跳到主要內容
Open In ColabOpen on GitHub

AwaDB

AwaDB 是一個 AI 原生資料庫,用於搜尋和儲存 LLM 應用程式使用的嵌入向量。

本筆記本說明如何在 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"])

此頁面是否對您有幫助?