o ZhF@stddlmZddlZddlmZmZmZmZddlm Z ddl m Z m Z ddl mZeeZGdddee ZdS) ) annotationsN)AnyDictListOptional) Embeddings)get_from_dict_or_envpre_init) BaseModelc@seZdZUdZdZded< dZded< dZded< d Zded < d Z ded < d Z ded< dZ ded< ded< e d"ddZ d#ddZd$d d!ZdS)%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_sizerclientvaluesrreturncCst|dd|d<t|dd|d<z$ddlm}||d|d}||d||d||d <W|Sty?td w) 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` r ZVOLC_ACCESSKEYr ZVOLC_SECRETKEYr) MaasServicerrrzMvolcengine package not found, please install it with `pip install volcengine`)rvolcengine.maasrZset_akZset_sk ImportError)clsrrrr`/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/embeddings/volcengine.pyvalidate_environment.s,   z&VolcanoEmbeddings.validate_environmenttext List[float]cCs||gdS)Nr)embed_documents)selfr rrr embed_query^szVolcanoEmbeddings.embed_querytexts List[str]List[List[float]]c sfddtdtjD}g}|D]9}jjd|d}zddlm}j|}| dd|dDWq|yN}zt d |d }~ww|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. csg|] }||jqSr)r).0ir#r%rr lsz5VolcanoEmbeddings.embed_documents..r)namer)rinput) MaasExceptioncSsg|]}|dqS)Z embeddingr)r(resrrrr+}sdatazembed by volcengine Error: N) rangelenrrrrr.rZ embeddingsextend ValueError) r#r%Ztext_in_chunkslstchunkreqr.resperr*rr"as&   z!VolcanoEmbeddings.embed_documents)rrrr)r rrr!)r%r&rr')__name__ __module__ __qualname____doc__r __annotations__r rrrrrr rr$r"rrrrr s,          /r ) __future__rloggingtypingrrrrZlangchain_core.embeddingsrZlangchain_core.utilsrr Zpydanticr getLoggerr:loggerr rrrrs