Ng(ddlmZddlZddlmZmZmZmZmZddl m Z ddl m Z ddl mZmZmZddlmZddlmZmZmZmZmZejeZd d d d ZddZddZGdd 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]c d}d}d}tdt|jt|||t t t ttj S)NT)minmax)reraisestopwaitr before_sleep) rr max_retriesrrrrloggerloggingWARNING)r multiplier min_seconds max_secondss d/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/embeddings/dashscope.py_create_retry_decoratorr,sgJKK   6 7 7 jk{ K K K&y11%fgo>>    kwargsrc Nt}|dfd }|di|S)z)Use tenacity to retry the embedding call.r.rrc,g}d}|d}t|trt|nd}t|dd}||krt|tr ||||zn||d<jjdi|}|jdkr||jdz }nY|jdvr'td |jd |j d |j td |jd |j d |j | ||z }||k|S)Nrinputrmodelrr)iiz status_code: z code: z message: z"HTTP error occurred: status_code: )response) isinstancelistlen BATCH_SIZEgetclientcall status_codeoutput ValueErrorcodemessager)r.resulti input_data input_len batch_sizeresprs r+_embed_with_retryz+embed_with_retry.._embed_with_retry0s G_ '1*d'C'CJC OOO ^^F7OR88 )mmj$// 1q:~-.. 7O *:$)33F33D3&&$+l33!Z// DD$4DD!YDD59\DD  D9IDD!YDD59\DD! OA))mm* r-)r.rrrr5)r,)rr.retry_decoratorrHs` r+embed_with_retryrJ,sQ-j99O_:   & &v & &&r-ceZdZUdZdZded< dZded<dZded <d Zd ed < e d Z e de ddZ ddZddZdS)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) Nrr;rstrr2z Optional[str]dashscope_api_keyintr$forbid)extrabefore)modevaluesrrcddl} t|dd|d<|d|_ ddl}|j|d<n#t$rt dwxYw|S)NrrMDASHSCOPE_API_KEYr;zZCould not import dashscope python package. Please install it with `pip install dashscope`.) dashscoper api_key TextEmbedding ImportError)clsrTrWs r+validate_environmentz(DashScopeEmbeddings.validate_environmentxs M&: ')<' ' "###67      (6F8     B    s 7Atexts List[str]List[List[float]]cNt||d|j}d|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. documentr1 text_typer2cg|] }|d S) embeddingr5).0items r+ z7DashScopeEmbeddings.embed_documents..sCCC${+CCCr-rJr2)selfr]rembedding_lists r+embed_documentsz#DashScopeEmbeddings.embed_documentss?& 4:   DC CCCr-text List[float]cNt||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. queryrbrreri)rjrmres r+ embed_queryzDashScopeEmbeddings.embed_querys;% tz     r-)rTrrr)r]r^rr_)rmrLrrn)__name__ __module__ __qualname____doc__r;__annotations__r2rMr$r model_configr classmethodr\rlrqr5r-r+rrQs6F$E$$$$'+++++K<:L_(###[$#&          r-)rrrr)rrr.rrr) __future__rr&typingrrrrrlangchain_core.embeddingsr langchain_core.utilsr pydanticr r r requests.exceptionsrtenacityrrrrr getLoggerrrr%r9r,rJrr5r-r+rs""""""100000555555;;;;;;;;;;))))))  8 $ $#%BUV W W     "'"'"'"'JWWWWW)ZWWWWWr-