探索Astra DB：利用其强大功能存储聊天消息历史

最新推荐文章于 2024-09-30 20:09:33 发布

jaioyfpo

最新推荐文章于 2024-09-30 20:09:33 发布

阅读量333

点赞数 4

文章标签：数据库 oracle python

本文链接：https://blog.csdn.net/jaioyfpo/article/details/142428041

版权

引言

随着人工智能聊天机器人的普及，存储聊天消息历史成为一项关键任务。DataStax的Astra DB作为一个无服务器、支持向量的数据库，提供了一个简单易用的JSON API，可以简化这一过程。在这篇文章中，我们将学习如何利用Astra DB来存储聊天消息历史。

主要内容

Astra DB概述

Astra DB是基于Cassandra构建的无服务器数据库，为开发者提供了一套强大的工具集。其向量功能和JSON API让数据存储和管理变得得心应手。

设置Astra DB

在开始使用Astra DB之前，你需要一个运行中的Astra DB实例。可以通过Astra仪表板获取连接密钥，其中API Endpoint示例为https://01234567-89ab-cdef-0123-456789abcdef-us-east1.apps.astra.datastax.com，Token示例为AstraCS:6gBhNmsk135....。

安装必要的Python包：

%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存储聊天消息

通过Astra的API接口，我们可以轻松添加和检索聊天消息。以下是使用AstraDBChatMessageHistory的示例：

from langchain_community.chat_message_histories import AstraDBChatMessageHistory

# 创建消息历史对象
message_history = AstraDBChatMessageHistory(
    session_id="test-session",
    api_endpoint=ASTRA_DB_API_ENDPOINT,  # 使用API代理服务提高访问稳定性
    token=ASTRA_DB_APPLICATION_TOKEN,
)

# 添加用户消息
message_history.add_user_message("hi!")

# 添加AI消息
message_history.add_ai_message("whats up?")

# 检索消息历史
print(message_history.messages)
# 输出: [HumanMessage(content='hi!'), AIMessage(content='whats up?')]