a bg(@sddlmZddlZddlmZmZmZmZmZddl m Z ddl m Z ddl mZmZmZddlmZddlmZmZmZmZmZeeZd d d d Zd d dddZd dddddZGdd 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-v3DashScopeEmbeddingszCallable[[Any], Any]) embeddingsreturncCs:d}d}d}tdt|jt|||dtttttj dS)NT)minmax)reraisestopwaitr before_sleep) rr max_retriesrrr rloggerloggingWARNING)r multiplierZ min_secondsZ max_secondsr&v/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/dashscope.py_create_retry_decorators  r(r)rkwargsrc s.t}|dddfdd }|fi|S)z)Use tenacity to retry the embedding call.r)r)rcsg}d}|d}t|tr"t|nd}t|dd}||krt|trX||||n||d<jjfi|}|jdkr||jd7}nP|jdvrt d |jd |j d |j n$t d |jd |j d |j |d ||7}q6|S)Nrinputrmodelrr)iiz status_code: z code: z message: z"HTTP error occurred: status_code: )response) isinstancelistlen BATCH_SIZEgetclientcall status_codeoutput ValueErrorcodemessager )r)resultiZ input_dataZ input_lenZ batch_sizeresprr&r'_embed_with_retry0s>     z+embed_with_retry.._embed_with_retry)r()rr)Zretry_decoratorr>r&r=r'embed_with_retry,sr?c@seZdZUdZdZded<dZded<dZded <d Zd ed <e d dZ e dde dddddZ dddddZdddddZdS)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) Nrr3rstrr+z Optional[str]dashscope_api_keyintr!Zforbid)extrabefore)moder)valuesrcCsXddl}t|dd|d<|d|_zddl}|j|d<WntyRtdYn0|S)NrrAZDASHSCOPE_API_KEYr3zZCould not import dashscope python package. Please install it with `pip install dashscope`.) dashscoper Zapi_keyZ TextEmbedding ImportError)clsrGrHr&r&r'validate_environmentxs   z(DashScopeEmbeddings.validate_environmentz List[str]zList[List[float]])textsrcCs$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' z7DashScopeEmbeddings.embed_documents..r?r+)selfrLrZembedding_listr&r&r'embed_documentss  z#DashScopeEmbeddings.embed_documentsz List[float])textrcCst||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. queryrMrrOrT)rUrWrOr&r&r' embed_querys  zDashScopeEmbeddings.embed_query)__name__ __module__ __qualname____doc__r3__annotations__r+rAr!r Z model_configr classmethodrKrVrYr&r&r&r'rQs     ) __future__rr#typingrrrrrZlangchain_core.embeddingsrZlangchain_core.utilsr Zpydanticr r r Zrequests.exceptionsr tenacityrrrrr getLoggerrZr"r1r(r?rr&r&r&r's      %