AwaDB

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

本 Notebook 介绍如何在 LangChain 中使用 AwaEmbeddings。

# pip install awadb

导入库

from langchain_community.embeddings import AwaEmbeddings

API Reference: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"])

Related

• Embedding model conceptual guide
• Embedding model how-to guides