o Zh(@sddlmZddlZddlmZmZmZmZmZddl m Z ddl m Z ddl mZmZmZddlmZddlmZmZmZmZmZeeZd d d d ZdddZdddZGdd d ee ZdS)) 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-v3 embeddingsDashScopeEmbeddingsreturnCallable[[Any], Any]cCs:d}d}d}tdt|jt|||dtttttj dS)NT)minmax)reraisestopwaitr before_sleep) rr max_retriesrrr rloggerloggingWARNING)r multiplierZ min_secondsZ max_secondsr'_/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/embeddings/dashscope.py_create_retry_decorators  r)kwargsrc s(t}|dfdd }|di|S) z)Use tenacity to retry the embedding call.r*rrcsg}d}|d}t|trt|nd}t|dd}||krut|tr,||||n||d<jjdi|}|jdkrF||jd7}n'|jdvr[t d |jd |j d |j t d |jd |j d |j |d ||7}||ks|S)Nrinputrmodelrr)iiz status_code: z code: z message: z"HTTP error occurred: status_code: )responser') isinstancelistlen BATCH_SIZEgetclientcall status_codeoutput ValueErrorcodemessager )r*resultiZ input_dataZ input_lenZ batch_sizeresprr'r(_embed_with_retry0s@    z+embed_with_retry.._embed_with_retryN)r*rrrr')r))rr*Zretry_decoratorr?r'r>r(embed_with_retry,sr@c@s|eZdZUdZdZded< dZded<dZded <d Zd ed < e d dZ e dde dddZ d ddZd!ddZdS)"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) Nrr4rstrr,z Optional[str]dashscope_api_keyintr"Zforbid)extrabefore)modevaluesrrcCsTddl} t|dd|d<|d|_z ddl}|j|d<W|Sty)tdw)NrrBZDASHSCOPE_API_KEYr4zZCould not import dashscope python package. Please install it with `pip install dashscope`.) dashscoper Zapi_keyZ TextEmbedding ImportError)clsrHrIr'r'r(validate_environmentxs   z(DashScopeEmbeddings.validate_environmenttexts List[str]List[List[float]]cCs$t||d|jd}dd|D}|S)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. Zdocumentr+ text_typer,cSsg|]}|dqS) embeddingr').0itemr'r'r( sz7DashScopeEmbeddings.embed_documents..r@r,)selfrMrZembedding_listr'r'r(embed_documentss  z#DashScopeEmbeddings.embed_documentstext List[float]cCst||d|jddd}|S)zCall out to DashScope's embedding endpoint for embedding query text. Args: text: The text to embed. Returns: Embedding for the text. queryrPrrRrV)rWrYrRr'r'r( embed_querys zDashScopeEmbeddings.embed_query)rHrrr)rMrNrrO)rYrArrZ)__name__ __module__ __qualname____doc__r4__annotations__r,rBr"r Z model_configr classmethodrLrXr\r'r'r'r(rQs      )rrrr)rrr*rrr) __future__rr$typingrrrrrZlangchain_core.embeddingsrZlangchain_core.utilsr Zpydanticr r r Zrequests.exceptionsr tenacityrrrrr getLoggerr]r#r2r)r@rr'r'r'r(s        %