Astra DB
DataStax Astra DB 是一个基于
Apache Cassandra®
并可通过易于使用的 JSON API 方便地访问的无服务器 AI 就绪数据库。
本笔记本将介绍如何使用 Astra DB 来存储聊天消息历史记录。
设置
要运行此笔记本,您需要一个正在运行的 Astra DB。在您的 Astra 控制台获取连接密钥
- API 端点类似于
https://01234567-89ab-cdef-0123-456789abcdef-us-east1.apps.astra.datastax.com
; - 数据库令牌类似于
AstraCS:aBcD0123...
。
!pip install "langchain-astradb>=0.6,<0.7"
设置数据库连接参数和密钥
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 = ········
示例
from langchain_astradb 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("hello, how are you?")
API 参考:AstraDBChatMessageHistory
API 参考: AstraDBChatMessageHistory
message_history.messages
[HumanMessage(content='hi!', additional_kwargs={}, response_metadata={}),
AIMessage(content='hello, how are you?', additional_kwargs={}, response_metadata={})]