Skip to main content
Open In ColabOpen on GitHub

AstraDB

DataStax Astra DB 是一个基于 Apache Cassandra® 构建的无服务器AI就绪型数据库，通过易于使用的JSON API便捷提供。

本笔记本介绍了如何使用 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

message_history.messages
[HumanMessage(content='hi!', additional_kwargs={}, response_metadata={}),
 AIMessage(content='hello, how are you?', additional_kwargs={}, response_metadata={})]