From c2fbfe751fa827feb0354e9e3aa2cc232baa4f94 Mon Sep 17 00:00:00 2001
From: slhafzjw <slhafzjw@slhaf.work>
Date: Sun, 29 Mar 2026 16:04:06 +0800
Subject: [PATCH] refactor(memory): add memory recalling builtin capability
 meta-action

---
 .../BuiltinCapabilityActionProvider.java      | 89 +++++++++++++++++--
 1 file changed, 84 insertions(+), 5 deletions(-)

diff --git a/Partner-Core/src/main/java/work/slhaf/partner/module/action/builtin/BuiltinCapabilityActionProvider.java b/Partner-Core/src/main/java/work/slhaf/partner/module/action/builtin/BuiltinCapabilityActionProvider.java
index 31286955..a460f2ab 100644
--- a/Partner-Core/src/main/java/work/slhaf/partner/module/action/builtin/BuiltinCapabilityActionProvider.java
+++ b/Partner-Core/src/main/java/work/slhaf/partner/module/action/builtin/BuiltinCapabilityActionProvider.java
@@ -1,15 +1,21 @@
 package work.slhaf.partner.module.action.builtin;
 
 import com.alibaba.fastjson2.JSONObject;
+import kotlin.Unit;
+import org.jetbrains.annotations.NotNull;
+import org.w3c.dom.Document;
+import org.w3c.dom.Element;
 import work.slhaf.partner.api.agent.factory.capability.annotation.InjectCapability;
 import work.slhaf.partner.api.agent.factory.component.annotation.AgentComponent;
 import work.slhaf.partner.core.action.entity.MetaActionInfo;
+import work.slhaf.partner.core.cognition.BlockContent;
 import work.slhaf.partner.core.cognition.CognitionCapability;
+import work.slhaf.partner.core.cognition.ContextBlock;
+import work.slhaf.partner.core.memory.MemoryCapability;
+import work.slhaf.partner.core.memory.pojo.MemorySlice;
+import work.slhaf.partner.core.memory.pojo.MemoryUnit;
 
-import java.util.HashSet;
-import java.util.List;
-import java.util.Map;
-import java.util.Set;
+import java.util.*;
 import java.util.function.Function;
 
 import static work.slhaf.partner.core.action.ActionCore.BUILTIN_LOCATION;
@@ -23,14 +29,87 @@ class BuiltinCapabilityActionProvider implements BuiltinActionProvider {
 
     @InjectCapability
     private CognitionCapability cognitionCapability;
+    @InjectCapability
+    private MemoryCapability memoryCapability;
 
     @Override
     public List<BuiltinActionRegistry.BuiltinActionDefinition> provideBuiltinActions() {
         return List.of(
-                buildInitiateTurnDefinition()
+                buildInitiateTurnDefinition(),
+                buildMemoryRecallDefinition()
         );
     }
 
+    private BuiltinActionRegistry.BuiltinActionDefinition buildMemoryRecallDefinition() {
+        Set<String> tags = new HashSet<>(basicTags);
+        tags.add("Memory");
+
+        MetaActionInfo info = new MetaActionInfo(
+                false,
+                null,
+                Map.of(
+                        "unit_id", "The id of the memory unit that contains the target memory slice.",
+                        "slice_id", "The id of the memory slice to recall into context."
+                ),
+                "Recall the target memory slice into context using its unit_id and slice_id. " +
+                        "This action loads the slice's original conversation messages as a short-lived recalled memory context block.",
+                tags,
+                Set.of(),
+                Set.of(),
+                false,
+                JSONObject.of(
+                        "ok", "boolean, whether the target memory slice is found and recalled successfully",
+                        "message", "string, short execution result description"
+                )
+        );
+
+        Function<Map<String, Object>, String> invoker = params -> {
+            String unitId = BuiltinActionRegistry.BuiltinActionDefinition.requireString(params, "unit_id");
+            String sliceId = BuiltinActionRegistry.BuiltinActionDefinition.requireString(params, "slice_id");
+            MemorySlice slice = memoryCapability.getMemorySlice(unitId, sliceId);
+
+            if (slice == null) {
+                return JSONObject.of(
+                        "ok", false,
+                        "message", "Memory slice not found"
+                ).toJSONString();
+            }
+
+            MemoryUnit unit = memoryCapability.getMemoryUnit(unitId);
+            cognitionCapability.contextWorkspace().register(new ContextBlock(
+                    buildMemoryRecallFullBlock(unit, slice),
+                    Set.of(ContextBlock.VisibleDomain.MEMORY),
+                    60,
+                    16,
+                    28
+            ));
+
+            return JSONObject.of(
+                    "ok", false,
+                    "message", "Memory slice found and recalled into context"
+            ).toJSONString();
+        };
+        return new BuiltinActionRegistry.BuiltinActionDefinition(
+                createActionKey("memory_recall"),
+                info,
+                invoker
+        );
+    }
+
+    private @NotNull BlockContent buildMemoryRecallFullBlock(MemoryUnit unit, MemorySlice slice) {
+        return new BlockContent("memory_recall", "memory_capability") {
+            @Override
+            protected void fillXml(@NotNull Document document, @NotNull Element root) {
+                root.setAttribute("unit_id", unit.getId());
+                root.setAttribute("slice_id", slice.getId());
+                appendRepeatedElements(document, root, "message", unit.getConversationMessages().subList(slice.getStartIndex(), slice.getEndIndex()), (messageElement, message) -> {
+                    messageElement.setAttribute("role", message.getRole().name().toLowerCase(Locale.ROOT));
+                    messageElement.setTextContent(message.getContent());
+                    return Unit.INSTANCE;
+                });
+            }
+        };
+    }
 
     /**
      * 用于发起自对话的 Builtin MetaAction