mr-memory 2.16.0 → 2.18.0

package/index.ts

@@ -178,6 +178,7 @@ type MemoryRouterConfig = {
   density?: "low" | "high" | "xhigh";
   mode?: "relay" | "proxy";
   logging?: boolean;
+  embeddings?: string;
 };
 
 // ──────────────────────────────────────────────────────
@@ -286,10 +287,12 @@ const memoryRouterPlugin = {
     const density = cfg?.density || "high";
     const mode = cfg?.mode || "relay";
     const logging = cfg?.logging ?? false;
+    const embeddings = cfg?.embeddings; // undefined = bge (default), "qwen" = qwen3-8b
     const log = (msg: string) => { if (logging) api.logger.info?.(msg); };
 
     if (memoryKey) {
-      api.logger.info?.(`memoryrouter: active (key: ${memoryKey.slice(0, 6)}..., mode: ${mode})`);
+      const modelLabel = embeddings ? `, embeddings: ${embeddings}` : "";
+      api.logger.info?.(`memoryrouter: active (key: ${memoryKey.slice(0, 6)}..., mode: ${mode}${modelLabel})`);
     } 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");
@@ -329,6 +332,8 @@ const memoryRouterPlugin = {
             for (const msg of event.historyMessages) {
               const m = msg as { role?: string; content?: unknown };
               if (!m.role) continue;
+              // Skip tool result/use messages — contains file reads, exec output, etc.
+              if (m.role !== "user" && m.role !== "assistant" && m.role !== "system") continue;
               let text = "";
               if (typeof m.content === "string") text = m.content;
               else if (Array.isArray(m.content)) {
@@ -361,7 +366,7 @@ const memoryRouterPlugin = {
               messages: contextPayload,
               session_id: ctx.sessionKey,
               density,
-              
+              ...(embeddings && { embeddings }),
             }),
           });
 
@@ -432,6 +437,8 @@ const memoryRouterPlugin = {
             for (const msg of event.messages) {
               const m = msg as { role?: string; content?: unknown };
               if (!m.role) continue;
+              // Skip tool result/use messages — contains file reads, exec output, etc.
+              if (m.role !== "user" && m.role !== "assistant" && m.role !== "system") continue;
               
               let text = "";
               if (typeof m.content === "string") {
@@ -471,7 +478,7 @@ const memoryRouterPlugin = {
               messages: contextPayload,
               session_id: ctx.sessionKey,
               density,
-              
+              ...(embeddings && { embeddings }),
             }),
           });
 
@@ -590,6 +597,7 @@ const memoryRouterPlugin = {
                 messages: toStore,
                 session_id: ctx.sessionKey,
                 model: "unknown",
+                ...(embeddings && { embeddings }),
               }),
             });
             if (!res.ok) {
@@ -759,6 +767,48 @@ const memoryRouterPlugin = {
             });
         }
 
+        // Embedding model command
+        mr.command("embeddings")
+          .description("Get or set embedding model (bge, qwen)")
+          .argument("[model]", "Embedding model: bge (default, 1024d) or qwen (4096d)")
+          .action(async (model: string | undefined) => {
+            if (!model) {
+              console.log(`Current embedding model: ${embeddings || "bge (default)"}`);
+              console.log(`\nAvailable models:`);
+              console.log(`  bge    BGE-M3 1024 dims  (Cloudflare Workers AI, ~18ms, free)`);
+              console.log(`  qwen   Qwen3-8B 4096 dims (HuggingFace, ~69ms, $0.80/hr)`);
+              console.log(`\nUsage: openclaw mr embeddings <model>`);
+              return;
+            }
+            const normalized = model.toLowerCase().trim();
+            const valid = ["bge", "bge-m3", "qwen", "qwen3", "qwen3-8b", "default"];
+            if (!valid.includes(normalized)) {
+              console.error(`Unknown model: "${model}". Valid options: bge, qwen`);
+              return;
+            }
+            if (!memoryKey) { console.error("Not configured. Run: openclaw mr <key>"); return; }
+            // "bge" / "default" → clear embeddings (use server default)
+            const newEmbeddings = (normalized === "bge" || normalized === "bge-m3" || normalized === "default")
+              ? undefined
+              : normalized.startsWith("qwen") ? "qwen" : normalized;
+            try {
+              await setPluginConfig(api, {
+                key: memoryKey,
+                endpoint: cfg?.endpoint,
+                density,
+                mode,
+                logging,
+                ...(newEmbeddings && { embeddings: newEmbeddings }),
+              });
+              const label = newEmbeddings || "bge (default)";
+              console.log(`✓ Embedding model set to ${label}`);
+              console.log(`  All future prepare/ingest/search will use ${label}`);
+              console.log(`\n  Restart gateway to apply: openclaw gateway restart`);
+            } catch (err) {
+              console.error(`Failed: ${err instanceof Error ? err.message : String(err)}`);
+            }
+          });
+
         mr.command("status")
           .description("Show MemoryRouter vault stats")
           .option("--json", "JSON output")
