[深入探索Astra DB：轻松存储和管理聊天历史]

最新推荐文章于 2024-10-08 17:30:34 发布

jaioyfpo

最新推荐文章于 2024-10-08 17:30:34 发布

阅读量66

点赞数 2

文章标签：数据库 python

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

版权

Astra DB：轻松存储和管理聊天历史

引言

在现代应用开发中，聊天功能是一个常见的需求，而管理聊天记录则是其中的重要组成部分。本文将介绍如何使用Astra DB，一个基于Cassandra的无服务器数据库，通过简单易用的JSON API存储聊天记录。Astra DB支持向量功能，提供高效和灵活的数据存储方案。

主要内容

1. Astra DB简介

Astra DB是DataStax推出的一种无服务器数据库服务，构建在Apache Cassandra之上。它提供了JSON API，使得开发者可以方便地进行数据操作，而无需复杂的配置。Astra DB还支持向量处理能力，为机器学习等应用提供了强大的支持。

2. 设置Astra DB

在开始使用Astra DB之前，我们需要在DataStax的Astra控制台中创建一个数据库，并获取连接密钥。

API Endpoint: 例如，https://<identifier>-us-east1.apps.astra.datastax.com
Token: 例如，AstraCS:your_token_value

接下来，安装必要的Python库：

%pip install --upgrade --quiet "astrapy>=0.7.1 langchain-community"

3. 配置数据库连接

使用Python中的getpass库来安全地输入API键。

import getpass

ASTRA_DB_API_ENDPOINT = input("ASTRA_DB_API_ENDPOINT = ")
ASTRA_DB_APPLICATION_TOKEN = getpass.getpass("ASTRA_DB_APPLICATION_TOKEN = ")

4. 建立数据库会话

根据使用环境（本地或云端）创建相应的数据库连接会话对象。以下是使用langchain_community库与Astra DB交互的示例：

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,
)

5. 添加和检索消息

通过以下代码添加用户和AI消息，并检索消息历史：

message_history.add_user_message("hi!")
message_history.add_ai_message("whats up?")

print(message_history.messages)
# 输出: [HumanMessage(content='hi!'), AIMessage(content='whats up?')]