Astra DB
DataStax Astra DB 是一個以 Cassandra 為基礎建構的無伺服器、具備向量功能的資料庫,並透過易於使用的 JSON API 方便地提供使用。
本筆記本說明如何使用 Astra DB 儲存聊天訊息歷史記錄。
設定
若要執行此筆記本,您需要執行中的 Astra DB。在您的 Astra 儀表板上取得連線密碼
- API 端點看起來像
https://01234567-89ab-cdef-0123-456789abcdef-us-east1.apps.astra.datastax.com
; - Token 看起來像
AstraCS:6gBhNmsk135...
。
%pip install --upgrade --quiet "astrapy>=0.7.1 langchain-community"
設定資料庫連線參數和密碼
import getpass
ASTRA_DB_API_ENDPOINT = input("ASTRA_DB_API_ENDPOINT = ")
ASTRA_DB_APPLICATION_TOKEN = getpass.getpass("ASTRA_DB_APPLICATION_TOKEN = ")
ASTRA_DB_API_ENDPOINT = https://01234567-89ab-cdef-0123-456789abcdef-us-east1.apps.astra.datastax.com
ASTRA_DB_APPLICATION_TOKEN = ········
根據是本機或雲端 Astra DB,建立對應的資料庫連線「Session」物件。
範例
from langchain_community.chat_message_histories import AstraDBChatMessageHistory
message_history = AstraDBChatMessageHistory(
session_id="test-session",
api_endpoint=ASTRA_DB_API_ENDPOINT,
token=ASTRA_DB_APPLICATION_TOKEN,
)
message_history.add_user_message("hi!")
message_history.add_ai_message("whats up?")
API 參考:AstraDBChatMessageHistory
message_history.messages
[HumanMessage(content='hi!'), AIMessage(content='whats up?')]