Skip to main content
Open In ColabOpen on GitHub

AwaDB

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

本笔记本解释了如何在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"])

这个页面有帮助吗?