ai-g3sb-backman2.0/backend/utils/__pycache__/embedding.cpython-312.pyc

Ë
ˆêÞié#ã	óü—UdZd
dl
Z
d
dlZd
dlmZ
d
dlmZmZmZm	Z	
d
dl
Zd
dlZeje«
ZeGd„d««Zdefd„ZGd	„d
«ZeZdaeeed<			ddeed
eededefd„Zdd„Zy)u–
Embedding �装：通过 OpenAI 兼容 ``/v1/embeddings`` 远程 API 获���
（ModelScope / OpenAI / DashScope 等由环境��选择）。
éN)
Ú	dataclass)ÚAnyÚListÚOptionalÚUnioncóD—eZ
dZUd
Zeed<eed<eed<eed<eed<y)Ú_RemoteEmbeddingEnvuG从环境å�˜é‡�解æž�出的远程 OpenAI-Compatible Embedding é…�置。Úapi_keyÚbase_urlÚmodelÚ	max_batchÚlabelN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚstrÚ__annotations__Úint©óúbackend/utils/embedding.pyr	r	s…áQà
ƒL؃M؃J؃NØ„Jrr	ÚreturncóZ—t
jd
d«j«}t
jdd«j«}
t
jdd«j«}t
jd«
x
s
t
jdd«j«}|r¸|
x
s
d}t
jd«
x
s
t
jd	d
«j«}t
jdd«j«}|j«rt	d
t|«
«n
d}|st
d«
‚
t||jd«
||d¬«St
jdd«j«}|rð|jd«
s|jd«
rt
d«
‚
t
jdd«j«x
s
d}	t
jdd«j«}
|
st
d«
‚
t
jdd«j«}|j«rt	d
t|«
«n
d}t||	jd«
|
|d¬«S|st
d«
‚
|st
d «
‚
t
jd!d«j«}|st
d"«
‚
t
jd#d$«j«}|j«rt	d
t|«
«n
d%}t||jd«
||d&¬«S)'u`
    优先 ModelScope（MODELSCOPE_*）；未�置时回退 DashScope（DASHSCOPE_*）。
    ÚMODELSCOPE_API_KEYÚÚMODELSCOPE_BASE_URLÚDASHSCOPE_API_KEYÚDASHSCOPE_BASE_URLÚDASHSCOPE_base_urlz&https://api-inference.modelscope.cn/v1ÚMODELSCOPE_EMBEDDING_MODELÚMODELSCOPE_MODELzQwen/Qwen3-Embedding-8BÚMODELSCOPE_EMBEDDING_MAX_BATCHÚ32é
é u>未é…�ç½® MODELSCOPE_EMBEDDING_MODEL（或 MODELSCOPE_MODEL）Ú
/Ú
ModelScope©r
rrr
rÚOPENAI_API_KEYzhttp://zhttps://u‘OPENAI_API_KEY ä¸�能填写为 URL：请将网关地å�€å†™åˆ° OPENAI_BASE_URL（例如 http://host:9080/v1），密钥å�•ç‹¬å†™åœ¨ OPENAI_API_KEYÚOPENAI_BASE_URLzhttps://api.openai.com/v1ÚOPENAI_EMBEDDING_MODELuB使用 OpenAI 兼容 Embedding 时请设置 OPENAI_EMBEDDING_MODELÚOPENAI_EMBEDDING_MAX_BATCHÚ100édÚOpenAIuÝ远程 Embedding 未é…�置：请设置 MODELSCOPE_API_KEY（å�Šå�¯é€‰ BASE_URL），或 OPENAI_API_KEY / OPENAI_EMBEDDING_MODEL（å�Šå�¯é€‰ OPENAI_BASE_URL），或 DASHSCOPE_API_KEY / DASHSCOPE_BASE_URL / DASHSCOPE_MODELu8未é…�ç½® DASHSCOPE_BASE_URL（或 DASHSCOPE_base_url）ÚDASHSCOPE_MODELu未é…�ç½® DASHSCOPE_MODELÚDASHSCOPE_EMBEDDING_MAX_BATCHÚ10é
Ú	DashScope)
ÚosÚgetenvÚstripÚisdigitÚmaxrÚ
ValueErrorr	ÚrstripÚ
startswith)Úms_keyÚms_baseÚds_keyÚds_baserrÚmbr
Úoa_keyÚoa_baseÚoa_models           rÚ_remote_embedding_from_envrFsÉ€ô�Y‰YÐ+¨RÓ
0×
6Ñ
6Ó
8€FÜ�i‰iÐ-¨rÓ2×8Ñ8Ó:€GÜ
�Y‰YÐ*¨BÓ
/×
5Ñ
5Ó
7€Fä
�	‰	Ð&Ó'ÒN¬2¯9©9Ð5IÈ2Ó+Nß�eƒgðñ
ØÒFÐFˆä�I‰IÐ2Ó3ò

H
Ü�y‰yÐ+Ð-FÓGß
‰%‹'ð	ô�Y‰YÐ7¸Ó
>×
DÑ
DÓ
FˆØ')§z¡z¤|”C˜
œ3˜r›7”O¸ˆ	ÙÜÐ]Ó^Ð^Ü"ØØ—_‘_ SÓ)ØØØô
ð	
ô�Y‰YÐ'¨Ó
,×
2Ñ
2Ó
4€FÙ
Ø×јYÔ'¨6×+<Ñ+<¸ZÔ+HÜð
W
óð
ô

�I‰IÐ'¨Ó,×2Ñ2Ó4ÒSÐ8Sð	ô—9‘9Ð5°rÓ:×@Ñ@ÓBˆÙÜØTóð
ô�Y‰YÐ3°UÓ
;×
AÑ
AÓ
CˆØ')§z¡z¤|”C˜
œ3˜r›7”O¸ˆ	Ü"ØØ—^‘^ CÓ(ØØØô
ð	
ñÜð
K
ó
ð	
ñ
ÜÐSÓTÐTÜ�I‰IÐ'¨Ó,×2Ñ2Ó4€EÙÜÐ4Ó5Ð5Ü	�‰Ð2°DÓ	9×	?Ñ	?Ó	A€BØ#%§:¡:¤<”�A”s˜2“w”°R€IÜØØ—‘ Ó$ØØØôðrc
óD
—eZ
dZ					ddeedeedeedeedeef
d„Zedefd	„«Zd
e	e
dd
fd„Z				ddeee	efd
ede
dede
dejfd„Zdejdejdejfd„Z	ddede	edede	efd„Zy
)ÚOpenAICompatibleRemoteEmbeddingNr
rrr
Úprovider_labelcó4—	d
dlm
}
|
�i|�g|�et|
j	«|j	«jd«
|j	«|�t
dt|«
«n
d|x
s
d¬«}n
t«}|j|_	|j|_
|j|_|j|_
|j|_||j|j¬	«|_t#j$d
d«j	«}	|	j'«rt|	«
n
d|_t*j-d|j|j|j|j«
y#t$r}td«
|‚d}~w
wxYw
)
Nr)
r0u?使用远程 Embedding 需è¦�安装 openai：pip install openair'r%r&Úcustomr))r
rÚ
VECTOR_DIMru?使用 %s Embedding API：model=%s，base_url=%s，max_batch=%s)Úopenair0ÚImportErrorr	r8r<r:rrFr
rrr
Ú_api_max_batchrÚ_provider_labelÚ_clientr6r7r9Ú_embedding_dimÚloggerÚinfo)
Úselfr
rrr
rIr0Ú
eÚcfgÚvds