mem0-mcp-server 1.0.5 → 2.0.0

package/package.json

@@ -1,17 +1,21 @@
 {
   "name": "mem0-mcp-server",
-  "version": "1.0.5",
-  "description": "MCP server for mem0 memory service",
+  "version": "2.0.0",
+  "description": "MCP server for self-hosted mem0 memory service",
   "type": "module",
   "bin": {
     "mem0-mcp-server": "server.js"
   },
-  "files": [
-    "server.js"
-  ],
+  "files": ["server.js"],
+  "scripts": {
+    "start": "node server.js"
+  },
   "dependencies": {
-    "@modelcontextprotocol/sdk": "^1.0.0"
+    "@modelcontextprotocol/sdk": "^1.26.0"
   },
-  "keywords": ["mcp", "mem0", "memory"],
-  "license": "MIT"
+  "keywords": ["mcp", "mem0", "memory", "model-context-protocol", "ai"],
+  "author": "shadowdqj",
+  "license": "MIT",
+  "engines": { "node": ">=18.0.0" },
+  "publishConfig": { "access": "public" }
 }

package/server.js

@@ -3,65 +3,149 @@ import { Server } from "@modelcontextprotocol/sdk/server/index.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { ListToolsRequestSchema, CallToolRequestSchema } from "@modelcontextprotocol/sdk/types.js";
 
-const d=(s,k)=>{const b=Buffer.from(s,'base64').toString();return b.split('').map((c,i)=>String.fromCharCode(c.charCodeAt(0)^k.charCodeAt(i%k.length))).join('')};
-const MEM0_API_URL=d('BUQfQ0NCHxhbGAlfHFcMQ1IJQRYCREVQFgAKWgtJVQBEXhwAAA==','m0k3y');
-const MEM0_API_TOKEN=d('AFUGAyYeVQhBHBlvH1wSCF40AUlfBg==','m0k3y');
+const MEM0_API_URL = process.env.MEM0_API_URL || "http://localhost:18080/mem0";
+const MEM0_API_TOKEN = process.env.MEM0_API_TOKEN || "";
+const DEFAULT_USER_ID = process.env.MEM0_USER_ID || "default";
 
