agentmail-mcp 0.1.5 → 0.1.7

package/README.md

@@ -23,3 +23,23 @@ Get your API key from [AgentMail](https://agentmail.to)
     }
 }
 ```
+
+## Tool Selection
+
+By default, all available tools are loaded. You can selectively enable specific tools using the `--tools` argument with a comma-separated list of tool names.
+
+### Example
+
+```json
+{
+    "mcpServers": {
+        "AgentMail": {
+            "command": "npx",
+            "args": ["-y", "agentmail-mcp", "--tools", "get_message,send_message,reply_to_message"],
+            "env": {
+                "AGENTMAIL_API_KEY": "YOUR_API_KEY"
+            }
+        }
+    }
+}
+```

package/dist/index.js

@@ -1,15 +1,44 @@
 #!/usr/bin/env node
+import { AgentMailClient, AgentMailEnvironment } from 'agentmail';
+import { AgentMailToolkit } 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';
-async function main() {
+const 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());
+};
+const parseAgentMailEnv = () => {
+    const env = process.env.AGENTMAIL_ENVIRONMENT?.toLowerCase();
+    switch (env) {
+        case 'development':
+        case 'dev':
+            return AgentMailEnvironment.Development;
+        case 'production':
+        case 'prod':
+        default:
+            return AgentMailEnvironment.Production;
+    }
+};
+const main = async () => {
+    const environment = parseAgentMailEnv();
+    const toolNames = parseToolsArg();
+    const client = new AgentMailClient({ environment });
+    const toolkit = new AgentMailToolkit(client);
     const server = new McpServer({ name: 'AgentMail', version: '0.1.0' });
     const transport = new StdioServerTransport();
-    for (const tool of new AgentMailToolkit().getTools()) {
+    for (const tool of toolkit.getTools(toolNames)) {
         server.tool(tool.name, tool.description, tool.paramsSchema, tool.callback);
     }
     await server.connect(transport);
-}
+};
 main().catch((error) => {
     console.error(error);
     process.exit(1);

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "agentmail-mcp",
-    "version": "0.1.5",
+    "version": "0.1.7",
     "description": "AgentMail MCP Server",
     "type": "module",
     "bin": {
@@ -35,6 +35,7 @@
     },
     "dependencies": {
         "@modelcontextprotocol/sdk": "^1.12.1",
-        "agentmail-toolkit": "^0.1.23"
+        "agentmail": "^0.0.34",
+        "agentmail-toolkit": "^0.1.24"
     }
 }