o Ag(@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 multiplier min_seconds max_secondsr)o/var/www/html/development/chatbot/venv/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,resulti input_data input_len batch_sizeresprr)r*_embed_with_retry0s@    z+embed_with_retry.._embed_with_retryN)r,rrrr))r+)rr,retry_decoratorrDr)rCr*embed_with_retry,srFc@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) Nrr6rstrr.z Optional[str]dashscope_api_keyintr"forbid)extrabefore)modevaluesrrcCsTddl} t|dd|d<|d|_z ddl}|j|d<W|Sty)tdw)NrrHDASHSCOPE_API_KEYr6zZCould not import dashscope python package. Please install it with `pip install dashscope`.) dashscoper api_key TextEmbedding ImportError)clsrOrQr)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. documentr- text_typer.cSsg|]}|dqS) embeddingr)).0itemr)r)r* sz7DashScopeEmbeddings.embed_documents..rFr.)selfrWrembedding_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. queryr[rr]ra)rbrer]r)r)r* embed_querys zDashScopeEmbeddings.embed_query)rOrrr)rWrXrrY)rerGrrf)__name__ __module__ __qualname____doc__r6__annotations__r.rHr"r model_configr classmethodrVrdrhr)r)r)r*rQs      )rrrr)rrr,rrr) __future__rr$typingrrrrrlangchain_core.embeddingsrlangchain_core.utilsr pydanticr r r requests.exceptionsr tenacityrrrrr getLoggerrir#r4r+rFrr)r)r)r*s        %