【LangChain4J】聊天数据持久化——Redis

git地址

第一步集成Redis

推荐本地安装Redis使用docker

pom文件添加
<dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>配置文件添加
spring:data:redis:# 连接地址host: 127.0.0.1# 端口port: 6379# 数据库database: 0password: 密码# 连接超时connect-timeout: 5s# 读超时timeout: 5s

第二步集成Redis组件序列化配置

@Configuration
public class RedisConfig {public RedisTemplate<String, Object> redisTemplate(RedisConnectionFactory factory){RedisTemplate<String, Object> template = new RedisTemplate<>();template.setConnectionFactory(factory);template.setKeySerializer(new StringRedisSerializer());template.setValueSerializer(new GenericJackson2JsonRedisSerializer());template.setHashKeySerializer(new StringRedisSerializer());template.setHashValueSerializer(new GenericJackson2JsonRedisSerializer());template.afterPropertiesSet();return template;}
}

第三步实现ChatMemoryStore接口

@Component
public class RedisChatMemoryStore implements ChatMemoryStore {@Autowiredprivate RedisTemplate<String, String> redisTemplate;private static final String REDIS_KEY = "chat_persistence:";@Overridepublic List<ChatMessage> getMessages(Object memoryId) {String value = redisTemplate.opsForValue().get(REDIS_KEY + memoryId);return ChatMessageDeserializer.messagesFromJson(value);}@Overridepublic void updateMessages(Object memoryId, List<ChatMessage> list) {redisTemplate.opsForValue().set(REDIS_KEY + memoryId, ChatMessageSerializer.messagesToJson(list));}@Overridepublic void deleteMessages(Object memoryId) {redisTemplate.delete(REDIS_KEY + memoryId);}
}

第四步定义Assistant接口并且装配

public interface ChatPersistenceAssistant {String chatPersistenceAssistant(@MemoryId Long userId, @UserMessage String prompt);}@Beanpublic ChatPersistenceAssistant chatPersistenceAssistant(ChatModel chatModel,RedisChatMemoryStore redisChatMemoryStore){return AiServices.builder(ChatPersistenceAssistant.class).chatModel(chatModel).chatMemoryProvider(memoryId -> {return MessageWindowChatMemory.builder().id(memoryId).maxMessages(100).chatMemoryStore(redisChatMemoryStore).build();}).build();}

第五步实现Controller并查看写入Redis

@RestController
public class ChatPersistenceController {@Autowiredprivate ChatPersistenceAssistant chatPersistenceAssistant;@GetMapping("/chat/persistence")public String chatPersistence(){String result1 = chatPersistenceAssistant.chatPersistenceAssistant(1L, "我的名字叫安静的狮子");System.out.println(result1);String result2 = chatPersistenceAssistant.chatPersistenceAssistant(1L, "你叫什么名字");System.out.println(result2);String result3 = chatPersistenceAssistant.chatPersistenceAssistant(2L, "你的名字叫CPDD");System.out.println(result3);String result4 = chatPersistenceAssistant.chatPersistenceAssistant(2L, "你叫什么名字");System.out.println(result4);return "OK";}
}