跳到主要內容
Open In ColabOpen on GitHub

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?")
message_history.messages
[HumanMessage(content='hi!'), AIMessage(content='whats up?')]

此頁面是否對您有幫助?