@@ -776,12 +826,15 @@ const memoryRouterPlugin = {
               return;
             }
             try {
-              const res = await fetch(`${endpoint}/v1/memory/stats`, {
+              const statsUrl = embeddings
+                ? `${endpoint}/v1/memory/stats?embeddings=${encodeURIComponent(embeddings)}`
+                : `${endpoint}/v1/memory/stats`;
+              const res = await fetch(statsUrl, {
                 headers: { Authorization: `Bearer ${memoryKey}` },
               });
               const data = (await res.json()) as { totalVectors?: number; totalTokens?: number };
               if (opts.json) {
-                console.log(JSON.stringify({ enabled: true, key: memoryKey, density, mode, stats: data }, null, 2));
+                console.log(JSON.stringify({ enabled: true, key: memoryKey, density, mode, embeddings: embeddings || "bge", stats: data }, null, 2));
               } else {
                 console.log("MemoryRouter Status");
                 console.log("───────────────────────────");
@@ -789,6 +842,7 @@ const memoryRouterPlugin = {
                 console.log(`Key:         ${memoryKey.slice(0, 6)}...${memoryKey.slice(-3)}`);
                 console.log(`Mode:        ${mode}`);
                 console.log(`Density:     ${density}`);
+                console.log(`Embeddings:  ${embeddings || "bge (default)"}`);
                 console.log(`Endpoint:    ${endpoint}`);
                 console.log(`Memories:    ${data.totalVectors ?? 0}`);
                 console.log(`Tokens:      ${data.totalTokens ?? 0}`);
@@ -815,7 +869,7 @@ const memoryRouterPlugin = {
                 ? path.resolve(configWorkspace.replace(/^~/, os.homedir()))
                 : path.join(os.homedir(), ".openclaw", "workspace");
             const { runUpload } = await import("./upload.js");
-            await runUpload({ memoryKey, endpoint, targetPath, stateDir, workspacePath, hasWorkspaceFlag: !!opts.workspace, hasBrainFlag: !!opts.brain });
+            await runUpload({ memoryKey, endpoint, targetPath, stateDir, workspacePath, hasWorkspaceFlag: !!opts.workspace, hasBrainFlag: !!opts.brain, embeddings });
           });
 
         mr.command("delete")
@@ -823,12 +877,16 @@ const memoryRouterPlugin = {
           .action(async () => {
             if (!memoryKey) { console.error("Not configured. Run: openclaw mr <key>"); return; }
             try {
-              const res = await fetch(`${endpoint}/v1/memory`, {
+              const deleteUrl = embeddings
+                ? `${endpoint}/v1/memory?embeddings=${encodeURIComponent(embeddings)}`
+                : `${endpoint}/v1/memory`;
+              const res = await fetch(deleteUrl, {
                 method: "DELETE",
                 headers: { Authorization: `Bearer ${memoryKey}` },
               });
               const data = (await res.json()) as { message?: string };
-              console.log(`✓ ${data.message || "Vault cleared"}`);
+              const modelLabel = embeddings ? ` (${embeddings})` : "";
+              console.log(`✓ ${data.message || `Vault cleared${modelLabel}`}`);
             } catch (err) {
               console.error(`Failed: ${err instanceof Error ? err.message : String(err)}`);
             }

package/package.json

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

package/upload.ts

@@ -353,8 +353,9 @@ export async function runUpload(params: {
   workspacePath?: string;
   hasWorkspaceFlag?: boolean;
   hasBrainFlag?: boolean;
+  embeddings?: string;
 }): Promise<void> {
-  const { memoryKey, endpoint, targetPath, stateDir } = params;
+  const { memoryKey, endpoint, targetPath, stateDir, embeddings } = params;
   const uploadUrl = `${endpoint}/v1/memory/upload`;
 
   // Validate API reachability
@@ -469,6 +470,7 @@ export async function runUpload(params: {
           headers: {
             Authorization: `Bearer ${memoryKey}`,
             "Content-Type": "text/plain",
+            ...(embeddings && { "X-Embedding-Model": embeddings }),
           },
           body: jsonlBody,
         },