重构: 简化ChatMemoryStore、ChatService, SysMessageService之间的职责。

workwind · workwind · commit cfd9445baf13 · 2025-06-19T22:59:09.000+08:00
diff --git a/src/main/java/com/xiaozhi/communication/common/ChatSession.java b/src/main/java/com/xiaozhi/communication/common/ChatSession.java
@@ -7,8 +7,6 @@
 import com.xiaozhi.entity.SysRole;
 import com.xiaozhi.enums.ListenMode;
 import lombok.Data;
-import org.springframework.ai.chat.memory.ChatMemory;
-import org.springframework.ai.chat.messages.Message;
 import org.springframework.ai.tool.ToolCallback;
 import reactor.core.publisher.Sinks;
 
@@ -69,10 +67,7 @@ public abstract class ChatSession {
      * 会话的最后有效活动时间
      */
     protected Instant lastActivityTime;
-    /**
-     * spring ai 聊天记忆
-     */
-    protected ChatMemory chatMemory;
+
     /**
      * 会话属性存储
      */
@@ -129,17 +124,8 @@ public List<ToolCallback> getToolCallbacks() {
         return toolsSessionHolder.getAllFunction();
     }
 
-    public void clearMemory() {
-        chatMemory.clear(sessionId);
-    }
 
-    public List<Message> getHistoryMessages() {
-        return chatMemory.get(sessionId);
-    }
 
-    public void addHistoryMessage(Message message){
-        chatMemory.add(sessionId, message);
-    }
     /**
      * 会话连接是否打开中
      * @return
diff --git a/src/main/java/com/xiaozhi/communication/common/MessageHandler.java b/src/main/java/com/xiaozhi/communication/common/MessageHandler.java
@@ -177,6 +177,8 @@ public void afterConnectionClosed(String sessionId) {
         audioService.cleanupSession(sessionId);
         // 清理对话
         dialogueService.cleanupSession(sessionId);
+        // 清理ChatService缓存的对话历史。
+        chatService.clearMessageCache(device.getDeviceId());
     }
 
     /**
diff --git a/src/main/java/com/xiaozhi/dialogue/llm/ChatService.java b/src/main/java/com/xiaozhi/dialogue/llm/ChatService.java
@@ -67,8 +67,7 @@ public class ChatService {
     // 新句子判断的字符阈值
     private static final int NEW_SENTENCE_TOKEN_THRESHOLD = 8;
 
-    // 历史记录默认限制数量
-    private static final int DEFAULT_HISTORY_LIMIT = 10;
+
 
     @Resource(name = "messageWindowChatMemory")
     private ChatMemoryStore chatMemoryStore;
@@ -92,52 +91,27 @@ public String chat(ChatSession session, String message, boolean useFunctionCall)
             // 获取ChatModel
             ChatModel chatModel = chatModelFactory.takeChatModel(session);
 
-            if (session.getChatMemory() == null) {// 如果记忆没初始化，则初始化一下
-                initializeHistory(session);
-            }
-
-            // 获取格式化的历史记录（包含当前用户消息）
-            List<Message> historyMessages = session.getHistoryMessages();
-
             ChatOptions chatOptions = ToolCallingChatOptions.builder()
                     .toolCallbacks(useFunctionCall ? session.getToolCallbacks() : new ArrayList<>())
                     .toolContext(TOOL_CONTEXT_SESSION_KEY, session)
                     .build();
-            UserMessage userMessage = new UserMessage(message);
-            Prompt prompt = Prompt.builder().messages(historyMessages).messages(userMessage).chatOptions(chatOptions)
-                    .build();
+
+            UserMessage userMessage = new UserMessage( message);
+            Prompt prompt = new Prompt(chatMemoryStore.prompt(device, userMessage),chatOptions);
 
             ChatResponse chatResponse = chatModel.call(prompt);
             if (chatResponse == null || chatResponse.getResult().getOutput().getText() == null) {
                 logger.warn("模型响应为空或无生成内容");
                 return "抱歉，我在处理您的请求时遇到了问题。请稍后再试。";
             }
-            String response = chatResponse.getResult().getOutput().getText();
-            boolean hasToolCalls = chatResponse.hasToolCalls();
-            String messageType = SysMessage.MESSAGE_TYPE_NORMAL;// 默认消息类型为普通消息
-            if (!hasToolCalls) {// 非function消息才加入对话历史，避免调用混乱
-                // 更新历史消息缓存
-                session.addHistoryMessage(userMessage);
-                if (response != null && !response.isEmpty()) {
-                    session.addHistoryMessage(new AssistantMessage(response));
-                }
-            } else {
-                // TODO 后续还需要根据元数据判断是function_call还是mcp调用
-                // 检查元数据中是否包含工具调用标识
-                // 发生了工具调用，获取函数调用的名称，通过名称反查类型
-                // String functionName = chatResponse.getMetadata().get("function_name");
-                messageType = SysMessage.MESSAGE_TYPE_FUNCTION_CALL;// function消息类型
-            }
-            final String finalMessageType = messageType;
+            AssistantMessage assistantMessage =chatResponse.getResult().getOutput();
+
             Thread.startVirtualThread(() -> {// 异步持久化
-                // 保存用户消息，会被持久化至数据库。
-                chatMemoryStore.addUserMessage(device, message, finalMessageType);
-                if (response != null && !response.isEmpty()) {
-                    // 保存AI消息，会被持久化至数据库。
-                    chatMemoryStore.addAssistantMessage(device, response, finalMessageType);
-                }
+
+                // 保存AI消息，会被持久化至数据库。
+                chatMemoryStore.addMessage(device, assistantMessage,null);
             });
-            return response;
+            return assistantMessage.getText();
 
         } catch (Exception e) {
             logger.error("处理查询时出错: {}", e.getMessage(), e);
@@ -162,21 +136,10 @@ public Flux<ChatResponse> chatStream(ChatSession session, SysDevice device, Stri
                 .toolContext(TOOL_CONTEXT_SESSION_KEY, session)
                 .build();
 
-        if (session.getChatMemory() == null) {// 如果记忆没初始化，则初始化一下
-            initializeHistory(session);
-        }
-        // 获取格式化的历史记录（包含当前用户消息）
-        List<Message> historyMessages = session.getHistoryMessages();
-
         UserMessage userMessage = new UserMessage(message);
-        historyMessages.add(userMessage);
-        Prompt prompt = Prompt.builder().messages(historyMessages).chatOptions(chatOptions).build();
+        Prompt prompt = new Prompt(chatMemoryStore.prompt(device, userMessage),chatOptions);
 
         // 调用实际的流式聊天方法
-        // return chatModel.stream(prompt).map(response -> (response.getResult() == null
-        // || response.getResult().getOutput() == null
-        // || response.getResult().getOutput().getText() == null) ? ""
-        // : response.getResult().getOutput().getText());
         return chatModel.stream(prompt);
     }
 
@@ -230,19 +193,9 @@ public void initializeHistory(ChatSession chatSession) {
             return;
         }
         SysDevice device = chatSession.getSysDevice();
-        // 如果缓存中不存在该设备的历史记录，则初始化缓存。默认情况下，只缓存当前会话的聊天记录。
-        // 同一个设备重新连接至服务器，会被标识为不同的sessionId。
-        // 可以将这理解为spring-ai的conversation会话,将sessionId作为conversationId
-        // 从数据库加载历史记录
-        List<SysMessage> history = chatMemoryStore.getMessages(device.getDeviceId(),
-                SysMessage.MESSAGE_TYPE_NORMAL, DEFAULT_HISTORY_LIMIT);
-        String systemMessage = chatMemoryStore.getSystemMessage(device.getDeviceId(), device.getRoleId());
-        MessageWindowChatMemory chatMemory = MessageWindowChatMemory.builder()
-                .maxMessages(DEFAULT_HISTORY_LIMIT)
-                .build();// 创建一个新的MessageWindowChatMemory实例，限制为10条消息滚动
-        chatMemory.add(device.getSessionId(), new SystemMessage(systemMessage));
-        chatMemory.add(device.getSessionId(), convert(history));
-        chatSession.setChatMemory(chatMemory);
+
+        // 从数据库加载历史记录，初始化缓存。
+        List<Message> history = chatMemoryStore.initHistory(device.getDeviceId());
         logger.info("已初始化设备 {} 的历史记录缓存，共 {} 条消息", device.getDeviceId(), history.size());
     }
 
@@ -255,46 +208,6 @@ public void clearMessageCache(String deviceId) {
         chatMemoryStore.clearMessages(deviceId);
     }
 
-
-    /**
-     * 通用添加消息
-     *
-     * @param message     消息内容
-     * @param role        角色名称
-     * @param messageType 消息类型
-     */
-    public void addMessage(SysDevice device, String message, String role, String messageType, String audioPath) {
-        chatMemoryStore.addMessage(device.getDeviceId(), device.getSessionId(), role, message, device.getRoleId(),
-                messageType, audioPath);
-    }
-
-    /**
-     * 将数据库记录的SysMessag转换为spring-ai的Message。
-     * 加载的历史都是普通消息(SysMessage.MESSAGE_TYPE_NORMAL)
-     * 
-     * @param messages
-     * @return
-     */
-    private List<Message> convert(List<SysMessage> messages) {
-        if (messages == null || messages.isEmpty()) {
-            return Collections.emptyList();
-        }
-        return messages.stream()
-                .filter(message -> MessageType.ASSISTANT.getValue().equals(message.getSender())
-                        || MessageType.USER.getValue().equals(message.getSender()))
-                .map(message -> {
-                    String role = message.getSender();
-                    // 一般消息("messageType", "NORMAL");//默认为普通消息
-                    Map<String, Object> metadata = Map.of("messageId", message.getMessageId(), "messageType",
-                            message.getMessageType());
-                    return switch (role) {
-                        case "assistant" -> new AssistantMessage(message.getMessage(), metadata);
-                        case "user" -> UserMessage.builder().text(message.getMessage()).metadata(metadata).build();
-                        default -> throw new IllegalArgumentException("Invalid role: " + role);
-                    };
-                }).collect(Collectors.toList());
-    }
-
     /**
      * 判断文本是否包含实质性内容（不仅仅是空白字符或标点符号）
      *
@@ -458,30 +371,21 @@ void persistMessages(String toolName) {
             // TODO
             // 需要进一步看看ChatModel在流式响应里是如何判断hasTools的，或者直接基于Flux<ChatResponse>已封装好的对象hasToolCalls判断
             boolean hasToolCalls = toolName != null && !toolName.isEmpty();
-            String messageType = hasToolCalls ? SysMessage.MESSAGE_TYPE_FUNCTION_CALL : SysMessage.MESSAGE_TYPE_NORMAL;// TODO
-                                                                                                                       // 后续可以根据名称区分function还是mcp，来细分类型
+            String messageType = hasToolCalls ? SysMessage.MESSAGE_TYPE_FUNCTION_CALL : SysMessage.MESSAGE_TYPE_NORMAL;
+            // TODO
+            // 后续可以根据名称区分function还是mcp，来细分类型
 
             UserMessage userMessage = new UserMessage(message);
 
-            // 获取当前对话ID
-            String dialogueId = session.getDialogueId();
-
-            if (!hasToolCalls) {// 非function消息才加入对话历史，避免调用混乱
-                session.addHistoryMessage(userMessage);
-                if (!fullResponse.isEmpty()) {
-                    AssistantMessage assistantMessage = new AssistantMessage(fullResponse.toString());
-                    session.addHistoryMessage(assistantMessage);
-                }
-            }
             Thread.startVirtualThread(() -> {// 异步持久化
                 String userAudioPath = session.getUserAudioPath();
-                addMessage(session.getSysDevice(), userMessage.getText(), userMessage.getMessageType().getValue(),
-                        messageType, userAudioPath);
+                chatMemoryStore.addMessage(session.getSysDevice(), userMessage,  userAudioPath);
+
                 if (!fullResponse.isEmpty()) {
                     AssistantMessage assistantMessage = new AssistantMessage(fullResponse.toString());
                     String assistAudioPath = session.getAssistantAudioPath();
-                    addMessage(session.getSysDevice(), assistantMessage.getText(),
-                            assistantMessage.getMessageType().getValue(), messageType, assistAudioPath);
+
+                    chatMemoryStore.addMessage(session.getSysDevice(), assistantMessage, assistAudioPath);
                 }
             });
         }
diff --git a/src/main/java/com/xiaozhi/dialogue/llm/memory/ChatMemoryStore.java b/src/main/java/com/xiaozhi/dialogue/llm/memory/ChatMemoryStore.java
@@ -2,41 +2,33 @@
 
 import com.xiaozhi.entity.SysDevice;
 import com.xiaozhi.entity.SysMessage;
+import org.springframework.ai.chat.messages.AssistantMessage;
+import org.springframework.ai.chat.messages.Message;
+import org.springframework.ai.chat.messages.UserMessage;
+
 import java.util.List;
 
 /**
  * 聊天记忆接口
  * 负责管理聊天历史记录
  */
 public interface ChatMemoryStore {
-    
-    /**
-     * 添加消息到历史记录
-     * 
-     * @param deviceId 设备ID
-     * @param sessionId 会话ID
-     * @param sender 发送者
-     * @param content 内容
-     * @param roleId 角色ID
-     * @param messageType 消息类型
-     */
-    void addMessage(String deviceId, String sessionId, String sender, String content, Integer roleId, String messageType, String audioPath);
+    // 历史记录默认限制数量
+    int DEFAULT_HISTORY_LIMIT = 10;
 
-    // TODO 最终要去掉
-    void addUserMessage(SysDevice device, String message, String messageType);
+    void addMessage(SysDevice device, UserMessage message, String audioPath);
 
-    void addAssistantMessage(SysDevice device, String message, String messageType);
+    void addMessage(SysDevice device, AssistantMessage message, String audioPath);
 
+    List<Message> prompt(SysDevice device,UserMessage userMessage);
 
     /**
-     * 获取历史消息
+     * 初始化历史消息
      *
-     * @param deviceId 设备ID
-     * @param messageType 指定查询的消息类型 - 传null查所有消息
-     * @param limit 消息数量限制
+     * @param deviceId 设备标识
      * @return 历史消息列表
      */
-    List<SysMessage> getMessages(String deviceId, String messageType, Integer limit);
+    List<Message> initHistory(String deviceId);
 
     /**
      * 清除设备的历史记录
diff --git a/src/main/java/com/xiaozhi/dialogue/llm/memory/DatabaseChatMemory.java b/src/main/java/com/xiaozhi/dialogue/llm/memory/DatabaseChatMemory.java
@@ -33,25 +33,28 @@ public class DatabaseChatMemory  {
     private TtsServiceFactory ttsService;
 
     public void addMessage(String deviceId, String sessionId, String sender, String content, Integer roleId, String messageType, String audioPath) {
-        try {
-            SysMessage message = new SysMessage();
-            message.setDeviceId(deviceId);
-            message.setSessionId(sessionId);
-            message.setSender(sender);
-            message.setMessage(content);
-            message.setRoleId(roleId);
-            message.setMessageType(messageType);
-            if (sender == "assistant") {
-                // 目前生成的语音保存采用默认的语音合成服务，后续可以考虑支持自定义语音合成服务
-                // todo
-                message.setAudioPath(ttsService.getDefaultTtsService().textToSpeech(content));
-            } else {
-                message.setAudioPath(audioPath);
+        // TODO 异步虚拟线程处理持久化。
+        Thread.startVirtualThread(() -> {
+            try {
+                SysMessage message = new SysMessage();
+                message.setDeviceId(deviceId);
+                message.setSessionId(sessionId);
+                message.setSender(sender);
+                message.setMessage(content);
+                message.setRoleId(roleId);
+                message.setMessageType(messageType);
+                if (sender == "assistant") {
+                    // 目前生成的语音保存采用默认的语音合成服务，后续可以考虑支持自定义语音合成服务
+                    // todo
+                    message.setAudioPath(ttsService.getDefaultTtsService().textToSpeech(content));
+                } else {
+                    message.setAudioPath(audioPath);
+                }
+                messageService.add(message);
+            } catch (Exception e) {
+                logger.error("保存消息时出错: {}", e.getMessage(), e);
             }
-            messageService.add(message);
-        } catch (Exception e) {
-            logger.error("保存消息时出错: {}", e.getMessage(), e);
-        }
+        });
     }
 
 
diff --git a/src/main/java/com/xiaozhi/dialogue/llm/memory/MessageWindowChatMemory.java b/src/main/java/com/xiaozhi/dialogue/llm/memory/MessageWindowChatMemory.java

Original file line number	Diff line number	Diff line change
`@@ -177,6 +177,8 @@ public void afterConnectionClosed(String sessionId) {`
`177`	`177`	`audioService.cleanupSession(sessionId);`
`178`	`178`	`// 清理对话`
`179`	`179`	`dialogueService.cleanupSession(sessionId);`
	`180`	`+ // 清理ChatService缓存的对话历史。`
	`181`	`+ chatService.clearMessageCache(device.getDeviceId());`
`180`	`182`	`}`
`181`	`183`
`182`	`184`	`/**`