From 3d1c2589447d042b2b4089a3ad3659a73fd07027 Mon Sep 17 00:00:00 2001
From: slhafzjw <slhafzjw@slhaf.work>
Date: Tue, 10 Mar 2026 21:15:35 +0800
Subject: [PATCH] refactor(memory): drop message cleanup before summarizing

---
 .../modules/memory/updater/MemoryUpdater.java | 23 +++----------------
 1 file changed, 3 insertions(+), 20 deletions(-)

diff --git a/Partner-Core/src/main/java/work/slhaf/partner/module/modules/memory/updater/MemoryUpdater.java b/Partner-Core/src/main/java/work/slhaf/partner/module/modules/memory/updater/MemoryUpdater.java
index efc21d5b..12039642 100644
--- a/Partner-Core/src/main/java/work/slhaf/partner/module/modules/memory/updater/MemoryUpdater.java
+++ b/Partner-Core/src/main/java/work/slhaf/partner/module/modules/memory/updater/MemoryUpdater.java
@@ -25,7 +25,6 @@ import work.slhaf.partner.module.modules.memory.updater.summarizer.entity.Summar
 import work.slhaf.partner.runtime.interaction.data.context.PartnerRunningFlowContext;
 
 import java.util.ArrayList;
-import java.util.Arrays;
 import java.util.List;
 import java.util.UUID;
 import java.util.concurrent.atomic.AtomicBoolean;
@@ -134,15 +133,14 @@ public class MemoryUpdater extends PostRunningAgentModule {
 
     private void updateMemory(List<Message> chatSnapshot) {
         log.debug("[MemoryUpdater] 记忆更新流程开始...");
-        List<Message> chatMessages = getCleanedMessages(chatSnapshot);
-        if (chatMessages.isEmpty()) {
+        if (chatSnapshot.isEmpty()) {
             return;
         }
-        SummarizeInput summarizeInput = new SummarizeInput(chatMessages, memoryRuntime.getTopicTree());
+        SummarizeInput summarizeInput = new SummarizeInput(chatSnapshot, memoryRuntime.getTopicTree());
         log.debug("[MemoryUpdater] 记忆更新-总结流程-输入: {}", JSONObject.toJSONString(summarizeInput));
         SummarizeResult summarizeResult = summarize(summarizeInput);
         log.debug("[MemoryUpdater] 记忆更新-总结流程-输出: {}", JSONObject.toJSONString(summarizeResult));
-        MemoryUnit memoryUnit = buildMemoryUnit(chatMessages, summarizeResult);
+        MemoryUnit memoryUnit = buildMemoryUnit(chatSnapshot, summarizeResult);
         memoryRuntime.recordMemory(
                 memoryUnit,
                 summarizeResult.getTopicPath(),
@@ -153,21 +151,6 @@ public class MemoryUpdater extends PostRunningAgentModule {
         log.debug("[MemoryUpdater] 记忆更新流程结束...");
     }
 
-    private List<Message> getCleanedMessages(List<Message> chatMessages) {
-        return chatMessages.stream()
-                .map(message -> {
-                    if (message.getRole() == Message.Character.ASSISTANT) {
-                        return message;
-                    }
-                    List<String> splitResult = Arrays.stream(message.getContent().split("\\*\\*")).toList();
-                    if (splitResult.isEmpty()) {
-                        return message;
-                    }
-                    String time = splitResult.getLast();
-                    return new Message(Message.Character.USER, message.getContent().replace("\r\n**" + time, ""));
-                }).toList();
-    }
-
     private SummarizeResult summarize(SummarizeInput summarizeInput) {
         singleSummarizer.execute(summarizeInput.getChatMessages());
         return multiSummarizer.execute(summarizeInput);