o AgF@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 VOLC_ACCESSKEYr VOLC_SECRETKEYr) MaasServicerrrzMvolcengine package not found, please install it with `pip install volcengine`)rvolcengine.maasrset_akset_sk ImportError)clsrrrr!p/var/www/html/development/chatbot/venv/lib/python3.10/site-packages/langchain_community/embeddings/volcengine.pyvalidate_environment.s,   z&VolcanoEmbeddings.validate_environmenttext List[float]cCs||gdS)Nr)embed_documents)selfr$r!r!r" 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)r!r" lsz5VolcanoEmbeddings.embed_documents..r)namer)rinput) MaasExceptioncSsg|]}|dqS) embeddingr!)r,resr!r!r"r/}sdatazembed by volcengine Error: N) rangelenrrrrr2r embeddingsextend ValueError) r'r)text_in_chunkslstchunkreqr2resper!r.r"r&as&   z!VolcanoEmbeddings.embed_documents)rrrr)r$rrr%)r)r*rr+)__name__ __module__ __qualname____doc__r __annotations__r rrrrrr r#r(r&r!r!r!r"r s,          /r ) __future__rloggingtypingrrrrlangchain_core.embeddingsrlangchain_core.utilsrr pydanticr getLoggerrAloggerr r!r!r!r"s