昺h4hddlmZmZmZmZddlmZddlmZm Z m Z ddl m Z GddeeZ dS))AnyDictListOptional) Embeddings) BaseModelExtraroot_validator)get_from_dict_or_envc 8eZdZUdZeed< eed< dZeed< dZe eed< dZ e eed<dZ e e ed < dZ e eed < d Zeed < Gd dZededefdZdddeede edeeefdZdddeede edeeefdZdeedeeefdZdeedeeefdZdedeefdZdedeefdZdS)CohereEmbeddingsaCohere embedding models. To use, you should have the ``cohere`` python package installed, and the environment variable ``COHERE_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 CohereEmbeddings cohere = CohereEmbeddings( model="embed-english-light-v3.0", cohere_api_key="my-api-key" ) client async_clientzembed-english-v2.0modelNtruncatecohere_api_key max_retriesrequest_timeout langchain user_agentc eZdZdZejZdS)CohereEmbeddings.Configz'Configuration for this pydantic object.N)__name__ __module__ __qualname____doc__r forbidextram/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain_community/embeddings/cohere.pyConfigr,s55 r r"valuesreturncBt|dd}|d}|d} ddl}|d}||||||d <||||||d <n#t $rt d wxYw|S) z?Validate that api key and python package exists in environment.rCOHERE_API_KEYrrrNr)rtimeout client_namerrzTCould not import cohere python package. Please install it with `pip install cohere`.)r getcohereClient AsyncClient ImportError ValueError)clsr#rrrr*r(s r!validate_environmentz%CohereEmbeddings.validate_environment1s. $&6  jj//  **%677  MMM .K%}}''' -  F8  &,%7%7''' &8&&F> " "    ?    s ABB input_typetextsr2ct|j|j|||jj}d|DS)Nrr3r2rcRg|]$}ttt|%Srlistmapfloat.0es r! z*CohereEmbeddings.embed..Z(888S]]##888r )rembedrr embeddingsselfr3r2rAs r!r@zCohereEmbeddings.embedQsN[&&*!] '    98Z8888r cK|j|j|||jjd{V}d|DS)Nr5cRg|]$}ttt|%Srr7r;s r!r>z+CohereEmbeddings.aembed..er?r )rr@rrrArBs r!aembedzCohereEmbeddings.aembed\sn ,22*!] 3    98Z8888r c0||dS)zEmbed a list of document texts. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. search_documentr1r@rCr3s r!embed_documentsz CohereEmbeddings.embed_documentsgszz%,=z>>>r c@K||dd{VS)zAsync call out to Cohere's embedding endpoint. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. rHr1NrFrJs r!aembed_documentsz!CohereEmbeddings.aembed_documentsrs1[[3D[EEEEEEEEEr textc>||gddS)zCall out to Cohere's embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. search_queryr1rrIrCrOs r! embed_queryzCohereEmbeddings.embed_query}s!zz4&^z<r]s,,,,,,,,,,,,000000GGGGGGGGGG555555IIIIIIIIIIy*IIIIIIIIIIr