昺hJxddlmZmZmZmZmZddlZddlmZddl m Z m Z m Z ddl mZdZGdde eZdS) )AnyDictListMappingOptionalN) Embeddings) BaseModelExtraroot_validatorget_from_dict_or_envz#sentence-transformers/clip-ViT-B-32ceZdZUdZeZeed< dZe ed< dZ eed< dZ eed< d Z e eed < d Ze eed <Gd d ZededefdZedeeeffdZdeedeeefdZdeedeeefdZdedeefdZd 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_tokenc eZdZdZejZdS)DeepInfraEmbeddings.Configz'Configuration for this pydantic object.N)__name__ __module__ __qualname____doc__r forbidextrap/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain_community/embeddings/deepinfra.pyConfigr5s55 rr!valuesreturnc2t|dd}||d<|S)z?Validate that api key and python package exists in environment.rDEEPINFRA_API_TOKENr )clsr"rs r validate_environmentz(DeepInfraEmbeddings.validate_environment:s/3 )+@  )<$% rcd|jiS)zGet the identifying parameters.r)r)selfs r _identifying_paramsz'DeepInfraEmbeddings._identifying_paramsCsDM**rinputc|jpi}d|jdd} tjd|j|||jd|}n.#tjj$r}td|d}~wwxYw|j dkrtd |j d |j  | }|d }n6#tjj $r}td |d |j d}~wwxYw|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: z, embeddingszError raised by inference API: z . Response: ) rrrequestspostrr exceptionsRequestException ValueError status_codetextr0JSONDecodeError)r)r+ _model_kwargsr/resetr2s r _embedzDeepInfraEmbeddings._embedHsY)/R Bt'?AA.    I-I$-II %DNTTmTCC "3 I I IGAGGHH H I ?c ! !*???CHH.   A<JJ"2   L!LL#(LL   s/)AA-A((A-B88C+ C&&C+textscNfd|D}|}|S)zEmbed documents using a Deep Infra deployed embedding model. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. c&g|] }j|Sr)r).0r9r)s r z7DeepInfraEmbeddings.embed_documents..qs'QQQ4 6>>>QQQr)r?)r)r@instruction_pairsr2s` r embed_documentsz#DeepInfraEmbeddings.embed_documentshs8RQQQ5QQQ[[!233 rr9cT|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)rr?)r)r9instruction_pair embeddings r embed_queryzDeepInfraEmbeddings.embed_queryus8#4rVs55555555555555000000GGGGGGGGGG5555558uuuuu)Zuuuuur