【SpringAI Alibaba】基于 Redis 实现连续对话与向量存储

目录

一、连续对话的实现

1.1 添加依赖

1.2 设置配置文件

1.3 添加Redis配置类

1.4 配置ChatClient实现连续对话

二、Redis向量存储与查询

2.1 安装 RediSearch​(Redis Stack)

2.2 添加依赖

2.3 配置 Redis 服务器信息

2.4 添加 RedisConfig

2.5 Redis 添加向量数据

2.6 避免数据重复添加

2.7 Redis 查询向量数据

一、连续对话的实现

因为大模型本身并不存储数据，因此想要实现连续对话，其实是把历史对话信息一次给大模型，这样大模型就可以根据历史对话实现“连续对话”的功能了。​

因此存储的历史数据可能很多，所以最好存储在非关系型数据库，我们本文以存储到 Redis 为例给大家演示连续对话的实现。

1.1 添加依赖

需要添加两个依赖：spring-ai-alibaba-starter-memory-redis 是 Redis 存储器，jedis 是 Redis 存储器需要操作 Redis 数据库的驱动：

<dependency>​
    <groupId>com.alibaba.cloud.ai</groupId>​
    <artifactId>spring-ai-alibaba-starter-memory-redis</artifactId>​
</dependency>​
​
<dependency>​
    <groupId>redis.clients</groupId>​
    <artifactId>jedis</artifactId>​
    <version>5.2.0</version>​
</dependency>

1.2 设置配置文件

设置配置文件，配置 Redis 连接信息：

spring:​
  ai:​
    memory:​
      redis:​
        host: localhost​
        port: 6379​
        timeout: 5000

1.3 添加Redis配置类

添加 Redis 配置类，注入 RedisChatMemoryRepository 对象，实现 Redis 自定义存储器注入：

import com.alibaba.cloud.ai.memory.redis.RedisChatMemoryRepository;​
import org.springframework.beans.factory.annotation.Value;​
import org.springframework.context.annotation.Bean;​
import org.springframework.context.annotation.Configuration;​
​
@Configuration​
public class RedisMemoryConfig {​
​
    @Value("${spring.ai.memory.redis.host}")​
    private String redisHost;​
    @Value("${spring.ai.memory.redis.port}")​
    private int redisPort;​
//    @Value("${spring.ai.memory.redis.password}")​
//    private String redisPassword;​
    @Value("${spring.ai.memory.redis.timeout}")​
    private int redisTimeout;​
​
    @Bean​
    public RedisChatMemoryRepository redisChatMemoryRepository() {​
        return RedisChatMemoryRepository.builder()​
                .host(redisHost)​
                .port(redisPort)​
                // 若没有设置密码则注释该项​
//           .password(redisPassword)​
                .timeout(redisTimeout)​
                .build();​
    }​
}