mr-memory 1.0.1 → 1.0.2

package/index.ts

@@ -190,7 +190,15 @@ const memoryRouterPlugin = {
             const stateDir = opts.brain ? path.resolve(opts.brain) : path.join(os.homedir(), ".openclaw");
             const workspacePath = opts.workspace ? path.resolve(opts.workspace) : process.cwd();
             const { runUpload } = await import("./upload.js");
-            await runUpload({ memoryKey, endpoint, targetPath, stateDir, workspacePath });
+            await runUpload({
+              memoryKey,
+              endpoint,
+              targetPath,
+              stateDir,
+              workspacePath,
+              hasWorkspaceFlag: !!opts.workspace,
+              hasBrainFlag: !!opts.brain,
+            });
           });
 
         mr.command("delete")

package/package.json

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

package/upload.ts

@@ -118,7 +118,7 @@ async function exists(p: string): Promise<boolean> {
   }
 }
 
-async function discoverFiles(workspacePath: string, stateDir: string): Promise<string[]> {
+async function discoverWorkspaceFiles(workspacePath: string): Promise<string[]> {
   const files: string[] = [];
 
   const memoryMd = path.join(workspacePath, "MEMORY.md");
@@ -136,6 +136,12 @@ async function discoverFiles(workspacePath: string, stateDir: string): Promise<s
     if (await exists(p)) files.push(p);
   }
 
+  return files;
+}
+
+async function discoverBrainFiles(stateDir: string): Promise<string[]> {
+  const files: string[] = [];
+
   const sessionsDir = path.join(stateDir, "agents", "main", "sessions");
   if (await exists(sessionsDir)) {
     const allSessionFiles = await fs.readdir(sessionsDir);
@@ -171,6 +177,8 @@ export async function runUpload(params: {
   targetPath?: string;
   stateDir: string;
   workspacePath?: string;
+  hasWorkspaceFlag?: boolean;
+  hasBrainFlag?: boolean;
 }): Promise<void> {
   const { memoryKey, endpoint, targetPath, stateDir } = params;
   const uploadUrl = `${endpoint}/v1/memory/upload`;
@@ -198,8 +206,17 @@ export async function runUpload(params: {
     } else {
       files = [resolved];
     }
+  } else if (params.hasBrainFlag && !params.hasWorkspaceFlag) {
+    // --brain only: upload sessions from brain path
+    files = await discoverBrainFiles(stateDir);
+  } else if (params.hasWorkspaceFlag && !params.hasBrainFlag) {
+    // --workspace only: upload workspace files only
+    files = await discoverWorkspaceFiles(workspacePath);
   } else {
-    files = await discoverFiles(workspacePath, stateDir);
+    // No flags or both flags: upload both workspace + sessions
+    const wsFiles = await discoverWorkspaceFiles(workspacePath);
+    const brainFiles = await discoverBrainFiles(stateDir);
+    files = [...wsFiles, ...brainFiles];
   }
 
   if (files.length === 0) {