a bgU@sUddlmZmZmZddlZddlmZddlmZddl m Z m Z m Z m Z mZddlmZddlmZdZeed <Gd d d e eZdS) )AnyListOptionalN) Embeddings)secret_from_env) BaseModel ConfigDictField SecretStrmodel_validator)RequestException)Selfz)https://api.baichuan-ai.com/v1/embeddingsBAICHUAN_API_URLc@seZdZUdZdZeed<edddZe ed<ede d d gd Z e ed <d Z eed<edddZeddedddZee eeeedddZee eeeedddZe eeedddZdS)BaichuanTextEmbeddingsaBaichuan Text Embedding models. Setup: To use, you should set the environment variable ``BAICHUAN_API_KEY`` to your API key or pass it as a named parameter to the constructor. .. code-block:: bash export BAICHUAN_API_KEY="your-api-key" Instantiate: .. code-block:: python from langchain_community.embeddings import BaichuanTextEmbeddings embeddings = BaichuanTextEmbeddings() Embed: .. code-block:: python # embed the documents vectors = embeddings.embed_documents([text1, text2, ...]) # embed the query vectors = embeddings.embed_query(text) NsessionzBaichuan-Text-Embeddingmodel)defaultalias model_nameZapi_keyZBAICHUAN_API_KEYZBAICHUAN_AUTH_TOKEN)rdefault_factorybaichuan_api_key chunk_sizeT)Zpopulate_by_nameZprotected_namespacesafter)mode)returncCs2t}|jd|jddd||_|S)z/Validate that auth token exists in environment.zBearer identityzapplication/json) AuthorizationzAccept-Encodingz Content-type)requestsSessionheadersupdaterZget_secret_valuer)selfrrru/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/baichuan.pyvalidate_environmentKsz+BaichuanTextEmbeddings.validate_environment)textsrc sfddtdtjD}g}|D]z}jjt|jdd}||jdkr| }| dg}t |dd d }| d d|Dq*t d |jd q*|S)aInternal method to call Baichuan Embedding API and return embeddings. Args: texts: A list of texts to embed. Returns: A list of list of floats representing the embeddings, or None if an error occurs. csg|]}||jqSr)r).0ir#r&rr$ csz1BaichuanTextEmbeddings._embed..r)inputr)jsondatacSs |ddS)Nindexrget)errr$sz/BaichuanTextEmbeddings._embed..)keycSsg|]}|dgqS)Z embeddingr0)r'resultrrr$r*vr4zError: Received status code z from `BaichuanEmbedding` API)rangelenrrpostrrraise_for_status status_coder,r1sortedextendr ) r#r&Z chunk_textsZ embed_resultschunkresponserespZ embeddingsZsorted_embeddingsrr)r$_embedYs(      zBaichuanTextEmbeddings._embedcCs ||S)zPublic method to get embeddings for a list of documents. Args: texts: The list of texts to embed. Returns: A list of embeddings, one for each text, or None if an error occurs. rAr)rrr$embed_documentss z&BaichuanTextEmbeddings.embed_documents)textrcCs ||g}|dur|dSdS)zPublic method to get embedding for a single query text. Args: text: The text to embed. Returns: Embeddings for the text, or None if an error occurs. NrrB)r#rDr6rrr$ embed_querys z"BaichuanTextEmbeddings.embed_query)__name__ __module__ __qualname____doc__rr__annotations__r rstrrrr rintrZ model_configr r r%rrfloatrArCrErrrr$r"s      ( r)typingrrrrZlangchain_core.embeddingsrZlangchain_core.utilsrZpydanticrrr r r r Ztyping_extensionsr rrKrJrrrrr$s