a bgF@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 dddddZ dddddZddddd 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_sizerclientr)valuesreturncCst|dd|d<t|dd|d<zFddlm}||d|d}||d||d||d <Wntytd Yn0|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` r ZVOLC_ACCESSKEYr ZVOLC_SECRETKEYr) MaasServicerrrzMvolcengine package not found, please install it with `pip install volcengine`)rvolcengine.maasrZset_akZset_sk ImportError)clsrrrrw/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/volcengine.pyvalidate_environment.s*    z&VolcanoEmbeddings.validate_environmentz List[float])textrcCs||gdS)Nr)embed_documents)selfr rrr embed_query^szVolcanoEmbeddings.embed_queryz List[str]zList[List[float]])textsrc sfddtdtjD}g}|D]}jjd|d}z4ddlm}j|}| dd|dDWq*|y}zt d |WYd }~q*d }~00q*|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(}datazembed by volcengine Error: N) rangelenrrrrr+rZ embeddingsextend ValueError) r"r$Ztext_in_chunkslstchunkreqr+resperr'rr!as"   &z!VolcanoEmbeddings.embed_documents)__name__ __module__ __qualname____doc__r __annotations__r rrrrrr rr#r!rrrrr s        /r ) __future__rloggingtypingrrrrZlangchain_core.embeddingsrZlangchain_core.utilsrr Zpydanticr getLoggerr8loggerr rrrrs