Agent AI 超级智能体企业级实战（26年必学） - 基于Redis存储持久化（SpringAI1.1.2版本）

这里我存在Redis缓存里面，避免数据库压力

• 引入依赖

▼
plain
复制代码
<!--RedisChatMemory-->
		<dependency>
			<groupId>org.springframework.boot</groupId>
			<artifactId>spring-boot-starter-data-redis</artifactId>
		</dependency>

• 配置Redis

▼
plain
复制代码
spring:
  data:
    redis:
      host: localhost
      port: 6379
      password:
      database: 2

• 自定义RedisChatMemoryRepository

注意咱们的Message在读取和存入的序列化问题(这里使用了Hutool工具)

▼
plain
复制代码
@Repository
public class RedisChatMemoryRepository implements ChatMemoryRepository {

    private final StringRedisTemplate redisTemplate;
    private static final String KEY_PREFIX = "chat:memory:";
    private static final String CONVERSATIONS_KEY = "chat:memory:conversations";
    private static final long TTL_HOURS = 24 * 7; // 会话保留7天

    public RedisChatMemoryRepository(StringRedisTemplate redisTemplate) {
        this.redisTemplate = redisTemplate;
    }


    @Override
    public List<String> findConversationIds() {
        Set<String> members = redisTemplate.opsForSet().members(CONVERSATIONS_KEY);
        return members != null ? new ArrayList<>(members) : List.of();
    }

    @Override
    public List<Message> findByConversationId(String conversationId) {
        Assert.hasText(conversationId, "Conversation ID must not be empty");
        String key = KEY_PREFIX + conversationId;

        List<String> jsonMessages = redisTemplate.opsForList().range(key, 0, -1);
        if (jsonMessages == null || jsonMessages.isEmpty()) {
            return List.of();
        }

        List<Message> messages = new ArrayList<>();
        for (String json : jsonMessages) {
            Message message = deserializeMessage(json);
            if (message != null) {
                messages.add(message);
            }
        }
        return messages;
    }

    @Override
    public void saveAll(String conversationId, List<Message> messages) {
        Assert.hasText(conversationId, "Conversation ID must not be empty");
        String key = KEY_PREFIX + conversationId;
        // 删除旧记录
        redisTemplate.delete(key);
        if (messages != null && !messages.isEmpty()) {
            List<String> jsonMessages = new ArrayList<>();
            for (Message message : messages) {
                String json = serializeMessage(message);
                if (json != null) {
                    jsonMessages.add(json);
                }
            }
            // 批量存入Redis List
            redisTemplate.opsForList().rightPushAll(key, jsonMessages);
            // 设置过期时间
            redisTemplate.expire(key, TTL_HOURS, TimeUnit.HOURS);
            // 记录会话ID到集合
            redisTemplate.opsForSet().add(CONVERSATIONS_KEY, conversationId);
        }
    }

    @Override
    public void deleteByConversationId(String conversationId) {
        Assert.hasText(conversationId, "Conversation ID must not be empty");
        String key = KEY_PREFIX + conversationId;
        redisTemplate.delete(key);
        redisTemplate.opsForSet().remove(CONVERSATIONS_KEY, conversationId);
    }

    /**
     * 序列化消息为JSON（用Hutool）
     */
    private String serializeMessage(Message message) {
        JSONObject json = JSONUtil.createObj()
                .set("messageType", message.getMessageType().name())
                .set("text", message.getText());

        // 如果有其他属性可以继续添加
        if (message.getMetadata() != null && !message.getMetadata().isEmpty()) {
            json.set("metadata", message.getMetadata());
        }

        return json.toString();
    }

    /**
     * 反序列化JSON为Message（用Hutool）
     */
    private Message deserializeMessage(String json) {
        if (StrUtil.isBlank(json)) {
            return null;
        }

        JSONObject jsonObject = JSONUtil.parseObj(json);
        String messageType = jsonObject.getStr("messageType");
        String text = jsonObject.getStr("text");

        if (messageType == null || text == null) {
            return null;
        }

        // 根据消息类型创建具体的Message对象
        return switch (messageType) {
            case "USER" -> new UserMessage(text);
            case "ASSISTANT" -> new AssistantMessage(text);
            case "SYSTEM" -> new SystemMessage(text);
            default -> null;
        };
    }
}

• 重写我们的对话记忆

▼
plain
复制代码
/**
 * 初始化AI 客户端
 * @param dashScopeChatModel
 */
public TravelApp(DashScopeChatModel dashScopeChatModel, RedisChatMemoryRepository redisChatMemoryRepository) {
    //基于内存的对话记忆
    //现在不再存在内存，而进行持久化存储（Redis）
    ChatMemory chatMemory = MessageWindowChatMemory.builder()
            .chatMemoryRepository(redisChatMemoryRepository)
            .maxMessages(10)
            .build();
    chatClient = ChatClient.builder(dashScopeChatModel)
            .defaultSystem(SYSTEM_PROMPT)
            .defaultAdvisors(
                    MessageChatMemoryAdvisor.builder(chatMemory).build(),
                    //自定义日志拦截器
                    new MyLoggerAdvisor()
                    // 自定义推理增强Advisor，可以按需开启(需要烧token)
                    //new ReReadingAdvisor()
            )
            .build();
}