Skip to main content

AwaEmbedding

这个笔记本解释了如何使用AwaEmbedding,在awadb中包含了如何在langchain中嵌入文本。

# pip install awadb

导入库

from langchain.embeddings import AwaEmbeddings
Embedding = AwaEmbeddings()

设置嵌入模型

用户可以使用Embedding.set_model()来指定嵌入模型。\ 此函数的输入是一个表示模型名称的字符串。\ 当前支持的模型列表可以在这里获取。

默认模型all-mpnet-base-v2,可以在不设置的情况下使用。

text = "our embedding test"

Embedding.set_model("all-mpnet-base-v2")
res_query = Embedding.embed_query("The test information")
res_document = Embedding.embed_documents(["test1", "another test"])