jUddlmZddlZddlmZmZmZmZmZddl m Z ddl m Z ddl mZmZmZddlmZddlmZmZmZmZmZej2eZd d 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-v3ztext-embedding-v4DashScopeEmbeddingsc d}d}d}tdt|jt|||t t t ttjS)NT)minmax)reraisestopwaitr before_sleep) rr max_retriesrrrrloggerloggingWARNING) embeddings multiplier min_seconds max_secondss r/var/www/html/Fitness-lenito-AI-main/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_retry5sL 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_retryrI1s/-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_environment}sv 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_documentssC& 4:: 9CC ${+ CDs .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)rrVsp6FC$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)rzst"15;;)   8 $   "'JW)ZWr+