[打造智能聊天记忆：使用MongoDB存储聊天记录的实用指南]

最新推荐文章于 2024-09-27 17:45:42 发布

mmlihaio

最新推荐文章于 2024-09-27 17:45:42 发布

阅读量500

点赞数 3

文章标签： mongodb 数据库 oracle python

本文链接：https://blog.csdn.net/mmlihaio/article/details/142428176

版权

引言

MongoDB是一种流行的NoSQL数据库，它使用类似JSON的文档结构，提供灵活的数据库模型。在当今以聊天机器人和智能助手为代表的应用场景中，存储和管理聊天记录变得非常重要。本篇文章将介绍如何利用MongoDBChatMessageHistory类，将聊天记录存储在MongoDB数据库中，实现智能聊天记忆。

主要内容

1. 环境设置

首先，我们需要安装langchain-mongodb包以集成MongoDB存储功能：

pip install -U --quiet langchain-mongodb

此外，设置LangSmith可以增强可观察性（可选）：

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

2. 使用`MongoDBChatMessageHistory`

要使用该类，我们需要提供两个基本信息：

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", # 使用API代理服务提高访问稳定性
    database_name="my_db",
    collection_name="chat_histories",
)

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

3. 与LCEL的集成

我们可以将消息记录类与LCEL的Runnables相结合，实现更复杂的链式操作。需要先安装OpenAI，并设置OPENAI_API_KEY环境变量：

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

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)