Skip to main content
Open In Colab在 GitHub 上打开

阿瓦数据库

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

此笔记本介绍如何使用AwaEmbeddings在 LangChain 中。

# 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"])