o Zh@sddlZddlZddlZddlmZmZmZddlZddlm Z ddl m Z ddl m Z ddlmZmZddlmZmZeeZe dd d Gd d d ee ZdS) N)DictListOptional) deprecated) Embeddings)run_in_executor)get_from_dict_or_envpre_init) BaseModel ConfigDictz0.0.13z8langchain_community.embeddings.QianfanEmbeddingsEndpoint)Zsince alternativec@seZdZUdZdZeeed<dZeeed<dZ eeed<dZ eeed<dZ e ed<d Z eed <eZed d Zed edefddZdedefddZd ddZdeedeeefddZdedeefddZdedeefddZdeedeeefddZdS)!ErnieEmbeddingsz'`Ernie Embeddings V1` embedding models.Nernie_api_baseernie_client_idernie_client_secret access_token chunk_sizezErnieBot-Embedding-V1 model_name)Zprotected_namespacesvaluesreturncCs6t|ddd|d<t|dd|d<t|dd|d<|S)NrZERNIE_API_BASEzhttps://aip.baidubce.comrZERNIE_CLIENT_IDrZERNIE_CLIENT_SECRET)r)clsrrr[/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/embeddings/ernie.pyvalidate_environment$sz$ErnieEmbeddings.validate_environmentjsoncCs6|jd}tj|dddid|ji|d}|S)Nz//rpc/2.0/ai_custom/v1/wenxinworkshop/embeddingsz /embedding-v1 Content-Typeapplication/jsonr)headersparamsr)rrequestspostrr)selfrbase_urlresprrr _embedding5s zErnieEmbeddings._embeddingcCsx|j/td|jd}tj|dddd|j|jdd}t| d|_ WddS1s5wYdS) NzRefreshing access tokenz/oauth/2.0/tokenr)rAcceptZclient_credentials)Z grant_typeZ client_idZ client_secret)rrr) _lockloggerdebugrr r!rrstrrgetr)r"r#r$rrr_refresh_access_token_with_lockCs   "z/ErnieEmbeddings._refresh_access_token_with_locktextscsjsfddtdtjD}g}|D]>}ddd|Di}|drN|ddkrGddd|Di}ntd ||d d|d Dq|S) zEmbed search docs. Args: texts: The list of texts to embed Returns: List[List[float]]: List of embeddings, one for each text. csg|] }||jqSr)r.0ir"r-rr asz3ErnieEmbeddings.embed_documents..rinputcSg|]}|qSrrr/textrrrr2g error_codeocSr4rrr5rrrr2kr7Error from Ernie: cSsg|]}|dqS) embeddingrr.rrrr2nsdata) rr,rangelenrr%r+ ValueErrorextend)r"r-Ztext_in_chunkslstchunkr$rr1rembed_documentsUs   zErnieEmbeddings.embed_documentsr6cCsn|js||d|gi}|dr/|ddkr(||d|gi}ntd||dddS)zEmbed query text. Args: text: The text to embed. Returns: List[float]: Embeddings for the text. r3r8r9r:r<rr;)rr,r%r+r?)r"r6r$rrr embed_queryqs  zErnieEmbeddings.embed_querycstd|j|IdHS)zAsynchronous Embed query text. Args: text: The text to embed. Returns: List[float]: Embeddings for the text. N)rrD)r"r6rrr aembed_querys zErnieEmbeddings.aembed_querycs(tjfdd|DIdH}t|S)zAsynchronous Embed search docs. Args: texts: The list of texts to embed Returns: List[List[float]]: List of embeddings, one for each text. csg|]}|qSr)rEr5r"rrr2sz4ErnieEmbeddings.aembed_documents..N)asynciogatherlist)r"r-resultrrFraembed_documentss z ErnieEmbeddings.aembed_documents)rN) __name__ __module__ __qualname____doc__rrr*__annotations__rrrrintr threadingLockr'r Z model_configr rrobjectdictr%r,rfloatrCrDrErKrrrrr s$     " r )rGloggingrRtypingrrrr Zlangchain_core._api.deprecationrZlangchain_core.embeddingsrZlangchain_core.runnables.configrZlangchain_core.utilsrr Zpydanticr r getLoggerrLr(r rrrrs