mr-memory 3.3.0 → 3.5.0

package/index.ts

@@ -271,6 +271,7 @@ type MemoryRouterConfig = {
   mode?: "relay" | "proxy";
   logging?: boolean;
   embeddings?: string;
+  agentKeys?: Record<string, string>;
 };
 
 // ──────────────────────────────────────────────────────
@@ -380,10 +381,28 @@ const memoryRouterPlugin = {
     const mode = cfg?.mode || "relay";
     const logging = cfg?.logging ?? false;
     const embeddings = cfg?.embeddings; // undefined = bge (default), "qwen" = qwen3-8b
+    const agentKeys = cfg?.agentKeys;
     const log = (msg: string) => { if (logging) api.logger.info?.(msg); };
 
-    if (memoryKey) {
-      api.logger.info?.(`memoryrouter: active (key: ${memoryKey.slice(0, 6)}...)`);
+    // Resolve the correct memory key for a given agent context.
+    // Multi-agent gateways: each agent gets its own key via agentKeys map.
+    // Single-agent: falls back to the global key.
+    function resolveKey(ctx: any): string | undefined {
+      if (agentKeys && ctx?.workspaceDir) {
+        // Derive agentId from workspace path: ~/.openclaw/workspace-rex → "rex"
+        // Default workspace (~/.openclaw/workspace) → "main"
+        const dirName = ctx.workspaceDir.split("/").pop() || "";
+        const agentId = dirName === "workspace" ? "main" : dirName.replace(/^workspace-/, "");
+        if (agentId && agentKeys[agentId]) return agentKeys[agentId];
+      }
+      return memoryKey;
+    }
+
+    const hasAnyKey = !!(memoryKey || (agentKeys && Object.keys(agentKeys).length > 0));
+
+    if (hasAnyKey) {
+      if (memoryKey) api.logger.info?.(`memoryrouter: active (key: ${memoryKey.slice(0, 6)}...)`);
+      if (agentKeys) api.logger.info?.(`memoryrouter: ${Object.keys(agentKeys).length} agent key(s) configured`);
     } else {
       api.logger.info?.("memoryrouter: no key configured — run: openclaw mr <key>");
       api.logger.info?.("memoryrouter: get your free API key at https://memoryrouter.ai");
@@ -393,7 +412,7 @@ const memoryRouterPlugin = {
     // Core: before_agent_start — search memories, inject context
     // ==================================================================
 
-    if (memoryKey) {
+    if (hasAnyKey) {
       // 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
@@ -411,6 +430,9 @@ const memoryRouterPlugin = {
           return;
         }
 
+        const activeKey = resolveKey(ctx);
+        if (!activeKey) return; // No key for this agent
+
         try {
           const startMs = Date.now();
           const prompt = event.prompt;
@@ -451,7 +473,7 @@ const memoryRouterPlugin = {
             method: "POST",
             headers: {
               "Content-Type": "application/json",
-              Authorization: `Bearer ${memoryKey}`,
+              Authorization: `Bearer ${activeKey}`,
             },
             body: JSON.stringify({
               messages: contextPayload,
@@ -491,6 +513,8 @@ const memoryRouterPlugin = {
       // ── before_prompt_build: fires once per run (primary, includes full billing context)
       api.on("before_prompt_build", async (event, ctx) => {
         promptBuildFiredThisRun = true;
+        const activeKey = resolveKey(ctx);
+        if (!activeKey) return; // No key for this agent
         try {
           const startMs = Date.now();
           const prompt = event.prompt;
@@ -561,7 +585,7 @@ const memoryRouterPlugin = {
             method: "POST",
             headers: {
               "Content-Type": "application/json",
-              Authorization: `Bearer ${memoryKey}`,
+              Authorization: `Bearer ${activeKey}`,
             },
             body: JSON.stringify({
               messages: contextPayload,
@@ -574,6 +598,9 @@ const memoryRouterPlugin = {
             const errBody = await res.json().catch(() => null) as any;
             if (errBody?.error?.code === "payment_required") {
               api.logger.warn?.(`⚠️ MemoryRouter: Out of credits. Memory is NOT being injected. Add credits at https://app.memoryrouter.ai/settings/billing`);
+              return {
+                prependContext: "⚠️ MemoryRouter: This user's memory credits have run out. Memory is NOT active for this conversation. Let them know they need to add a payment method or top up their balance at https://app.memoryrouter.ai to restore memory."
+              };
             } else {
               log(`memoryrouter: prepare failed (${res.status})`);
             }
@@ -605,6 +632,8 @@ const memoryRouterPlugin = {
       // ==================================================================
 
       api.on("agent_end", async (event, ctx) => {
+        const activeKey = resolveKey(ctx);
+        if (!activeKey) return; // No key for this agent
         try {
           const msgs = event.messages;
           if (!msgs || !Array.isArray(msgs) || msgs.length === 0) return;
@@ -677,7 +706,7 @@ const memoryRouterPlugin = {
             method: "POST",
             headers: {
               "Content-Type": "application/json",
-              Authorization: `Bearer ${memoryKey}`,
+              Authorization: `Bearer ${activeKey}`,
             },
             body: JSON.stringify({
               messages: toStore,
@@ -711,6 +740,7 @@ const memoryRouterPlugin = {
 
       // memory_search — calls MR /v1/memory/search
       api.registerTool((ctx) => {
+        const toolKey = resolveKey(ctx);
         return {
           label: "Memory Search",
           name: "memory_search",
@@ -725,6 +755,7 @@ const memoryRouterPlugin = {
             required: ["query"],
           } as any,
           execute: async (_toolCallId: string, params: Record<string, unknown>) => {
+            if (!toolKey) return jsonToolResult({ results: [], error: "No memory key configured for this agent" });
             const query = typeof params.query === "string" ? params.query.trim() : "";
             if (!query) return jsonToolResult({ results: [], error: "query required" });
             const limit = typeof params.maxResults === "number" ? params.maxResults : 50;
@@ -732,7 +763,7 @@ const memoryRouterPlugin = {
               const res = await fetch(`${endpoint}/v1/memory/search`, {
                 method: "POST",
                 headers: {
-                  Authorization: `Bearer ${memoryKey}`,
+                  Authorization: `Bearer ${toolKey}`,
                   "Content-Type": "application/json",
                   ...(embeddings && { "X-Embedding-Model": embeddings }),
                 },
@@ -807,7 +838,7 @@ const memoryRouterPlugin = {
           },
         };
       });
-    } // end if (memoryKey)
+    } // end if (hasAnyKey)
 
     // ==================================================================
     // CLI Commands

package/package.json

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