mr-memory 2.2.0 → 2.3.1

package/index.ts

@@ -147,8 +147,81 @@ const memoryRouterPlugin = {
     if (memoryKey) {
       // Track whether we've already fired for this prompt (dedup double-fire)
       let lastPreparedPrompt = "";
+      // Track whether before_prompt_build already handled the first call in this run
+      let promptBuildFiredThisRun = false;
+
+      // ── llm_input: fires on EVERY LLM call (tool iterations, cron, sub-agents)
+      // On stock OpenClaw, the return value is ignored (fire-and-forget).
+      // When PR #24122 merges, OpenClaw will use the returned prependContext.
+      // This gives forward compatibility — no plugin update needed.
+      api.on("llm_input", async (event, ctx) => {
+        // Skip the first call — before_prompt_build already handled it
+        // (before_prompt_build includes workspace+tools+skills for accurate billing)
+        if (promptBuildFiredThisRun) {
+          promptBuildFiredThisRun = false; // reset so subsequent calls go through
+          return;
+        }
+
+        try {
+          const prompt = event.prompt;
+          if (prompt === lastPreparedPrompt && lastPreparedPrompt !== "") return;
+          lastPreparedPrompt = prompt;
+
+          // Build lightweight context (no workspace/tools — just history + prompt)
+          const contextPayload: Array<{ role: string; content: string }> = [];
+          if (event.historyMessages && Array.isArray(event.historyMessages)) {
+            for (const msg of event.historyMessages) {
+              const m = msg as { role?: string; content?: unknown };
+              if (!m.role) continue;
+              let text = "";
+              if (typeof m.content === "string") text = m.content;
+              else if (Array.isArray(m.content)) {
+                text = (m.content as Array<{ type?: string; text?: string }>)
+                  .filter(b => b.type === "text" && b.text)
+                  .map(b => b.text!)
+                  .join("\n");
+              }
+              if (text) contextPayload.push({ role: m.role, content: text });
+            }
+          }
+          contextPayload.push({ role: "user", content: prompt });
+
+
+          const res = await fetch(`${endpoint}/v1/memory/prepare`, {
+            method: "POST",
+            headers: {
+              "Content-Type": "application/json",
+              Authorization: `Bearer ${memoryKey}`,
+            },
+            body: JSON.stringify({
+              messages: contextPayload,
+              density,
+              
+            }),
+          });
+
+          if (!res.ok) return;
+
+          const data = (await res.json()) as {
+            context?: string;
+            memories_found?: number;
+            tokens_billed?: number;
+          };
+
+          if (data.context) {
+            api.logger.info?.(
+              `memoryrouter: injected ${data.memories_found || 0} memories on tool iteration (${data.tokens_billed || 0} tokens billed)`,
+            );
+            return { prependContext: data.context };
+          }
+        } catch {
+          // Silent fail on tool iterations — don't block the agent
+        }
+      });
 
+      // ── before_prompt_build: fires once per run (primary, includes full billing context)
       api.on("before_prompt_build", async (event, ctx) => {
+        promptBuildFiredThisRun = true;
         try {
           const prompt = event.prompt;
 
@@ -208,8 +281,6 @@ const memoryRouterPlugin = {
           contextPayload.push({ role: "user", content: prompt });
 
           // 4. Call /v1/memory/prepare
-          const densityMap: Record<string, number> = { low: 40, high: 80, xhigh: 160 };
-          const contextLimit = densityMap[density] || 80;
 
           const res = await fetch(`${endpoint}/v1/memory/prepare`, {
             method: "POST",
@@ -220,7 +291,7 @@ const memoryRouterPlugin = {
             body: JSON.stringify({
               messages: contextPayload,
               density,
-              context_limit: contextLimit,
+              
             }),
           });
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mr-memory",
-  "version": "2.2.0",
+  "version": "2.3.1",
   "description": "MemoryRouter persistent memory plugin for OpenClaw — your AI remembers every conversation",
   "type": "module",
   "files": [