AwaDB
AwaDB 是一個 AI 原生資料庫，用於 LLM 應用程式使用的嵌入向量的搜尋和儲存。
您需要使用 pip install -qU langchain-community 安裝 langchain-community 才能使用此整合
此筆記本示範如何使用與 AwaDB 相關的功能。
%pip install --upgrade --quiet awadb
from langchain_community.document_loaders import TextLoader
from langchain_community.vectorstores import AwaDB
from langchain_text_splitters import CharacterTextSplitter
loader = TextLoader("../../how_to/state_of_the_union.txt")
documents = loader.load()
text_splitter = CharacterTextSplitter(chunk_size=100, chunk_overlap=0)
docs = text_splitter.split_documents(documents)
db = AwaDB.from_documents(docs)
query = "What did the president say about Ketanji Brown Jackson"
docs = db.similarity_search(query)
print(docs[0].page_content)
And I did that 4 days ago, when I nominated Circuit Court of Appeals Judge Ketanji Brown Jackson. One of our nation’s top legal minds, who will continue Justice Breyer’s legacy of excellence.
相似度搜尋與分數
傳回的距離分數介於 0-1 之間。 0 是不相似，1 是最相似
docs = db.similarity_search_with_score(query)
print(docs[0])
(Document(page_content='And I did that 4 days ago, when I nominated Circuit Court of Appeals Judge Ketanji Brown Jackson. One of our nation’s top legal minds, who will continue Justice Breyer’s legacy of excellence.', metadata={'source': '../../how_to/state_of_the_union.txt'}), 0.561813814013747)
還原之前建立並新增資料的表格
AwaDB 會自動持久保存新增的文件資料。
如果您可以還原之前建立並新增的表格，您可以如下操作
import awadb
awadb_client = awadb.Client()
ret = awadb_client.Load("langchain_awadb")
if ret:
print("awadb load table success")
else:
print("awadb load table failed")
awadb 載入表格成功