agentmail-mcp 0.1.4 → 0.1.6

package/dist/index.js

@@ -1,9 +1,25 @@
 #!/usr/bin/env node
-import { AgentMailMcpServer } from 'agentmail-toolkit/mcp';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { AgentMailToolkit } from 'agentmail-toolkit/mcp';
+function parseToolsArg() {
+    const args = process.argv.slice(2);
+    const toolsIndex = args.indexOf('--tools');
+    if (toolsIndex === -1)
+        return undefined;
+    const toolsArg = args[toolsIndex + 1];
+    if (!toolsArg) {
+        console.error('Error: --tools flag requires a comma-separated list of tool names');
+        process.exit(1);
+    }
+    return toolsArg.split(',').map((tool) => tool.trim());
+}
 async function main() {
-    const server = new AgentMailMcpServer();
+    const server = new McpServer({ name: 'AgentMail', version: '0.1.0' });
     const transport = new StdioServerTransport();
+    for (const tool of new AgentMailToolkit().getTools(parseToolsArg())) {
+        server.tool(tool.name, tool.description, tool.paramsSchema, tool.callback);
+    }
     await server.connect(transport);
 }
 main().catch((error) => {

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "agentmail-mcp",
-    "version": "0.1.4",
+    "version": "0.1.6",
     "description": "AgentMail MCP Server",
     "type": "module",
     "bin": {
@@ -35,6 +35,6 @@
     },
     "dependencies": {
         "@modelcontextprotocol/sdk": "^1.12.1",
-        "agentmail-toolkit": "^0.1.21"
+        "agentmail-toolkit": "^0.1.24"
     }
 }