7|h(ddlmZddlZddlmZmZmZmZmZddl m Z ddl m Z ddl mZmZmZddlmZddlmZmZmZmZmZej2eZd d d d Zdd ZddZGdd ee Zy)) annotationsN)AnyCallableDictListOptional) Embeddings)get_from_dict_or_env) BaseModel ConfigDictmodel_validator) HTTPError)before_sleep_logretryretry_if_exception_typestop_after_attemptwait_exponential)text-embedding-v1ztext-embedding-v2ztext-embedding-v3DashScopeEmbeddingsc d}d}d}tdt|jt|||t t t ttjS)NT)minmax)reraisestopwaitr before_sleep) rr max_retriesrrrrloggerloggingWARNING) embeddings multiplier min_seconds max_secondss g/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/embeddings/dashscope.py_create_retry_decoratorr*sSJKK   6 6 7 jk{ K&y1%fgoo>  c @t}|dfd }|di|S)z)Use tenacity to retry the embedding call.cDg}d}|d}t|tr t|nd}tj |dd}||krt|tr||||zn||d<j j di|}|jdk(r||jdz }nt|jdvr2td |jd |jd |jtd |jd |jd |j| ||z }||kr|S)Nrinputrmodelrr%)iiz status_code: z code: z message: z"HTTP error occurred: status_code: )response) isinstancelistlen BATCH_SIZEgetclientcall status_codeoutput ValueErrorcodemessager)kwargsresulti input_data input_len batch_sizerespr%s r)_embed_with_retryz+embed_with_retry.._embed_with_retry0sL G_ '1*d'CC O ^^F7OR8 )mj$/1q:~. 7O *:$$))3F3D3&$++l33!!Z/ #D$4$4#56!YYK}T\\ND  89I9I8JK!YYK}T\\ND! OA))m* r+)r?rreturnrr2)r*)r%r?retry_decoratorrFs` r)embed_with_retryrI,s/-j9O:  &v &&r+ceZdZUdZdZded< dZded<dZded <d Zd ed < e d Z e de ddZ ddZddZy)ra`DashScope embedding models. To use, you should have the ``dashscope`` python package installed, and the environment variable ``DASHSCOPE_API_KEY`` set with your API key or pass it as a named parameter to the constructor. Example: .. code-block:: python from langchain_community.embeddings import DashScopeEmbeddings embeddings = DashScopeEmbeddings(dashscope_api_key="my-api-key") Example: .. code-block:: python import os os.environ["DASHSCOPE_API_KEY"] = "your DashScope API KEY" from langchain_community.embeddings.dashscope import DashScopeEmbeddings embeddings = DashScopeEmbeddings( model="text-embedding-v1", ) text = "This is a test query." query_result = embeddings.embed_query(text) Nrr8rstrr/z Optional[str]dashscope_api_keyintr!forbid)extrabefore)modecddl} t|dd|d<|d|_ ddl}|j|d<|S#t$r t dwxYw)NrrLDASHSCOPE_API_KEYr8zZCould not import dashscope python package. Please install it with `pip install dashscope`.) dashscoper api_key TextEmbedding ImportError)clsvaluesrUs r)validate_environmentz(DashScopeEmbeddings.validate_environmentxsv M&: ')<' "###67   (66F8    B  s 6A cft||d|j}|Dcgc]}|d }}|Scc}w)zCall out to DashScope's embedding endpoint for embedding search docs. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. documentr. text_typer/ embeddingrIr/)selftextsr%itemembedding_lists r)embed_documentsz#DashScopeEmbeddings.embed_documentssB& 4:: 9CC${+CCDs .cDt||d|jdd}|S)zCall out to DashScope's embedding endpoint for embedding query text. Args: text: The text to embed. Returns: Embedding for the text. queryr^rr`ra)rbtextr`s r) embed_queryzDashScopeEmbeddings.embed_querys3% tzz   r+)rZrrGr)rcz List[str]rGzList[List[float]])rirKrGz List[float])__name__ __module__ __qualname____doc__r8__annotations__r/rLr!r model_configr classmethodr[rfrjr2r+r)rrQsp6FC$E3$'+}+K<L(#$&  r+)r%rrGzCallable[[Any], Any])r%rr?rrGr) __future__rr#typingrrrrrlangchain_core.embeddingsr langchain_core.utilsr pydanticr r r requests.exceptionsrtenacityrrrrr getLoggerrkr"r6r*rIrr2r+r)rzsl"15;;)   8 $#%BUV W  "'JW)ZWr+