NgFddlmZddlZddlmZmZmZmZddlm Z ddl m Z m Z ddl mZejeZGddee ZdS) ) annotationsN)AnyDictListOptional) Embeddings)get_from_dict_or_envpre_init) BaseModelceZdZUdZdZded< dZded< dZded< d Zded < d Z ded < d Z ded< dZ ded< ded< e ddZ d dZd!dZdS)"VolcanoEmbeddingsz)`Volcengine Embeddings` embedding models.Nz Optional[str] volcano_ak volcano_skz*maas-api.ml-platform-cn-beijing.volces.comstrhostz cn-beijingregionz bge-large-zhmodelz1.0versiondint chunk_sizerclientvaluesrreturncFt|dd|d<t|dd|d< ddlm}||d|d}||d||d||d <n#t $rt d wxYw|S) a6 Validate whether volcano_ak and volcano_sk in the environment variables or configuration file are available or not. init volcano embedding client with `ak`, `sk`, `host`, `region` Args: values: a dictionary containing configuration information, must include the fields of volcano_ak and volcano_sk Returns: a dictionary containing configuration information. If volcano_ak and volcano_sk are not provided in the environment variables or configuration file,the original values will be returned; otherwise, values containing volcano_ak and volcano_sk will be returned. Raises: ValueError: volcengine package not found, please install it with `pip install volcengine` rVOLC_ACCESSKEYrVOLC_SECRETKEYr) MaasServicerrrzMvolcengine package not found, please install it with `pip install volcengine`)r volcengine.maasrset_akset_sk ImportError)clsrrrs e/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/embeddings/volcengine.pyvalidate_environmentz&VolcanoEmbeddings.validate_environment.s. 4     | 4     |  3 3 3 3 3 3 [1ABBF MM&. / / / MM&. / / /%F8     +    s ABBtext List[float]c:||gdS)Nr)embed_documents)selfr&s r$ embed_queryzVolcanoEmbeddings.embed_query^s##TF++A..texts List[str]List[List[float]]cffdtdtjD}g}|D]y}jjd|d} ddlm}j|}| d|dD[#|$r}td|d }~wwxYw|S) a_ Embeds a list of text documents using the AutoVOT algorithm. Args: texts (List[str]): A list of text documents to embed. Returns: List[List[float]]: A list of embeddings for each document in the input list. Each embedding is represented as a list of float values. c4g|]}||jzS)r).0ir*r-s r$ z5VolcanoEmbeddings.embed_documents..ls:    !a$/)) *   r,r)namer)rinput) MaasExceptioncg|] }|d S) embeddingr2)r3ress r$r5z5VolcanoEmbeddings.embed_documents..}sEEEC ,EEEr,datazembed by volcengine Error: N) rangelenrrrrr8r embeddingsextend ValueError) r*r-text_in_chunkslstchunkreqr8respes `` r$r)z!VolcanoEmbeddings.embed_documentsas     1c%jj$/::   # D DE!J#| C D999999{--c22 EEV EEEFFFF  D D D !Bq!B!BCCC D s ABB.B))B.)rrrr)r&rrr')r-r.rr/)__name__ __module__ __qualname____doc__r__annotations__rrrrrrr r%r+r)r2r,r$r r s33 $J$$$$I!%J$$$$I=D<<<<EFE E GJ2KKK ---X-^////r,r ) __future__rloggingtypingrrrrlangchain_core.embeddingsrlangchain_core.utilsr r pydanticr getLoggerrHloggerr r2r,r$rUs"""""",,,,,,,,,,,,000000????????  8 $ $sssss :sssssr,