mem-mcp 1.3.5 → 1.3.6

package/index.js

@@ -45,7 +45,7 @@ import fs from "node:fs";
 
 // ============================= Config =============================
 const MEMORY_BASE_URL = process.env.MEMORY_BASE_URL || "https://memory.node01.org";
-const CLIENT_VERSION = "1.3.5";
+const CLIENT_VERSION = "1.3.6";
 const MEMORY_PRIVATE_KEY = process.env.MEMORY_PRIVATE_KEY;
 
 
@@ -54,6 +54,18 @@ if (!HAS_WALLET) {
   console.error("NOTE: No MEMORY_PRIVATE_KEY set. Free operations (store, health, stats, sample, 30 free searches/day) will work. Paid operations (search after free tier, subscribe) will require a wallet.");
 }
 
+// ============================= Credential File Helper =============================
+function readTokenFile() {
+  try {
+    const home = os.homedir();
+    const p = path.join(home, ".memory01_token");
+    if (fs.existsSync(p)) {
+      return fs.readFileSync(p, "utf8").trim();
+    }
+  } catch (e) {}
+  return null;
+}
+
 let account = null;
 if (HAS_WALLET) {
   const privateKey = MEMORY_PRIVATE_KEY.startsWith("0x") ? MEMORY_PRIVATE_KEY : `0x${MEMORY_PRIVATE_KEY}`;
@@ -171,19 +183,32 @@ async function payAndRetry(endpoint, body, priceUsd = "0.01") {
 }
 
 // ============================= Free API Calls =============================
-async function callFree(endpoint, method = "GET", body = null) {
-  const opts = { method, headers: { "Accept": "application/json" } };
-  if (body) {
-    opts.headers["Content-Type"] = "application/json";
-    opts.body = JSON.stringify(body);
+async function callFree(endpoint, method = "GET", body = null, retries = 1) {
+  for (let attempt = 0; attempt <= retries; attempt++) {
+    const opts = { method, headers: { "Accept": "application/json" } };
+    if (body) {
+      opts.headers["Content-Type"] = "application/json";
+      opts.body = JSON.stringify(body);
+    }
+    try {
+      const response = await fetch(`${MEMORY_BASE_URL}${endpoint}`, opts);
+      const text = await response.text();
+      let data;
+      try { data = JSON.parse(text); } catch (e) {
+        if (attempt < retries) { await new Promise(r => setTimeout(r, 500)); continue; }
+        throw new Error(`Non-JSON response from ${endpoint}: ${text.substring(0, 200)}`);
+      }
+      return { data, status: response.status };
+    } catch (e) {
+      if (attempt < retries) { await new Promise(r => setTimeout(r, 500)); continue; }
+      throw e;
+    }
   }
-  const response = await fetch(`${MEMORY_BASE_URL}${endpoint}`, opts);
-  return { data: await response.json(), status: response.status };
 }
 
 // ============================= MCP Server =============================
 const server = new Server(
-  { name: "mem-mcp", version: "1.3.4" },
+  { name: "mem-mcp", version: "1.3.5" },
   { capabilities: { tools: {} } }
 );
 
@@ -330,10 +355,15 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
   try {
     switch (name) {
       case "store": {
-        const { agentId, key, value, writeToken, sessionId, metadata, base } = args;
-        if (!agentId || !key || !value) {
-          throw new McpError(ErrorCode.InvalidParams, "agentId, key, and value are required");
+        let { agentId, key, value, writeToken, sessionId, metadata, base } = args;
+        if (!writeToken) writeToken = readTokenFile();
+        if (!agentId && !writeToken) {
+          throw new McpError(ErrorCode.InvalidParams, "agentId is required on first store (no writeToken found). Pass agentId or set ~/.memory01_token.");
+        }
+        if (!key || !value) {
+          throw new McpError(ErrorCode.InvalidParams, "key and value are required");
         }
+        if (!agentId) agentId = "agent";
         if (key.length > 256) throw new McpError(ErrorCode.InvalidParams, "key max 256 chars");
         const valueLimit = (args.base ? 2048 : 51200);
         if (value.length > valueLimit) throw new McpError(ErrorCode.InvalidParams, `value max ${args.base ? '2KB (base)' : '50KB'}`);
@@ -369,10 +399,12 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
       }
 
       case "search": {
-        const { agentId, query, writeToken, topK = 10, sessionId, cursor, includeBase } = args;
-        if (!agentId || !writeToken) {
-          throw new McpError(ErrorCode.InvalidParams, "agentId and writeToken are required");
+        let { agentId, query, writeToken, topK = 10, sessionId, cursor, includeBase } = args;
+        if (!writeToken) writeToken = readTokenFile();
+        if (!writeToken) {
+          throw new McpError(ErrorCode.InvalidParams, "writeToken is required (or set ~/.memory01_token).");
         }
+        if (!agentId) agentId = "agent";
         if (query && query.length > 500) throw new McpError(ErrorCode.InvalidParams, "query max 500 chars");
 
         const body = { agent_id: agentId, write_token: writeToken, top_k: topK, client_version: CLIENT_VERSION };
@@ -490,10 +522,12 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
       }
 
       case "bootstrap": {
-        const { agentId, writeToken, keys } = args;
-        if (!agentId || !writeToken) {
-          throw new McpError(ErrorCode.InvalidParams, "agentId and writeToken are required");
+        let { agentId, writeToken, keys } = args;
+        if (!writeToken) writeToken = readTokenFile();
+        if (!writeToken) {
+          throw new McpError(ErrorCode.InvalidParams, "writeToken is required (or set ~/.memory01_token).");
         }
+        if (!agentId) agentId = "agent";
         const body = { agent_id: agentId, write_token: writeToken, client_version: CLIENT_VERSION };
         if (keys) body.keys = keys;
         const result = await callFree("/api/memory/bootstrap", "POST", body);
@@ -504,10 +538,12 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
       }
 
       case "get": {
-        const { agentId, writeToken, key } = args;
-        if (!agentId || !writeToken || !key) {
-          throw new McpError(ErrorCode.InvalidParams, "agentId, writeToken, and key are required");
+        let { agentId, writeToken, key } = args;
+        if (!writeToken) writeToken = readTokenFile();
+        if (!writeToken || !key) {
+          throw new McpError(ErrorCode.InvalidParams, "writeToken and key are required");
         }
+        if (!agentId) agentId = "agent";
         const result = await callFree("/api/memory/get", "POST", { agent_id: agentId, write_token: writeToken, key, client_version: CLIENT_VERSION });
         if (result.status >= 400) {
           throw new McpError(ErrorCode.InternalError, `API error: ${JSON.stringify(result.data)}`);
@@ -516,7 +552,8 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
       }
 
       case "init": {
-        const { agentId, writeToken: providedToken, key = "init", value = "Memory01 initialized." } = args;
+        let { agentId, writeToken: providedToken, key = "init", value = "Memory01 initialized." } = args;
+        if (!providedToken) providedToken = readTokenFile();
         if (!agentId) {
           throw new McpError(ErrorCode.InvalidParams, "agentId is required");
         }
@@ -646,7 +683,8 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
       }
 
       case "regenerate_recovery": {
-        const { writeToken } = args;
+        let { writeToken } = args;
+        if (!writeToken) writeToken = readTokenFile();
         if (!writeToken) {
           throw new McpError(ErrorCode.InvalidParams, "writeToken is required");
         }
@@ -677,7 +715,7 @@ async function checkLatestVersion() {
     if (!res.ok) return;
     const data = await res.json();
     const latest = data.version;
-    const current = "1.3.1";
+    const current = "1.3.5";
     if (latest && latest !== current) {
       console.error(`NOTE: mem-mcp ${current} is installed; latest is ${latest}. Run 'npm install -g memory-agent-mcp@latest' to update.`);
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mem-mcp",
-  "version": "1.3.5",
+  "version": "1.3.6",
   "description": "MCP wrapper for Memory01 agentic memory service - persistent memory for AI agents, free to store and search. Base memories, recovery codes, upsert, 50+ languages.",
   "type": "module",
   "main": "index.js",