Skip to main content
Open In ColabOpen on GitHub

AwaDB

AwaDB 是一个面向大语言模型应用的 AI 原生数据库，用于嵌入向量的搜索与存储。

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

# pip install awadb

导入库​

from langchain_community.embeddings import AwaEmbeddings
API 参考:AwaEmbeddings
Embedding = AwaEmbeddings()

设置嵌入模型

用户可以使用 Embedding.set_model() 用以指定嵌入模型。
该函数的输入是一个字符串，表示模型的名称。
目前支持的模型列表可以获取here \ \

默认模型 为 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"])