langchain_community.embeddings.nlpcloud — 🦜🔗 LangChain 0.2.0

from typing import Any, Dict, List

from langchain_core.embeddings import Embeddings
from langchain_core.pydantic_v1 import BaseModel, root_validator
from langchain_core.utils import get_from_dict_or_env


[docs]class NLPCloudEmbeddings(BaseModel, Embeddings):
    """NLP Cloud嵌入模型。

    要使用，应该已安装nlpcloud python包

    示例:
        .. code-block:: python

            from langchain_community.embeddings import NLPCloudEmbeddings

            embeddings = NLPCloudEmbeddings()"""

    model_name: str  # Define model_name as a class attribute
    gpu: bool  # Define gpu as a class attribute
    client: Any  #: :meta private:

    def __init__(
        self,
        model_name: str = "paraphrase-multilingual-mpnet-base-v2",
        gpu: bool = False,
        **kwargs: Any,
    ) -> None:
        super().__init__(model_name=model_name, gpu=gpu, **kwargs)

    @root_validator()
    def validate_environment(cls, values: Dict) -> Dict:
        """验证环境中是否存在API密钥和Python包。"""
        nlpcloud_api_key = get_from_dict_or_env(
            values, "nlpcloud_api_key", "NLPCLOUD_API_KEY"
        )
        try:
            import nlpcloud

            values["client"] = nlpcloud.Client(
                values["model_name"], nlpcloud_api_key, gpu=values["gpu"], lang="en"
            )
        except ImportError:
            raise ImportError(
                "Could not import nlpcloud python package. "
                "Please install it with `pip install nlpcloud`."
            )
        return values

[docs]    def embed_documents(self, texts: List[str]) -> List[List[float]]:
        """使用NLP Cloud嵌入文档列表。

参数：
    texts: 要嵌入的文本列表。

返回：
    每个文本的嵌入列表。
"""

        return self.client.embeddings(texts)["embeddings"]

[docs]    def embed_query(self, text: str) -> List[float]:
        """使用NLP Cloud嵌入一个查询。

参数：
    text: 要嵌入的文本。

返回：
    文本的嵌入。
"""
        return self.client.embeddings([text])["embeddings"][0]
Source code for langchain_community.embeddings.nlpcloud
