U h@snddlmZmZmZmZddlmZddlmZm Z m Z ddl m Z m Z mZddlmZGdddeeZdS) )AnyDictListOptional) Embeddings) BaseModelField SecretStr)convert_to_secret_strget_from_dict_or_envpre_init)Requestsc@seZdZUdZedddZeeed<dZ e ed<dZ ee ed<Gd d d Z e eed d d Zee dddZee eeedddZee eeedddZe eedddZdS)EdenAiEmbeddingsz~EdenAI embedding. environment variable ``EDENAI_API_KEY`` set with your API key, or pass it as a named parameter. NzEdenAI API Token) descriptionedenai_api_keyZopenaiprovidermodelc@seZdZdZdS)zEdenAiEmbeddings.ConfigZforbidN)__name__ __module__ __qualname__extrarrI/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/embeddings/edenai.pyConfigsr)valuesreturncCstt|dd|d<|S)z,Validate that api key exists in environment.rZEDENAI_API_KEY)r r )clsrrrrvalidate_environment"s z%EdenAiEmbeddings.validate_environment)rcCsddlm}d|S)Nr __version__z langchain/)Zlangchain_communityrrrrrget_user_agent*s zEdenAiEmbeddings.get_user_agent)textsrc Cs$d}ddd|j|d}||jd}|jdk rH|j|ji|d<t|d}|j||d }|jd kr|td |jn>|jd krt d |j n"|jdkrtd|jd|j | }||j}| ddkr| di d} t| g} ||jdD]} | d} | | q| S)z$Compute embeddings using EdenAi api.z)https://api.edenai.run/v2/text/embeddingszapplication/jsonzBearer )acceptz content-type authorizationz User-Agent)r!Z providersNsettings)headers)urldataizEdenAI Server: Error iz$EdenAI received an invalid payload: z3EdenAI returned an unexpected response with status z: statusZfailerrormessageitems embedding)rZget_secret_valuer rrr post status_code Exception ValueErrortextjsongetappend) selfr!r&r%payloadrequestresponsetempZprovider_responseerr_msgZ embeddingsZ embed_itemr-rrr_generate_embeddings0s:       z%EdenAiEmbeddings._generate_embeddingscCs ||S)zEmbed a list of documents using EdenAI. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. r<)r6r!rrrembed_documents[s z EdenAiEmbeddings.embed_documents)r2rcCs||gdS)zEmbed a query using EdenAI. Args: text: The text to embed. Returns: Embeddings for the text. rr=)r6r2rrr embed_querygs zEdenAiEmbeddings.embed_query)rrr__doc__rrrr __annotations__rstrrrr rr staticmethodr rfloatr<r>r?rrrrrs  + rN)typingrrrrZlangchain_core.embeddingsrZlangchain_core.pydantic_v1rrr Zlangchain_core.utilsr r r Z&langchain_community.utilities.requestsr rrrrrs