[使用MongoDB存储聊天记录：从入门到精通]

jaioyfpo

于 2024-10-08 13:58:42 发布

阅读量130

点赞数 1

文章标签： mongodb 数据库 python

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

版权

# 使用MongoDB存储聊天记录：从入门到精通

## 引言

MongoDB是一种流行的NoSQL数据库，使用类似JSON的文档来存储数据。它特别适合灵活的、无模式的存储需求。在这篇文章中，我将向您展示如何使用`MongoDBChatMessageHistory`类将聊天消息历史记录存储到MongoDB中。

## 主要内容

### 安装与设置

首先，您需要安装`langchain-mongodb`库，这是集成MongoDB和LangChain所必需的：

```bash
pip install -U langchain-mongodb

另外，如果需要，可以设置LangSmith以便更好地监控：

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

使用指南

要使用MongoDB存储聊天记录，您需要提供两个主要的参数：

Session Id：比如用户名、邮箱、聊天ID等。
Connection String：用于数据库连接的字符串。

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

from langchain_mongodb.chat_message_histories import MongoDBChatMessageHistory

chat_message_history = MongoDBChatMessageHistory(
    session_id="test_session",
    connection_string="mongodb://mongo_user:password123@mongo:27017",  # 使用API代理服务提高访问稳定性
    database_name="my_db",
    collection_name="chat_histories",
)

chat_message_history.add_user_message("Hello")
chat_message_history.add_ai_message("Hi")

链接OpenAI消息处理

我们可以将消息历史记录与OpenAI服务结合使用。在开始之前，请设置OPENAI_API_KEY环境变量：

import os

assert os.environ["OPENAI_API_KEY"], "Set the OPENAI_API_KEY environment variable."

然后，您可以像这样定义和运行带有历史记录的链：

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

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",
        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": "What's my name"}, config=config)