redis相關準備

添加依賴

我利用redission來實現

<dependency><groupId>org.redisson</groupId><artifactId>redisson</artifactId><version>3.37.0</version>
</dependency>

添加配置文件

spring:redis:database: 5host: 127.0.0.1port: 6379password: 1234

創建RedisProperties文件

@Data
@Component
@ConfigurationProperties(prefix = "spring.redis")
public class RedisProperties {private String host;private int port;private String password;private int database;
}

創建redissionConfig配置文件

@Configuration
@RequiredArgsConstructor
public class RedisConfig {private final RedisProperties redisProperties;@Beanpublic RedissonClient redissonClient(){Config config = new Config();config.useSingleServer().setAddress("redis://" +  redisProperties.getHost() + ":" + redisProperties.getPort()).setDatabase(redisProperties.getDatabase()).setPassword(redisProperties.getPassword());config.setCodec(new StringCodec(StandardCharsets.UTF_8));return Redisson.create(config);}
}

創建RedissionUtil工具類

package com.sde.util;import org.redisson.api.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.jackson.JacksonProperties;
import org.springframework.stereotype.Component;
import java.time.Duration;
import java.time.Instant;
import java.util.Arrays;
import java.util.Set;
import java.util.concurrent.TimeUnit;
import java.util.stream.StreamSupport;@Component
public class RedissonUtil {private static RedissonClient redissonClient;@Autowiredpublic void setRedissonClient(RedissonClient client) {redissonClient = client;}/*** 設置數據的過期時間* @param key 要設置的鍵* @param seconds 過期時間（秒）*/public static boolean expire(String key, long seconds) {return redissonClient.getKeys().expire(key, seconds, TimeUnit.SECONDS);}/*** 設置數據的過期時間* @param key 要設置的鍵* @param duration 時長* @param unit 時間單位* @return*/public static boolean expire(String key, long duration, TimeUnit unit) {return redissonClient.getKeys().expire(key, duration, unit);}/*** 將字符串添加到 Redis List 的尾部** @param key Redis 鍵* @param value 要添加的值*/public static void addToList(String key, String value) {RList<String> list = redissonClient.getList(key);list.add(value);}/*** 獲取列表* @param key* @return*/public static RList<String> getList(String key) {return redissonClient.getList(key);}/*** 獲取 RList 對象，用于后續操作如 trim、remove 等*/public static RList<String> getRList(String key) {return redissonClient.getList(key);}/*** 刪除數據* @param key 要刪除的鍵* @return*/public static boolean remove(final String key) {return redissonClient.getKeys().delete(key) >= 0;}/*** 檢查鍵是否存在* @param key 鍵* @return 是否存在*/public static boolean exists(String key) {RBucket<String> bucket = redissonClient.getBucket(key);return bucket.isExists();}}

SpringAI相關準備

不用阿里云的maven倉庫，用默認的maven中央倉庫。

版本控制

    <properties><java.version>21</java.version><spring-ai.version>1.0.0-SNAPSHOT</spring-ai.version></properties>

    <dependencyManagement><dependencies><dependency><groupId>org.springframework.ai</groupId><artifactId>spring-ai-bom</artifactId><version>${spring-ai.version}</version><type>pom</type><scope>import</scope></dependency></dependencies></dependencyManagement>

添加依賴

        <dependency><groupId>org.springframework.ai</groupId><artifactId>spring-ai-starter-model-openai</artifactId></dependency>

因為我對接的阿里云百煉平臺

添加配置

spring:ai:openai:base-url: https://dashscope.aliyuncs.com/compatible-mode/api-key: ${ALIYUN_AK}chat:options:model: qwen-max

我的api-key 存儲在我環境變量中

編寫代碼

創建保存到redis中的實體

@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class StoredMessage implements Serializable {@JSONField(ordinal = 0)private String role;@JSONField(ordinal = 1)private String content;public static StoredMessage from(Message message) {return new StoredMessage(message.getMessageType().name(), message.getText());}public Message toMessage() {return switch (role) {case "USER" -> new UserMessage(content);case "ASSISTANT" -> new AssistantMessage(content);default -> new SystemMessage(content);};}
}

創建RedisChatMemory

@Component
public class RedisChatMemory implements ChatMemory {// redis key 的前綴private static final String CHAT_MEMORY_KEY = "chat:conversationid:";private static final long TTL_MILLIS = 2 * 24 * 60 * 60 * 1000; // 2天private final int maxMessage;public RedisChatMemory() {this(10); // 默認保留 10 條消息}public RedisChatMemory(int maxMessages) {this.maxMessage = maxMessages;}@Overridepublic void add(String conversationId, Message message) {String key = getKey(conversationId);StoredMessage storedMessage = StoredMessage.from(message);String json = toJson(storedMessage);RedissonUtil.addToList(key, json);trimMessages(key);//刷新過期時間為2天RedissonUtil.expire(key, TTL_MILLIS, TimeUnit.MILLISECONDS);}@Overridepublic void add(String conversationId, List<Message> messages) {messages.forEach(message -> add(conversationId, message));}@Overridepublic List<Message> get(String conversationId) {String key = getKey(conversationId);//檢查key是否存在if (!RedissonUtil.exists(key)) {return Collections.emptyList();}//獲取所有 json字符串格式的消息List<String> messages = RedissonUtil.getList(key);//轉換成 Message 列表return messages.stream().map(this::fromJson).filter(Objects::nonNull).map(StoredMessage::toMessage).toList();}@Overridepublic void clear(String conversationId) {String key = getKey(conversationId);RedissonUtil.remove(key);}/*** 構建redis的key* @param conversationId* @return*/private String getKey(String conversationId){return CHAT_MEMORY_KEY + conversationId;}/*** 將消息轉換為 JSON 字符串*/private String toJson(StoredMessage msg) {SerializeConfig config = new SerializeConfig();return JSON.toJSONString(msg,config, SerializerFeature.SortField);}/*** 將 JSON 字符串轉換為 StoredMessage 對象*/private StoredMessage fromJson(String json) {return JSON.parseObject(json, StoredMessage.class);}/*** 控制消息數量，保留最近 maxMessages 條*/private void trimMessages(String key) {RList<String> list = RedissonUtil.getRList(key);if (list.size() > maxMessage) {int toRemove = list.size() - maxMessage;for (int i = 0; i < toRemove; i++) {list.remove(0);}}}}

創建Controller

@Slf4j
@RestController
@RequestMapping("/memory")
public class ChatMemoryController {private final ChatClient chatClient;public ChatMemoryController(ChatModel qwen) {this.chatClient = ChatClient.builder(qwen).build();}@Autowiredprivate RedisChatMemory redisChatMemory;@GetMapping("/redis")public Flux<String> redis(String question, String conversationId){MessageChatMemoryAdvisor advisor = MessageChatMemoryAdvisor.builder(redisChatMemory).conversationId(conversationId).order(5).build();return chatClient.prompt().system("你是我的學習助手名字叫小菜包").user(question).advisors(advisor).stream().content();}
}

測試