o Zh+@srddlmZmZmZmZddlmZddlmZm Z m Z ddl m Z m Z mZmZddlmZGddde eZdS) )AnyDictListOptional) Embeddings)convert_to_secret_strget_from_dict_or_envpre_init) BaseModel ConfigDictField SecretStr)Requestsc@seZdZUdZedddZeeed<dZ e ed< dZ ee ed< e d d Z ed ed efd dZed e fddZdee d eeefddZdee d eeefddZde d eefddZdS)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_keyZopenaiprovidermodelZforbid)extravaluesreturncCstt|dd|d<|S)z,Validate that api key exists in environment.rZEDENAI_API_KEY)rr)clsrr\/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/embeddings/edenai.pyvalidate_environment$s z%EdenAiEmbeddings.validate_environmentcCsddlm}d|S)Nr __version__z langchain/)Zlangchain_communityrrrrrget_user_agent,s  zEdenAiEmbeddings.get_user_agenttextsc Csd}ddd|j|d}||jd}|jdur$|j|ji|d<t|d}|j||d }|jd kr=td |j|jd krJt d |j |jdkr[td|jd|j | }||j}| ddkrx| 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 providersNsettings)headers)urldataizEdenAI Server: Error iz$EdenAI received an invalid payload: z3EdenAI returned an unexpected response with status z: statusZfailerrormessageitems embedding)rZget_secret_valuerrrrpost status_code Exception ValueErrortextjsongetappend) selfrr$r#payloadrequestresponsetempZprovider_responseerr_msgZ embeddingsZ embed_itemr+rrr_generate_embeddings2sB        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:)r4rrrrembed_documents]s z EdenAiEmbeddings.embed_documentsr0cCs||gdS)zEmbed a query using EdenAI. Args: text: The text to embed. Returns: Embeddings for the text. rr;)r4r0rrr embed_queryis zEdenAiEmbeddings.embed_query)__name__ __module__ __qualname____doc__r rrr __annotations__rstrrr Z model_configr rr staticmethodrrfloatr:r<r=rrrrrs"  + rN)typingrrrrZlangchain_core.embeddingsrZlangchain_core.utilsrrr Zpydanticr r r r Z&langchain_community.utilities.requestsrrrrrrs