使用Google Cloud Memorystore for Redis存储聊天消息历史记录

最新推荐文章于 2025-04-27 15:36:21 发布

bhawfgrcbtwny

最新推荐文章于 2025-04-27 15:36:21 发布

阅读量686

点赞数 22

文章标签： redis bootstrap 数据库 python

本文链接：https://blog.csdn.net/bhawfgrcbtwny/article/details/142015698

版权

使用Google Cloud Memorystore for Redis存储聊天消息历史记录

引言

Google Cloud Memorystore for Redis 是一个全托管服务，利用Redis内存数据存储技术，为应用构建提供亚毫秒级数据访问的缓存。这篇文章旨在介绍如何在Google Cloud Memorystore for Redis中存储聊天消息历史记录，并借助Langchain的集成，实现AI驱动的体验。

主要内容

1. 准备工作

在运行此Notebook之前，您需要完成以下步骤：

创建一个Google Cloud项目
启用Memorystore for Redis API
创建一个Memorystore for Redis实例，确保版本不低于5.0

确认在此Notebook的运行时环境中可以访问数据库后，填写以下值并运行单元格。例如：

# 指定与实例关联的端点，供演示用途
ENDPOINT = "redis://127.0.0.1:6379"  # 使用API代理服务提高访问稳定性

2. 安装必要的库

Langchain集成存储在langchain-google-memorystore-redis包中，需要安装该包。

%pip install --upgrade --quiet langchain-google-memorystore-redis

3. 设置Google Cloud项目

设置您的Google Cloud项目，以便在此Notebook中利用Google Cloud资源。

PROJECT_ID = "my-project-id"  # 请填写您的Google Cloud项目ID

# 设置项目ID
!gcloud config set project {PROJECT_ID}

4. 认证

如果您使用Colab运行此Notebook，请使用以下单元格进行认证。

from google.colab import auth
auth.authenticate_user()

5. 基本用法

初始化MemorystoreChatMessageHistory类仅需提供两个参数：

redis_client - Memorystore Redis的一个实例。
session_id - 每个聊天消息历史记录对象必须有唯一的会话ID。

以下是具体实现步骤。

代码示例

import redis
from langchain_google_memorystore_redis import MemorystoreChatMessageHistory

# 连接到Memorystore for Redis实例
redis_client = redis.from_url("redis://127.0.0.1:6379")  # 使用API代理服务提高访问稳定性

# 创建聊天消息历史记录对象
message_history = MemorystoreChatMessageHistory(redis_client, session_id="session1")

# 打印存储的消息
print(message_history.messages)