a bg@sVddlmZmZmZddlmZddlmZmZddl m Z m Z Gddde eZ dS))AnyDictList) Embeddings)get_from_dict_or_envpre_init) BaseModel ConfigDictcseZdZUdZeed<eed<eed<eddZ deeed d fd d Z e e e d ddZ eeeeedddZeeedddZZS)NLPCloudEmbeddingsa NLP Cloud embedding models. To use, you should have the nlpcloud python package installed Example: .. code-block:: python from langchain_community.embeddings import NLPCloudEmbeddings embeddings = NLPCloudEmbeddings() model_namegpuclient)Zprotected_namespaces%paraphrase-multilingual-mpnet-base-v2FN)r r kwargsreturnc stjf||d|dS)N)r r )super__init__)selfr r r __class__ru/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/nlpcloud.pyrszNLPCloudEmbeddings.__init__)valuesrcCsVt|dd}z*ddl}|j|d||ddd|d <WntyPtd Yn0|S) z?Validate that api key and python package exists in environment.nlpcloud_api_keyZNLPCLOUD_API_KEYrNr r en)r langr zXCould not import nlpcloud python package. Please install it with `pip install nlpcloud`.)rnlpcloudZClient ImportError)clsrrrrrrvalidate_environment#s  z'NLPCloudEmbeddings.validate_environment)textsrcCs|j|dS)zEmbed a list of documents using NLP Cloud. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. embeddingsr r!)rr rrrembed_documents6s z"NLPCloudEmbeddings.embed_documents)textrcCs|j|gddS)zEmbed a query using NLP Cloud. Args: text: The text to embed. Returns: Embeddings for the text. r!rr")rr$rrr embed_queryBs zNLPCloudEmbeddings.embed_query)rF)__name__ __module__ __qualname____doc__str__annotations__boolrr Z model_configrrrrrfloatr#r% __classcell__rrrrr s"    r N) typingrrrZlangchain_core.embeddingsrZlangchain_core.utilsrrZpydanticrr r rrrrs