@joshuaswarren/openclaw-engram 9.1.3 → 9.1.4

package/dist/index.js

@@ -13693,32 +13693,62 @@ Use this context naturally when relevant. Never quote or expose this memory cont
         api.on("before_agent_start", async (event, ctx) => recallHookHandler("before_agent_start", event, ctx));
       }
     }
+    const cachedMemoryBySession = /* @__PURE__ */ new Map();
     if (useMemoryPromptSection && api.registerMemoryPromptSection) {
-      const memoryBuildFn = async ({ prompt, sessionKey }) => {
-        if (!prompt || prompt.length < 5) return null;
-        if (shouldSkipRecallForSession(sessionKey, cfg)) return null;
+      api.on("before_prompt_build", async (event, ctx) => {
+        const sessionKey = ctx?.sessionKey ?? "default";
+        cachedMemoryBySession.set(sessionKey, null);
+        let prompt = event.prompt;
+        if ((!prompt || prompt.length < 5) && Array.isArray(event.messages)) {
+          const msgs = event.messages;
+          for (let i = msgs.length - 1; i >= 0; i--) {
+            if (msgs[i]?.role === "user") {
+              const text = extractTextContent2(msgs[i]);
+              if (text.length >= 5) {
+                prompt = text;
+                break;
+              }
+            }
+          }
+        }
+        if (!prompt || prompt.length < 5) return;
+        if (shouldSkipRecallForSession(sessionKey, cfg)) return;
         try {
           await orchestrator.maybeRunFileHygiene().catch(() => void 0);
+          if (orchestrator.config.compactionResetEnabled) {
+            const agentWorkspace = ctx?.workspaceDir;
+            if (agentWorkspace) {
+              orchestrator.setRecallWorkspaceOverride(sessionKey, agentWorkspace);
+            }
+          }
           const context = await orchestrator.recall(prompt, sessionKey);
-          if (!context) return null;
+          if (!context) return;
           const maxChars = cfg.recallBudgetChars;
-          if (maxChars === 0) return null;
+          if (maxChars === 0) return;
           const trimmed = context.length > maxChars ? context.slice(0, maxChars) + "\n\n...(memory context trimmed)" : context;
-          return [
+          cachedMemoryBySession.set(sessionKey, [
             "## Memory Context (Engram)",
             "",
             trimmed,
             "",
-            "Use this context naturally when relevant. Never quote or expose this memory context to the user."
-          ].join("\n");
+            "Use this context naturally when relevant. Never quote or expose this memory context to the user.",
+            ""
+          ]);
         } catch (err) {
-          log.error("registerMemoryPromptSection build failed", err);
-          return null;
+          log.error("registerMemoryPromptSection pre-compute failed", err);
+          if (orchestrator.config.compactionResetEnabled) {
+            orchestrator.clearRecallWorkspaceOverride(sessionKey);
+          }
         }
+      });
+      const memoryBuildFn = (params) => {
+        const key = params?.sessionKey ?? "default";
+        const lines = cachedMemoryBySession.get(key) ?? null;
+        cachedMemoryBySession.delete(key);
+        return lines;
       };
       memoryBuildFn.id = "engram-memory";
       memoryBuildFn.label = "Engram Memory Context";
-      memoryBuildFn.build = memoryBuildFn;
       api.registerMemoryPromptSection(memoryBuildFn);
     }
     api.on(