Sentence Transformers Embeddings

SentenceTransformers 嵌入是通过 HuggingFaceEmbeddings 集成来调用的。我们还为那些更熟悉直接使用该软件包的用户添加了 SentenceTransformerEmbeddings 的别名。

SentenceTransformers 是一个可以生成文本和图像嵌入的 Python 软件包，源自 Sentence-BERT。

pip install sentence_transformers > /dev/null

    
    [notice] A new release of pip is available: 23.0.1 -> 23.1.1
    [notice] To update, run: pip install --upgrade pip

from langchain.embeddings import HuggingFaceEmbeddings, SentenceTransformerEmbeddings

embeddings = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
# 等同于 SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")

text = "这是一个测试文档。"

query_result = embeddings.embed_query(text)

doc_result = embeddings.embed_documents([text, "这不是一个测试文档。"])