U hN@s|ddlmZmZmZmZddlmZddlmZddl m Z m Z ddl m Z ddlmZeddd d Gd d d e eZd S))AnyDictListOptional) deprecated) Embeddings) BaseModelroot_validator)get_from_dict_or_env)_create_retry_decoratorz0.0.30z1.0z!langchain_cohere.CohereEmbeddings)ZsinceZremovalZalternative_importc@sneZdZUdZeed<eed<dZeed<dZe eed<dZ e eed<d Z e ed <dZ e eed <d Zeed <GdddZeddeedddZeedddZeedddZddeee eeeedddZddeee eeeedddZeeeeed d!d"Zeeeeed d#d$Zeeed%d&d'Zeeed%d(d)ZdS)*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_timeoutZ langchain user_agentc@seZdZdZdS)zCohereEmbeddings.ConfigZforbidN)__name__ __module__ __qualname__extrarrI/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/embeddings/cohere.pyConfig4srT)pre)valuesreturncCstt|dd}|d}z._embed_with_retryr rrr.r(Zretry_decoratorr/rr-rembed_with_retryTs z!CohereEmbeddings.embed_with_retryc s,tj}|ttdfdd }|f|S)r)r'csjjf|IdHSr*)rr+r,r-rrr/bsz=CohereEmbeddings.aembed_with_retry.._embed_with_retryr0r1rr-raembed_with_retry^s z"CohereEmbeddings.aembed_with_retry input_type)textsr5rcCs&|j|j|||jdj}dd|DS)Nrr6r5rcSsg|]}ttt|qSrlistmapfloat.0errr qsz*CohereEmbeddings.embed..)r2rr embeddingsr.r6r5r@rrrr+hszCohereEmbeddings.embedcs,|j|j|||jdIdHj}dd|DS)Nr7cSsg|]}ttt|qSrr8r<rrrr?~sz+CohereEmbeddings.aembed..)r3rrr@rArrraembedss zCohereEmbeddings.aembed)r6rcCs|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_documentr4r+r.r6rrrembed_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. rCr4NrBrErrraembed_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_queryr4rrDr.rIrrr 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. rJr4NrrGrKrrr aembed_querys zCohereEmbeddings.aembed_query)rrr__doc__r__annotations__rstrrrrrintrr;rrr rr&r2r3rr+rBrFrHrLrMrrrrr s<           r N)typingrrrrZlangchain_core._api.deprecationrZlangchain_core.embeddingsrZlangchain_core.pydantic_v1rr Zlangchain_core.utilsr Zlangchain_community.llms.coherer r rrrrs