[利用MongoDB存储聊天记录：一步步教程]

用户57716616653
178 阅读2分钟 
# 利用MongoDB存储聊天记录：一步步教程

## 引言

在现代应用中，处理和存储聊天记录是一个常见需求。MongoDB作为一种NoSQL数据库，以其灵活的JSON-like文档存储方式成为开发者的理想选择。本篇文章将介绍如何使用`MongoDBChatMessageHistory`类将聊天记录存储在MongoDB中，同时探讨一些常见问题和解决方案。

## 主要内容

### 安装和设置

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

```bash
pip install -U langchain-mongodb

您还可以配置LangSmith来实现更好的可观察性，但这不是必须的：

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

使用MongoDB存储聊天记录

要使用MongoDB存储聊天记录，您需要提供以下信息：

  • Session Id: 会话的唯一标识符，比如用户名、电子邮件、聊天ID等。
  • Connection string: 数据库连接字符串，将传递给MongoDB的create_engine函数。

您还可以自定义数据库和集合的名称。

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

将聊天记录与LCEL组合

可以将消息历史类与LCEL Runnables结合使用。

首先，确保安装langchain-openai，并设置OPENAI_API_KEY环境变量：

import os
from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
from langchain_core.runnables.history import RunnableWithMessageHistory
from langchain_openai import ChatOpenAI

assert os.environ["OPENAI_API_KEY"], "Set the OPENAI_API_KEY environment variable with your OpenAI API key."

prompt = ChatPromptTemplate.from_messages(
    [
        ("system", "You are a helpful assistant."),
        MessagesPlaceholder(variable_name="history"),
        ("human", "{question}"),
    ]
)

chain = prompt | ChatOpenAI()

chain_with_history = RunnableWithMessageHistory(
    chain,
    lambda session_id: MongoDBChatMessageHistory(
        session_id=session_id,
        connection_string="mongodb://mongo_user:password123@mongo:27017",  # 使用API代理服务提高访问稳定性
        database_name="my_db",
        collection_name="chat_histories",
    ),
    input_messages_key="question",
    history_messages_key="history",
)

config = {"configurable": {"session_id": "<SESSION_ID>"}}

chain_with_history.invoke({"question": "Hi! I'm bob"}, config=config)
chain_with_history.invoke({"question": "Whats my name"}, config=config)

常见问题和解决方案

  1. 连接问题：在某些地区，由于网络限制，连接MongoDB可能失败。可以考虑使用API代理服务来提高访问稳定性。
  2. 权限问题：确保您的MongoDB用户具有对指定数据库和集合的读写权限。

总结和进一步学习资源

通过本文，您应该能够利用MongoDB存储和管理聊天记录。对于进一步学习，您可以参考以下资源：

参考资料

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


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