a bg@sddlmZmZmZmZddlmZddlmZddl m Z ddl m Z m Z mZddlmZeddd d Gd d d e eZd S))AnyDictListOptional) deprecated) Embeddings)get_from_dict_or_env) BaseModel ConfigDictmodel_validator)_create_retry_decoratorz0.0.30z1.0z!langchain_cohere.CohereEmbeddings)ZsinceZremovalZalternative_importc@sveZdZUdZdZeed<dZeed<dZe ed<dZ e e ed<dZ e e ed<d Z eed <dZe eed <d Ze ed <eddZeddeeedddZeedddZeedddZddee e e eeedddZddee e e eeedddZee eeed d!d"Zee eeed d#d$Ze eed%d&d'Ze eed%d(d)Z dS)*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" ) Nclient async_clientzembed-english-v2.0modeltruncatecohere_api_key max_retriesrequest_timeoutZ langchain user_agentZforbid)extrabefore)mode)valuesreturncCsrt|dd}|d}z._embed_with_retryr rrr,r&Zretry_decoratorr-r"r+r#embed_with_retryVs z!CohereEmbeddings.embed_with_retryc s0tj}|ttdfdd }|fi|S)r'r%csjjfi|IdHSr()rr)r*r+r"r#r-dsz=CohereEmbeddings.aembed_with_retry.._embed_with_retryr.r/r"r+r#aembed_with_retry`s z"CohereEmbeddings.aembed_with_retry input_type)textsr3rcCs&|j|j|||jdj}dd|DS)Nrr4r3rcSsg|]}ttt|qSr"listmapfloat.0er"r"r# sz*CohereEmbeddings.embed..)r0rr embeddingsr,r4r3r?r"r"r#r)jszCohereEmbeddings.embedcs,|j|j|||jdIdHj}dd|DS)Nr5cSsg|]}ttt|qSr"r6r:r"r"r#r=r>z+CohereEmbeddings.aembed..)r1rrr?r@r"r"r#aembedus zCohereEmbeddings.aembed)r4rcCs|j|ddS)zEmbed a list of document texts. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. search_documentr2r)r,r4r"r"r#embed_documentss z CohereEmbeddings.embed_documentscs|j|ddIdHS)zAsync call out to Cohere's embedding endpoint. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. rBr2NrArDr"r"r#aembed_documentss z!CohereEmbeddings.aembed_documents)textrcCs|j|gdddS)zCall out to Cohere's embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. search_queryr2rrCr,rHr"r"r# embed_querys zCohereEmbeddings.embed_querycs|j|gddIdHdS)zAsync call out to Cohere's embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. rIr2NrrFrJr"r"r# aembed_querys zCohereEmbeddings.aembed_query)!__name__ __module__ __qualname____doc__rr__annotations__rrstrrrrrintrr9rr Z model_configr classmethodrr$r0r1rr)rArErGrKrLr"r"r"r#r s>               r N)typingrrrrZlangchain_core._api.deprecationrZlangchain_core.embeddingsrZlangchain_core.utilsrZpydanticr r r Zlangchain_community.llms.coherer r r"r"r"r#s