[掌握MongoDB：使用LangChain存储聊天历史的实用指南]

tt_jishu

于 2024-09-27 17:45:42 发布

阅读量165

点赞数 2

文章标签： mongodb langchain 数据库

本文链接：https://blog.csdn.net/tt_jishu/article/details/142289060

版权

# 掌握MongoDB：使用LangChain存储聊天历史的实用指南

## 引言

MongoDB 是一个广泛使用的NoSQL数据库，以其灵活的JSON-like文档结构而闻名。在现代应用中，尤其是聊天机器人和会话应用的开发中，如何有效地存储聊天记录是一个常见需求。本篇文章将介绍如何利用LangChain的MongoDBChatMessageHistory类，在MongoDB中存储聊天历史。

## 主要内容

### 安装和设置

在着手开发之前，首先需要安装 `langchain-mongodb` 包：

```bash
pip install -U --quiet langchain-mongodb

为了获得更好的可观测性，建议设置LangSmith，虽然不是必须的：

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

使用方法

利用MongoDB存储聊天历史需要两个核心要素：

Session Id：会话的唯一标识符，如用户名、电子邮件、聊天ID等。
Connection String：指定数据库连接的字符串。

如果需要自定义数据库和集合名称，可以设置额外的参数：

database_name：使用的数据库名称
collection_name：数据库中集合的名称

以下是如何创建一个 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")

链式调用

可以将 MongoDBChatMessageHistory 与 LangChain 的其他组件结合使用，创建复杂的会话逻辑。以下是与 OpenAI 的集成示例：

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

import os

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

# 配置会话ID
config = {"configurable": {"session_id": "<SESSION_ID>"}}

# 执行调用
chain_with_history.invoke({"question": "Hi! I'm bob"}, config=config)