a bgP@snddlmZmZmZmZmZddlZddlmZddl m Z m Z ddl m Z mZdZdZGdd d e eZdS) )AnyDictListMappingOptionalN) Embeddings)get_from_dict_or_envpre_init) BaseModel ConfigDictz#sentence-transformers/clip-ViT-B-32ic@seZdZUdZeZeed<dZe ed<dZ eed<dZ eed<d Z e eed <d Ze eed <eZeed <ed ddZeeedddZeeeefdddZeeeeedddZeeeeedddZeeedddZ d S)DeepInfraEmbeddingsaDeep Infra's embedding inference service. To use, you should have the environment variable ``DEEPINFRA_API_TOKEN`` set with your API token, or pass it as a named parameter to the constructor. There are multiple embeddings models available, see https://deepinfra.com/models?type=embeddings. Example: .. code-block:: python from langchain_community.embeddings import DeepInfraEmbeddings deepinfra_emb = DeepInfraEmbeddings( model_id="sentence-transformers/clip-ViT-B-32", deepinfra_api_token="my-api-key" ) r1 = deepinfra_emb.embed_documents( [ "Alpha is the first letter of Greek alphabet", "Beta is the second letter of Greek alphabet", ] ) r2 = deepinfra_emb.embed_query( "What is the second letter of Greek alphabet" ) model_idF normalizez passage: embed_instructionzquery: query_instructionN model_kwargsdeepinfra_api_token batch_sizeZforbid)extraZprotected_namespaces)valuesreturncCst|dd}||d<|S)z?Validate that api key and python package exists in environment.rZDEEPINFRA_API_TOKEN)r)clsrrrrv/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/deepinfra.pyvalidate_environment;s z(DeepInfraEmbeddings.validate_environment)rcCs d|jiS)zGet the identifying parameters.r )r selfrrr_identifying_paramsDsz'DeepInfraEmbeddings._identifying_params)inputrc Cs|jpi}d|jdd}z(tjd|j|||jd|d}Wn6tjjyz}ztd|WYd}~n d}~00|j dkrtd |j |j fz| }|d }Wn>tjj y}z"td |d |j WYd}~n d}~00|S) Nzbearer zapplication/json) Authorizationz Content-Typez'https://api.deepinfra.com/v1/inference/)inputsr)headersjsonz$Error raised by inference endpoint: z/Error raised by inference API HTTP code: %s, %s embeddingszError raised by inference API: z . Response: ) rrrequestspostr r exceptionsRequestException ValueError status_codetextr"JSONDecodeError)rrZ _model_kwargsr!resetr$rrr_embedIs4    $   zDeepInfraEmbeddings._embed)textsrcsTg}fdd|DfddtdtjD}|D]}||7}q<|S)aQEmbed documents using a Deep Infra deployed embedding model. For larger batches, the input list of texts is chunked into smaller batches to avoid exceeding the maximum request size. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. csg|]}j|qSr)r).0r+rrr vz7DeepInfraEmbeddings.embed_documents..csg|]}||jqSr)r)r2iZinstruction_pairsrrrr3xsr)rangelenrr0)rr1r$chunkschunkrr6rembed_documentsis  z#DeepInfraEmbeddings.embed_documents)r+rcCs"|j|}||gd}|S)zEmbed a query using a Deep Infra deployed embedding model. Args: text: The text to embed. Returns: Embeddings for the text. r)rr0)rr+Zinstruction_pairZ embeddingrrr embed_querys zDeepInfraEmbeddings.embed_query)!__name__ __module__ __qualname____doc__DEFAULT_MODEL_IDr str__annotations__rboolrrrrdictrMAX_BATCH_SIZErintr Z model_configr rrpropertyrrrrfloatr0r;r<rrrrr s        r )typingrrrrrr%Zlangchain_core.embeddingsrZlangchain_core.utilsrr Zpydanticr r rArFr rrrrs