如何使用AWS DynamoDB存储聊天消息历史记录

最新推荐文章于 2025-08-05 09:46:16 发布

原创最新推荐文章于 2025-08-05 09:46:16 发布 · 358 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#aws #大数据 #数据库 #python

引言

在快速发展的数字时代，数据的存储和处理变得越来越重要。Amazon AWS的DynamoDB是一个完全托管的NoSQL数据库服务，提供快速和可预见的性能，并具有无缝的可扩展性。在这篇文章中，我们将探讨如何利用DynamoDB来存储聊天消息历史记录，使用DynamoDBChatMessageHistory类进行操作。

主要内容

1. 环境设置

开始之前，请确保已正确配置AWS CLI，并安装了langchain-community和boto3包。可以通过以下命令安装必需的包：

pip install -U langchain-community boto3

另外，设置LangSmith可提高可观察性，但不是必须的：

# os.environ["LANGCHAIN_TRACING_V2"] = "true"
# os.environ["LANGCHAIN_API_KEY"] = getpass.getpass()

2. 创建DynamoDB表

在开始存储消息之前，我们需要创建一个DynamoDB表来保存数据：

import boto3

# 获取服务资源
dynamodb = boto3.resource("dynamodb")

# 创建DynamoDB表
table = dynamodb.create_table(
    TableName="SessionTable",
    KeySchema=[{"AttributeName": "SessionId", "KeyType": "HASH"}],
    AttributeDefinitions=[{"AttributeName": "SessionId", "AttributeType": "S"}],
    BillingMode="PAY_PER_REQUEST",
)

# 等待表存在
table.meta.client.get_waiter("table_exists").wait(TableName="SessionTable")

# 输出表的数据
print(table.item_count)

3. 使用`DynamoDBChatMessageHistory`类记录消息

一旦表创建完成，我们就可以使用DynamoDBChatMessageHistory类来记录和检索聊天消息：

from langchain_community.chat_message_histories import DynamoDBChatMessageHistory

# 初始化消息历史记录
history = DynamoDBChatMessageHistory(table_name="SessionTable", session_id="0")

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

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

# 检索消息历史记录
print(history.messages)

4. 自定义端点URL

在某些情况下，指定AWS端点的URL是有用的，例如在本地运行时连接到Localstack。可以通过endpoint_url参数指定URL：

history = DynamoDBChatMessageHistory(
    table_name="SessionTable",
    session_id="0",
    endpoint_url="http://localhost.localstack.cloud:4566",  # 使用API代理服务提高访问稳定性
)

5. 使用复合键

有时，你可能需要修改键结构来适应现有的DynamoDB表设计，比如包含排序键：

composite_table = dynamodb.create_table(
    TableName="CompositeTable",
    KeySchema=[
        {"AttributeName": "PK", "KeyType": "HASH"},
        {"AttributeName": "SK", "KeyType": "RANGE"},
    ],
    AttributeDefinitions=[
        {"AttributeName": "PK", "AttributeType": "S"},
        {"AttributeName": "SK", "AttributeType": "S"},
    ],
    BillingMode="PAY_PER_REQUEST",
)

# 等待表存在
composite_table.meta.client.get_waiter("table_exists").wait(TableName="CompositeTable")

my_key = {
    "PK": "session_id::0",
    "SK": "langchain_history",
}

composite_key_history = DynamoDBChatMessageHistory(
    table_name="CompositeTable",
    session_id="0",
    endpoint_url="http://localhost.localstack.cloud:4566",  # 使用API代理服务提高访问稳定性
    key=my_key,
)

composite_key_history.add_user_message("hello, composite dynamodb table!")
print(composite_key_history.messages)