Enhanced MCP to work with both direct parameters and Smithery config

Author: rashidazarang

index.js

@@ -8,10 +8,12 @@ const { spawn } = require('child_process');
 const args = process.argv.slice(2);
 let tokenIndex = args.indexOf('--token');
 let baseIndex = args.indexOf('--base');
+let configIndex = args.indexOf('--config');
 
-// Extract token and base ID
+// Extract token, base ID and config
 const token = tokenIndex !== -1 && tokenIndex + 1 < args.length ? args[tokenIndex + 1] : null;
 const baseId = baseIndex !== -1 && baseIndex + 1 < args.length ? args[baseIndex + 1] : null;
+const config = configIndex !== -1 && configIndex + 1 < args.length ? args[configIndex + 1] : null;
 
 console.log('🔌 Airtable MCP - Connecting your AI to Airtable');
 console.log('-----------------------------------------------');
@@ -76,18 +78,33 @@ if (token) {
 if (baseId) {
   scriptArgs.push('--base', baseId);
 }
-
-console.log(`🚀 Starting Airtable MCP Server using ${pythonPath}`);
-if (token) {
-  console.log('✅ Using provided API token');
+if (config) {
+  scriptArgs.push('--config', config);
+  
+  // Try to extract and log info from config
+  try {
+    const configObj = JSON.parse(config);
+    if (configObj.airtable_token) {
+      console.log('✅ Using API token from config');
+    }
+    if (configObj.base_id) {
+      console.log(`✅ Using base ID from config: ${configObj.base_id}`);
+    }
+  } catch (e) {
+    console.warn('⚠️ Could not parse config JSON, passing it directly to Python script');
+  }
 } else {
-  console.log('⚠️ No API token provided, will try to use .env file');
-}
+  if (token) {
+    console.log('✅ Using provided API token');
+  } else {
+    console.log('⚠️ No API token provided, will try to use .env file');
+  }
 
-if (baseId) {
-  console.log(`✅ Using base ID: ${baseId}`);
-} else {
-  console.log('ℹ️ No base ID provided, will need to set one later');
+  if (baseId) {
+    console.log(`✅ Using base ID: ${baseId}`);
+  } else {
+    console.log('ℹ️ No base ID provided, will need to set one later');
+  }
 }
 
 // Execute the Python script

inspector_server.py

@@ -23,6 +23,7 @@ def parse_args():
     parser = argparse.ArgumentParser(description="Airtable MCP Server")
     parser.add_argument("--token", dest="api_token", help="Airtable Personal Access Token")
     parser.add_argument("--base", dest="base_id", help="Airtable Base ID")
+    parser.add_argument("--config", dest="config_json", help="Configuration as JSON (for Smithery integration)")
     return parser.parse_args()
 
 # Set up logging
@@ -32,28 +33,45 @@ def parse_args():
 # Parse arguments
 args = parse_args()
 
+# Handle config JSON from Smithery if provided
+config = {}
+if args.config_json:
+    try:
+        # Strip any trailing quotes or backslashes that might be present
+        config_str = args.config_json.rstrip('\\"')
+        logger.info(f"Parsing config: {config_str}")
+        config = json.loads(config_str)
+        logger.info(f"Successfully parsed config: {config}")
+    except json.JSONDecodeError as e:
+        logger.error(f"Failed to parse config JSON: {e}")
+        logger.error(f"Raw config string: {args.config_json}")
+
 # Create MCP server
 app = FastMCP("Airtable Tools")
 
-# Get token from arguments or environment
-token = args.api_token or os.environ.get("AIRTABLE_PERSONAL_ACCESS_TOKEN", "")
-base_id = args.base_id or os.environ.get("AIRTABLE_BASE_ID", "")
+# Get token from arguments, config, or environment
+token = args.api_token or config.get("airtable_token", "") or os.environ.get("AIRTABLE_PERSONAL_ACCESS_TOKEN", "")
+# Clean up token if it has trailing quote
+if token and token.endswith('"'):
+    token = token[:-1]
+    
+base_id = args.base_id or config.get("base_id", "") or os.environ.get("AIRTABLE_BASE_ID", "")
 
 if not token:
-    logger.warning("No Airtable API token provided. Use --token or set AIRTABLE_PERSONAL_ACCESS_TOKEN environment variable.")
+    logger.warning("No Airtable API token provided. Use --token, --config, or set AIRTABLE_PERSONAL_ACCESS_TOKEN environment variable.")
 else:
     logger.info(f"Using Airtable token: {token[:5]}...{token[-5:]}")
 
 if base_id:
     logger.info(f"Using base ID: {base_id}")
 else:
-    logger.warning("No base ID provided. Use --base or set AIRTABLE_BASE_ID environment variable.")
+    logger.warning("No base ID provided. Use --base, --config, or set AIRTABLE_BASE_ID environment variable.")
 
 # Helper functions for Airtable API calls
 async def api_call(endpoint, method="GET", data=None, params=None):
     """Make an Airtable API call"""
     if not token:
-        return {"error": "No Airtable API token provided. Use --token or set AIRTABLE_PERSONAL_ACCESS_TOKEN environment variable."}
+        return {"error": "No Airtable API token provided. Use --token, --config, or set AIRTABLE_PERSONAL_ACCESS_TOKEN environment variable."}
 
     headers = {
         "Authorization": f"Bearer {token}",

smithery.yaml

@@ -17,13 +17,11 @@ startCommand:
     required: ["airtable_token"]
   commandFunction: |
     (config) => {
-      // Use Python directly with the new inspector_server.py
+      // Pass config as a JSON string to the inspector_server.py
+      const configStr = JSON.stringify(config);
       return {
         command: "python3.10",
-        args: ["inspector_server.py", 
-              "--token", config.airtable_token,
-              ...(config.base_id ? ["--base", config.base_id] : [])
-        ],
+        args: ["inspector_server.py", "--config", configStr],
         env: {}
       };
     }