o Zh@sVddlmZmZmZddlmZddlmZmZddl m Z m Z Gddde eZ dS))AnyDictList) Embeddings)get_from_dict_or_envpre_init) BaseModel ConfigDictc seZdZUdZeed<eed<eed<eddZ  ddeded ed d ffd d Z e de d e fddZ deed eeefddZded eefddZZS)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-v2FkwargsreturnNc stjd||d|dS)N)r r r)super__init__)selfr r r __class__r^/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/embeddings/nlpcloud.pyrszNLPCloudEmbeddings.__init__valuescCsPt|dd}zddl}|j|d||ddd|d <W|Sty'td w) 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_environmenttextscCs|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_documentstextcCs|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