U h@sRddlmZmZmZddlmZddlmZddlm Z m Z GdddeeZ dS))AnyDictList) Embeddings) BaseModel)get_from_dict_or_envpre_initcseZdZUdZeed<eed<eed<deeeddfd d Ze e e d d d Z e ee e e dddZee e dddZZS)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%paraphrase-multilingual-mpnet-base-v2FN)r r kwargsreturnc stjf||d|dS)N)r r )super__init__)selfr r r __class__K/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/embeddings/nlpcloud.pyrszNLPCloudEmbeddings.__init__)valuesrcCsXt|dd}z*ddl}|j|d||ddd|d <Wntk rRtd YnX|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 )rrrrrembed_documents4s 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_query@s zNLPCloudEmbeddings.embed_query)r F)__name__ __module__ __qualname____doc__str__annotations__boolrrrrrrfloatr"r$ __classcell__rrrrr s   r N) typingrrrZlangchain_core.embeddingsrZlangchain_core.pydantic_v1rZlangchain_core.utilsrrr rrrrs