昺h`ddlmZddlZddlmZmZmZmZmZddl m Z ddl m Z m Z mZddlmZddlmZddlmZmZmZmZmZejeZdd ZddZGdd e e ZdS)) annotationsN)AnyCallableDictListOptional) Embeddings) BaseModelExtraroot_validator)get_from_dict_or_env) HTTPError)before_sleep_logretryretry_if_exception_typestop_after_attemptwait_exponential 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 p/var/www/html/services/rag-service/venv/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+rrcg}d}|d}|t|kr|||dz|d<jjd i|}|jdkr||jdz }nY|jdvr't d|jd|jd |jtd |jd|jd |j| |dz }|t|k|S) Nrinputr)iiz status_code: z code: z message: z"HTTP error occurred: status_code: )response) lenclientcall status_codeoutput ValueErrorcodemessager)r+resulti input_dataresprs r(_embed_with_retryz+embed_with_retry.._embed_with_retry.sL G_ #j//!!(QV4F7O):$)33F33D3&&$+l33!Z// DD$4DD!YDD59\DD  D9IDD!YDD59\DD! GA!#j//!!" r*)r+rrrr2)r))rr+retry_decoratorr?s` r(embed_with_retryrA*sQ-j99O_.   & &v & &&r*ceZdZUdZded< dZded<dZded <d Zd ed < Gd dZe ddZ ddZ ddZ dS)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) rr4ztext-embedding-v1strmodelNz Optional[str]dashscope_api_keyintr!c eZdZdZejZdS)DashScopeEmbeddings.Configz'Configuration for this pydantic object.N)__name__ __module__ __qualname____doc__r forbidextrar2r*r(ConfigrIls55 r*rPvaluesrrcddl} t|dd|d<|d|_ ddl}|j|d<n#t$rt dwxYw|S)NrrEDASHSCOPE_API_KEYr4zZCould not import dashscope python package. Please install it with `pip install dashscope`.) dashscoper api_key TextEmbedding ImportError)clsrQrTs r(validate_environmentz(DashScopeEmbeddings.validate_environmentqsM&: ')<' ' "###67      (6F8     B    s 7Atexts List[str]List[List[float]]cNt||d|j}d|D}|S)aPCall out to DashScope's embedding endpoint for embedding search docs. Args: texts: The list of texts to embed. chunk_size: The chunk size of embeddings. If None, will use the chunk size specified by the class. Returns: List of embeddings, one for each text. documentr. text_typerDcg|] }|d S) embeddingr2).0items r( z7DashScopeEmbeddings.embed_documents..sCCC${+CCCr*rArD)selfrZrembedding_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. queryr_rrbrf)rgrjrbs r( embed_queryzDashScopeEmbeddings.embed_querys;% tz     r*)rQrrr)rZr[rr\)rjrCrrk) rJrKrLrM__annotations__rDrEr!rPr rYrirnr2r*r(rrIs6KKK$E$$$$'+++++K< ^&"      r*)rrrr)rrr+rrr) __future__rr#typingrrrrrlangchain_core.embeddingsr langchain_core.pydantic_v1r r r langchain_core.utilsr requests.exceptionsrtenacityrrrrr getLoggerrJr"r)rArr2r*r(rxs~""""""100000GGGGGGGGGG555555))))))  8 $ $    ''''>YYYYY)ZYYYYYr*