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