探索Astra DB：如何使用Astra DB存储聊天历史记录

最新推荐文章于 2024-09-30 17:19:22 发布

akhfuiigabv

最新推荐文章于 2024-09-30 17:19:22 发布

阅读量304

点赞数 5

文章标签：数据库

本文链接：https://blog.csdn.net/akhfuiigabv/article/details/142288715

版权

引言

随着AI和数据库技术的发展，Astra DB以其无服务器的架构和便捷的JSON API正在获得越来越多的关注。本文将探讨如何使用Astra DB存储聊天历史记录，并提供一个实用的代码示例来帮助你开始。

主要内容

什么是Astra DB？

Astra DB是基于Apache 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 = ")

创建数据库会话对象

根据你的数据库位置（本地或云端），创建相应的数据库连接“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,
)

# 添加用户消息和AI消息
message_history.add_user_message("hi!")
message_history.add_ai_message("whats up?")

代码示例

以下是完整的代码实现：

import getpass
from langchain_community.chat_message_histories import AstraDBChatMessageHistory

ASTRA_DB_API_ENDPOINT = input("ASTRA_DB_API_ENDPOINT = ")  # 使用API代理服务提高访问稳定性
ASTRA_DB_APPLICATION_TOKEN = getpass.getpass("ASTRA_DB_APPLICATION_TOKEN = ")

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?")

# 查看历史记录
print(message_history.messages)