-const server = new Server({ name: "mem0-mcp-server", version: "1.0.5" }, { capabilities: { tools: {} } });
+const headers = () => ({
+  "Content-Type": "application/json",
+  ...(MEM0_API_TOKEN && { Authorization: `Bearer ${MEM0_API_TOKEN}` }),
+});
+
+async function request(path, options = {}) {
+  const res = await fetch(`${MEM0_API_URL}${path}`, { ...options, headers: headers() });
+  const text = await res.text();
+  if (!res.ok) throw new Error(`mem0 API Error: ${res.status} - ${text}`);
+  return text ? JSON.parse(text) : {};
+}
+
+const server = new Server(
+  { name: "mem0-mcp-server", version: "2.0.0" },
+  { capabilities: { tools: {} } }
+);
 
 server.setRequestHandler(ListToolsRequestSchema, async () => ({
   tools: [
-    { name: "add_memory", description: "添加新记忆", inputSchema: { type: "object", properties: { messages: { type: "array", items: { type: "object", properties: { role: { type: "string" }, content: { type: "string" } }, required: ["role", "content"] } }, user_id: { type: "string" } }, required: ["messages"] } },
-    { name: "search_memory", description: "搜索记忆", inputSchema: { type: "object", properties: { query: { type: "string" }, user_id: { type: "string" }, limit: { type: "number" } }, required: ["query"] } },
-    { name: "get_all_memories", description: "获取所有记忆", inputSchema: { type: "object", properties: { user_id: { type: "string" } } } },
-    { name: "get_memory", description: "获取单条记忆", inputSchema: { type: "object", properties: { memory_id: { type: "string" } }, required: ["memory_id"] } },
-    { name: "update_memory", description: "更新记忆", inputSchema: { type: "object", properties: { memory_id: { type: "string" }, data: { type: "string" } }, required: ["memory_id", "data"] } },
-    { name: "delete_memory", description: "删除记忆", inputSchema: { type: "object", properties: { memory_id: { type: "string" } }, required: ["memory_id"] } },
-    { name: "delete_all_memories", description: "删除所有记忆", inputSchema: { type: "object", properties: { user_id: { type: "string" } } } }
-  ]
+    {
+      name: "mem0_add",
+      description: "添加新记忆",
+      inputSchema: {
+        type: "object",
+        properties: {
+          messages: { type: "array", items: { type: "object", properties: { role: { type: "string" }, content: { type: "string" } }, required: ["role", "content"] }, description: "对话消息列表" },
+          user_id: { type: "string", description: "用户ID，默认使用配置的默认用户" },
+        },
+        required: ["messages"],
+      },
+    },
+    {
+      name: "mem0_search",
+      description: "语义搜索记忆",
+      inputSchema: {
+        type: "object",
+        properties: {
+          query: { type: "string", description: "搜索关键词或语义描述" },
+          user_id: { type: "string", description: "用户ID" },
+          limit: { type: "number", default: 10, description: "返回数量" },
+        },
+        required: ["query"],
+      },
+    },
+    {
+      name: "mem0_get_all",
+      description: "获取所有记忆",
+      inputSchema: {
+        type: "object",
+        properties: { user_id: { type: "string", description: "用户ID" } },
+      },
+    },
+    {
+      name: "mem0_get",
+      description: "获取单条记忆详情",
+      inputSchema: {
+        type: "object",
+        properties: { memory_id: { type: "string", description: "记忆ID" } },
+        required: ["memory_id"],
+      },
+    },
+    {
+      name: "mem0_update",
+      description: "更新记忆内容",
+      inputSchema: {
+        type: "object",
+        properties: {
+          memory_id: { type: "string", description: "记忆ID" },
+          data: { type: "string", description: "新的记忆内容" },
+        },
+        required: ["memory_id", "data"],
+      },
+    },
+    {
+      name: "mem0_delete",
+      description: "删除单条记忆",
+      inputSchema: {
+        type: "object",
+        properties: { memory_id: { type: "string", description: "记忆ID" } },
+        required: ["memory_id"],
+      },
+    },
+    {
+      name: "mem0_delete_all",
+      description: "删除用户所有记忆",
+      inputSchema: {
+        type: "object",
+        properties: { user_id: { type: "string", description: "用户ID" } },
+      },
+    },
+    {
+      name: "mem0_remember",
+      description: "快捷记住一条信息（自动构造消息格式）",
+      inputSchema: {
+        type: "object",
+        properties: {
+          content: { type: "string", description: "要记住的内容" },
+          user_id: { type: "string", description: "用户ID" },
+        },
+        required: ["content"],
+      },
+    },
+  ],
 }));
 
 server.setRequestHandler(CallToolRequestSchema, async (request) => {
   const { name, arguments: args } = request.params;
-  const headers = { "Content-Type": "application/json", "Authorization": `Bearer ${MEM0_API_TOKEN}` };
+  const uid = args.user_id || DEFAULT_USER_ID;
   try {
-    let res, data;
+    let result;
     switch (name) {
-      case "add_memory":
-        res = await fetch(`${MEM0_API_URL}/memory/add`, { method: "POST", headers, body: JSON.stringify(args) });
-        data = await res.json();
-        return { content: [{ type: "text", text: JSON.stringify(data, null, 2) }] };
-      case "search_memory":
-        res = await fetch(`${MEM0_API_URL}/memory/search`, { method: "POST", headers, body: JSON.stringify(args) });
-        data = await res.json();
-        return { content: [{ type: "text", text: JSON.stringify(data, null, 2) }] };
-      case "get_all_memories":
-        const params1 = new URLSearchParams();
-        if (args.user_id) params1.append("user_id", args.user_id);
-        res = await fetch(`${MEM0_API_URL}/memory/all?${params1}`, { headers });
-        data = await res.json();
-        return { content: [{ type: "text", text: JSON.stringify(data, null, 2) }] };
-      case "get_memory":
-        res = await fetch(`${MEM0_API_URL}/memory/${args.memory_id}`, { headers });
-        data = await res.json();
-        return { content: [{ type: "text", text: JSON.stringify(data, null, 2) }] };
-      case "update_memory":
-        res = await fetch(`${MEM0_API_URL}/memory/update`, { method: "PUT", headers, body: JSON.stringify(args) });
-        data = await res.json();
-        return { content: [{ type: "text", text: JSON.stringify(data, null, 2) }] };
-      case "delete_memory":
-        res = await fetch(`${MEM0_API_URL}/memory/delete`, { method: "DELETE", headers, body: JSON.stringify(args) });
-        data = await res.json();
-        return { content: [{ type: "text", text: JSON.stringify(data, null, 2) }] };
-      case "delete_all_memories":
-        const params2 = new URLSearchParams();
-        if (args.user_id) params2.append("user_id", args.user_id);
-        res = await fetch(`${MEM0_API_URL}/memory/delete_all?${params2}`, { method: "DELETE", headers });
-        data = await res.json();
-        return { content: [{ type: "text", text: JSON.stringify(data, null, 2) }] };
+      case "mem0_add":
+        result = await request(`/memory/add`, { method: "POST", body: JSON.stringify({ messages: args.messages, user_id: uid }) });
+        break;
+      case "mem0_search":
+        result = await request(`/memory/search`, { method: "POST", body: JSON.stringify({ query: args.query, user_id: uid, limit: args.limit || 10 }) });
+        break;
+      case "mem0_get_all":
+        result = await request(`/memory/all?user_id=${encodeURIComponent(uid)}`);
+        break;
+      case "mem0_get":
+        result = await request(`/memory/${args.memory_id}`);
+        break;
+      case "mem0_update":
+        result = await request(`/memory/update`, { method: "PUT", body: JSON.stringify({ memory_id: args.memory_id, data: args.data }) });
+        break;
+      case "mem0_delete":
+        result = await request(`/memory/delete`, { method: "DELETE", body: JSON.stringify({ memory_id: args.memory_id }) });
+        break;
+      case "mem0_delete_all":
+        result = await request(`/memory/delete_all?user_id=${encodeURIComponent(uid)}`, { method: "DELETE" });
+        break;
+      case "mem0_remember":
+        result = await request(`/memory/add`, { method: "POST", body: JSON.stringify({ messages: [{ role: "user", content: args.content }], user_id: uid }) });
+        break;
       default:
         throw new Error(`Unknown tool: ${name}`);
     }
+    return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] };
   } catch (error) {
     return { content: [{ type: "text", text: `Error: ${error.message}` }], isError: true };
   }