U h@sddlmZddlZddlmZmZmZmZmZddl m Z ddl m Z m Z ddlmZddlmZddlmZmZmZmZmZeeZd d d d d Zd dddddZGdd d e e ZdS)) annotationsN)AnyCallableDictListOptional) Embeddings) BaseModelroot_validator)get_from_dict_or_env) HTTPError)before_sleep_logretryretry_if_exception_typestop_after_attemptwait_exponentialDashScopeEmbeddingszCallable[[Any], Any]) embeddingsreturncCs:d}d}d}tdt|jt|||dtttttj dS)NT)minmax)reraisestopwaitrZ before_sleep) rr max_retriesrrr r loggerloggingWARNING)rZ multiplierZ min_secondsZ max_secondsr L/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/embeddings/dashscope.py_create_retry_decorators  r"r)rkwargsrc s*t}|dddfdd }|f|S)z)Use tenacity to retry the embedding call.r)r#rcsg}d}|d}t|tr"t|nd}||krt|trH|||dn||d<jjf|}|jdkrx||jd7}nP|jdkrtd|jd |jd |j n$t d |jd |jd |j |d |d7}q&|S) Nrinputrr)iiz status_code: z code: z message: z"HTTP error occurred: status_code: )response) isinstancelistlenclientcall status_codeoutput ValueErrorcodemessager )r#resultiZ input_dataZ input_lenresprr r!_embed_with_retry.s(   z+embed_with_retry.._embed_with_retry)r")rr#Zretry_decoratorr6r r5r!embed_with_retry*sr7c@seZdZUdZded<dZded<dZded <d Zd ed <Gd ddZe dddddddZ dddddZ dddddZ 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) rr+ztext-embedding-v1strmodelNz Optional[str]dashscope_api_keyintrc@seZdZdZdS)zDashScopeEmbeddings.ConfigZforbidN)__name__ __module__ __qualname__extrar r r r!ConfigosrAT)prer)valuesrcCsZddl}t|dd|d<|d|_zddl}|j|d<Wntk rTtdYnX|S)Nrr:ZDASHSCOPE_API_KEYr+zZCould not import dashscope python package. Please install it with `pip install dashscope`.) dashscoper Zapi_keyZ TextEmbedding ImportError)clsrCrDr r r!validate_environmentrs  z(DashScopeEmbeddings.validate_environmentz List[str]zList[List[float]])textsrcCs$t||d|jd}dd|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_typer9cSsg|] }|dqS) embeddingr ).0itemr r r! sz7DashScopeEmbeddings.embed_documents..r7r9)selfrHrZembedding_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. queryrJrrLrP)rQrSrLr r r! embed_querys zDashScopeEmbeddings.embed_query) r=r>r?__doc____annotations__r9r:rrAr rGrRrUr r r r!rLs    ) __future__rrtypingrrrrrZlangchain_core.embeddingsrZlangchain_core.pydantic_v1r r Zlangchain_core.utilsr Zrequests.exceptionsr Ztenacityr rrrr getLoggerr=rr"r7rr r r r!s     "