如何使用 MongoDBChatMessageHistory 存储聊天记录

bBADAS

关注

最新推荐文章于 2026-05-22 07:50:10 发布 · 573 阅读

原创 ·

本内容遵循 CC 4.0 BY-SA 版权协议

· GEO检测

LangChain入门(九)-使用Memory实现一个带记忆的对话机器人在第五章中我们使用的是通过自定义一个列表来存储对话的方式来保存历史的。当然，你也可以使用自带的 memory 对象来实现这一点。阅读详情

在现代应用中，保存用户和AI之间的聊天记录是至关重要的。MongoDB 提供了一种有效的方法来存储这些数据。本文将演示如何使用 MongoDBChatMessageHistory 类将聊天记录存储在 MongoDB 数据库中。

技术背景介绍

MongoDB 是一个跨平台的文档导向数据库，属于 NoSQL 数据库，使用类似 JSON 的文档。它的灵活性和可扩展性使其成为处理大型和复杂数据的理想选择。在本教程中，我们将重点介绍如何利用 MongoDBChatMessageHistory 类为聊天应用存储历史记录。

核心原理解析

MongoDBChatMessageHistory 是 langchain-mongodb 包的一部分，它允许我们将聊天记录保存在 MongoDB 数据库中。我们只需要提供会话ID和连接字符串，并可以自定义数据库名称和集合名称。

代码实现演示

首先，我们需要安装 langchain-mongodb 包：

pip install -U --quiet langchain-mongodb

接下来，使用 MongoDBChatMessageHistory 来存储聊天记录：

from langchain_mongodb.chat_message_histories import MongoDBChatMessageHistory

# 初始化数据库连接
chat_message_history = MongoDBChatMessageHistory(
    session_id="test_session",
    connection_string="mongodb://mongo_user:password123@mongo:27017",
    database_name="my_db",
    collection_name="chat_histories",
)

# 添加消息
chat_message_history.add_user_message("Hello")
chat_message_history.add_ai_message("Hi")

# 查看存储的消息
print(chat_message_history.messages)
# 输出: [HumanMessage(content='Hello'), AIMessage(content='Hi')]