跳至主要内容

AwaDB

AwaDB 是一个用于存储和搜索 LLM 应用程序使用的嵌入向量的 AI 原生数据库。

此笔记本解释了如何在 LangChain 中使用 AwaEmbeddings

# pip install awadb

导入库

from langchain_community.embeddings import AwaEmbeddings
API 参考: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"])

此页面是否有帮助?


您还可以留下详细的反馈 在 GitHub 上.