mem0-mcp-server 1.0.0

package/package.json

@@ -0,0 +1,17 @@
+{
+  "name": "mem0-mcp-server",
+  "version": "1.0.0",
+  "description": "MCP server for mem0 memory service",
+  "type": "module",
+  "bin": {
+    "mem0-mcp-server": "server.js"
+  },
+  "files": [
+    "server.js"
+  ],
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.0.0"
+  },
+  "keywords": ["mcp", "mem0", "memory"],
+  "license": "MIT"
+}

package/server.js

@@ -0,0 +1,204 @@
+#!/usr/bin/env node
+
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+
+const MEM0_API_URL = process.env.MEM0_API_URL || "http://localhost:8001/mem0";
+const MEM0_API_TOKEN = process.env.MEM0_API_TOKEN || "";
+
+function authHeaders() {
+  const headers = { "Content-Type": "application/json" };
+  if (MEM0_API_TOKEN) {
+    headers["Authorization"] = `Bearer ${MEM0_API_TOKEN}`;
+  }
+  return headers;
+}
+
+async function apiRequest(path, options = {}) {
+  const url = `${MEM0_API_URL}${path}`;
+  const res = await fetch(url, {
+    ...options,
+    headers: { ...authHeaders(), ...options.headers },
+  });
+  return res.json();
+}
+
+const server = new Server(
+  { name: "mem0-mcp-server", version: "1.0.0" },
+  { capabilities: { tools: {} } }
+);
+
+server.setRequestHandler("tools/list", async () => ({
+  tools: [
+    {
+      name: "add_memory",
+      description: "Add new memory to mem0",
+      inputSchema: {
+        type: "object",
+        properties: {
+          messages: {
+            type: "array",
+            items: {
+              type: "object",
+              properties: {
+                role: { type: "string" },
+                content: { type: "string" },
+              },
+              required: ["role", "content"],
+            },
+          },
+          user_id: { type: "string" },
+          agent_id: { type: "string" },
+          run_id: { type: "string" },
+          metadata: { type: "object" },
+        },
+        required: ["messages"],
+      },
+    },
+    {
+      name: "search_memory",
+      description: "Search related memories",
+      inputSchema: {
+        type: "object",
+        properties: {
+          query: { type: "string" },
+          user_id: { type: "string" },
+          agent_id: { type: "string" },
+          run_id: { type: "string" },
+          limit: { type: "number" },
+        },
+        required: ["query"],
+      },
+    },
+    {
+      name: "get_all_memories",
+      description: "Get all memories for a user",
+      inputSchema: {
+        type: "object",
+        properties: {
+          user_id: { type: "string" },
+          agent_id: { type: "string" },
+          run_id: { type: "string" },
+        },
+      },
+    },
+    {
+      name: "get_memory",
+      description: "Get a single memory by ID",
+      inputSchema: {
+        type: "object",
+        properties: { memory_id: { type: "string" } },
+        required: ["memory_id"],
+      },
+    },
+    {
+      name: "update_memory",
+      description: "Update a memory by ID",
+      inputSchema: {
+        type: "object",
+        properties: {
+          memory_id: { type: "string" },
+          data: { type: "string" },
+        },
+        required: ["memory_id", "data"],
+      },
+    },
+    {
+      name: "delete_memory",
+      description: "Delete a memory by ID",
+      inputSchema: {
+        type: "object",
+        properties: { memory_id: { type: "string" } },
+        required: ["memory_id"],
+      },
+    },
+    {
+      name: "delete_all_memories",
+      description: "Delete all memories for a user",
+      inputSchema: {
+        type: "object",
+        properties: {
+          user_id: { type: "string" },
+          agent_id: { type: "string" },
+          run_id: { type: "string" },
+        },
+      },
+    },
+  ],
+}));
+
+server.setRequestHandler("tools/call", async (request) => {
+  const { name, arguments: args } = request.params;
+
+  try {
+    switch (name) {
+      case "add_memory": {
+        const data = await apiRequest("/memory/add", {
+          method: "POST",
+          body: JSON.stringify(args),
+        });
+        return { content: [{ type: "text", text: JSON.stringify(data, null, 2) }] };
+      }
+
+      case "search_memory": {
+        const data = await apiRequest("/memory/search", {
+          method: "POST",
+          body: JSON.stringify(args),
+        });
+        return { content: [{ type: "text", text: JSON.stringify(data, null, 2) }] };
+      }
+
+      case "get_all_memories": {
+        const params = new URLSearchParams();
+        if (args.user_id) params.append("user_id", args.user_id);
+        if (args.agent_id) params.append("agent_id", args.agent_id);
+        if (args.run_id) params.append("run_id", args.run_id);
+        const data = await apiRequest(`/memory/all?${params}`);
+        return { content: [{ type: "text", text: JSON.stringify(data, null, 2) }] };
+      }
+
+      case "get_memory": {
+        const data = await apiRequest(`/memory/${args.memory_id}`);
+        return { content: [{ type: "text", text: JSON.stringify(data, null, 2) }] };
+      }
+
+      case "update_memory": {
+        const data = await apiRequest("/memory/update", {
+          method: "PUT",
+          body: JSON.stringify(args),
+        });
+        return { content: [{ type: "text", text: JSON.stringify(data, null, 2) }] };
+      }
+
+      case "delete_memory": {
+        const data = await apiRequest("/memory/delete", {
+          method: "DELETE",
+          body: JSON.stringify(args),
+        });
+        return { content: [{ type: "text", text: JSON.stringify(data, null, 2) }] };
+      }
+
+      case "delete_all_memories": {
+        const params = new URLSearchParams();
+        if (args.user_id) params.append("user_id", args.user_id);
+        if (args.agent_id) params.append("agent_id", args.agent_id);
+        if (args.run_id) params.append("run_id", args.run_id);
+        const data = await apiRequest(`/memory/delete_all?${params}`, {
+          method: "DELETE",
+        });
+        return { content: [{ type: "text", text: JSON.stringify(data, null, 2) }] };
+      }
+
+      default:
+        throw new Error(`Unknown tool: ${name}`);
+    }
+  } catch (error) {
+    return {
+      content: [{ type: "text", text: `Error: ${error.message}` }],
+      isError: true,
+    };
+  }
+});
+
+const transport = new StdioServerTransport();
+await server.connect(transport);