7|hFddlmZddlZddlmZmZmZmZddlm Z ddl m Z m Z ddl mZejeZGddee Zy) ) 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 ddZddZy)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_sizerclientct|dd|d<t|dd|d< ddlm}||d|d}|j|d|j |d||d <|S#t $r t d wxYw) 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)clsvaluesrrs h/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/embeddings/volcengine.pyvalidate_environmentz&VolcanoEmbeddings.validate_environment.s. 4    | 4    |  3 1ABF MM&. / MM&. /%F8    +  s AA&&A;c,|j|gdS)Nr)embed_documents)selftexts r# embed_queryzVolcanoEmbeddings.embed_query^s##TF+A..ctdt||jDcgc]}||||jz}}g}|D]e}|j|jd|d} ddlm}|jj|}|j|dD cgc]} | d c} g|Scc}wcc} w#$r} td| d} ~ wwxYw) 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. r)namer)rinput) MaasExceptiondata embeddingzembed by volcengine Error: N) rangelenrrrrr.r embeddingsextend ValueError) r'textsitext_in_chunkslstchunkreqr.respreses r#r&z!VolcanoEmbeddings.embed_documentsas1c%j$//:  !a$//) *  # DE!JJ#|| C D9{{--c2 V EC ,EF D ) "F  D #>qc!BCC Ds/B,"3B6 B1 !B61B66C;C  C)r"rreturnr)r(rr?z List[float])r6z List[str]r?zList[List[float]])__name__ __module__ __qualname____doc__r__annotations__rrrrrrr r$r)r&r*r#r r s3 $J $I!%J $I=D#<EFCE E3 GSJ2 K --^/r*r ) __future__rloggingtypingrrrrlangchain_core.embeddingsrlangchain_core.utilsr r pydanticr getLoggerr@loggerr rEr*r#rNs<",,0?   8 $s :sr*