7|hUUddlmZmZmZddlZddlmZddlmZddl m Z m Z m Z m Z mZddlmZddlmZdZeed <Gd d e eZy) )AnyListOptionalN) Embeddings)secret_from_env) BaseModel ConfigDictField SecretStrmodel_validator)RequestException)Selfz)https://api.baichuan-ai.com/v1/embeddingsBAICHUAN_API_URLc$eZdZUdZdZeed<eddZe ed< ede d d g Z e ed < d Z eed< eddZeddefdZdee deeeefdZdee deeeefdZde deeefdZy)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}|jjd|jj ddd||_|S)z/Validate that auth token exists in environment.zBearer identityzapplication/json) AuthorizationzAccept-Encodingz Content-type)requestsSessionheadersupdaterget_secret_valuer)selfrs f/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/embeddings/baichuan.pyvalidate_environmentz+BaichuanTextEmbeddings.validate_environmentKsW""$#*4+@+@+Q+Q+S*T!U#- 2    textsc 0tdt||jDcgc]}||||jz}}g}|D]}|jj t ||j d}|j|jdk(r`|j}|jdg}t|d} |j| D cgc]} | jdgc} td |jd |Scc}wcc} w) 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. r)inputr)jsondatac&|jddS)Nindexr)get)es r-z/BaichuanTextEmbeddings._embed..ssQUU7TUEVr/)key embeddingzError: Received status code z from `BaichuanEmbedding` API)rangelenrrpostrrraise_for_status status_coder3r8sortedextendr ) r,r0i chunk_texts embed_resultschunkresponseresp embeddingssorted_embeddingsresults r-_embedzBaichuanTextEmbeddings._embedYs&1c%j$//:  !a$//) *     E||(( 'Q)H  % % '##s*}}!XXfb1 $*:;V$W!$$?PQVVZZ R0Q '283G3G2HI..' .9 &Rs DD c$|j|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. rM)r,r0s r-embed_documentsz&BaichuanTextEmbeddings.embed_documentss{{5!!r/textc8|j|g}||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. NrrO)r,rQrLs r- embed_queryz"BaichuanTextEmbeddings.embed_querys(dV$".vay8D8r/)__name__ __module__ __qualname____doc__rr__annotations__r rstrrrr rintr model_configr rr.rrfloatrMrPrSrr/r-rr"s6GS$=WMJM0"'');=R(ST#iRJ2t"ML'" d # &DI&(4U 3D*E&P "T#Y "8DerbsC&&0&"C#C t9Y t9r/