NgUUddlmZmZmZddlZddlmZddlmZddl m Z m Z m Z m Z mZddlmZddlmZdZeed <Gd d e eZdS) )AnyListOptionalN) Embeddings)secret_from_env) BaseModel ConfigDictField SecretStrmodel_validator)RequestException)Selfz)https://api.baichuan-ai.com/v1/embeddingsBAICHUAN_API_URLceZdZUdZdZeed<eddZe ed< ede d d g Z e ed < d Z eed< eddZeddefdZdee deeeefdZdee deeeefdZde deeefdZdS)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_nameapi_keyBAICHUAN_API_KEYBAICHUAN_AUTH_TOKEN)rdefault_factorybaichuan_api_key chunk_sizeT)populate_by_nameprotected_namespacesafter)modereturnctj}|jd|jddd||_|S)z/Validate that auth token exists in environment.zBearer identityzapplication/json) AuthorizationzAccept-Encodingz Content-type)requestsSessionheadersupdaterget_secret_valuer)selfrs c/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/embeddings/baichuan.pyvalidate_environmentz+BaichuanTextEmbeddings.validate_environmentKsd"$$!U4+@+Q+Q+S+S!U!U#- 2       textscfdtdtjD}g}|D]}jt |jd}||jdkr\| }| dg}t|d}| d |Dtd |jd |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. c4g|]}||jzSr)r).0ir,r0s r- z1BaichuanTextEmbeddings._embed..cs:    !a$/)) *   r/r)inputr)jsondatac.|ddS)Nindexrget)es r-z/BaichuanTextEmbeddings._embed..ssQUU7TUEVEVr/)keyc:g|]}|dgS) embeddingr<)r3results r-r5z1BaichuanTextEmbeddings._embed..vs&QQQVVZZ R00QQQr/zError: Received status code z from `BaichuanEmbedding` API)rangelenrrpostrrraise_for_status status_coder7r=sortedextendr ) r,r0 chunk_texts embed_resultschunkresponseresp embeddingssorted_embeddingss `` r-_embedzBaichuanTextEmbeddings._embedYsC     1c%jj$/::       E|(( 'Q'Q)H  % % ' ' '#s**}}!XXfb11 $*:;V;V$W$W$W!$$QQ?PQQQ '.83G...r/c,||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. rR)r,r0s r-embed_documentsz&BaichuanTextEmbeddings.embed_documentss{{5!!!r/textcF||g}||dndS)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. NrrT)r,rVrCs r- embed_queryz"BaichuanTextEmbeddings.embed_querys*dV$$".vayyD8r/)__name__ __module__ __qualname____doc__rr__annotations__r rstrrrr rintr model_configr rr.rrfloatrRrUrXrr/r-rr"s6GSe$=WMMMJMMM0"'%');=R(STT###iRJ2:t"MMML_'""" d   #" &DI&(4U 3D*E&&&&P "T#Y "8Dergs&&&&&&&&&&&000000&%%%%%""""""C#CCC t9t9t9t9t9Y t9t9t9t9t9r/