metabase-ai-assistant 3.4.0 → 3.4.2

package/README_MCP.md

@@ -45,7 +45,7 @@ After updating the configuration file, restart your MCP client to load the new s
 
 ---
 
-## Available Tools (107 Total)
+## Available Tools (111 Total)
 
 ### Database Operations
 - **db_list**: List all databases in Metabase
@@ -55,6 +55,10 @@ After updating the configuration file, restart your MCP client to load the new s
 
 ### SQL Operations
 - **sql_execute**: Execute native SQL query
+- **sql_submit**: Asynchronously create a long-running SQL query job
+- **sql_status**: Check the status of a SQL job
+- **sql_cancel**: Cancel a running SQL job
+- **db_table_profile**: Profile a table (smart dim/ref detection)
 - **ai_sql_generate**: Generate SQL from natural language description
 - **ai_sql_optimize**: Optimize SQL query for performance
 - **ai_sql_explain**: Explain what a SQL query does
@@ -176,6 +180,9 @@ After updating the configuration file, restart your MCP client to load the new s
 ## Features Summary
 
 ### Currently Available
+- **Async Query Management** (for long-running queries)
+- **Smart Response Optimization** (no truncation for DDL/definitions)
+- **Table Profiling** (dim/ref table detection)
 - Metabase API integration
 - SQL query execution
 - Question/Dashboard creation
@@ -209,7 +216,34 @@ After updating the configuration file, restart your MCP client to load the new s
 ```json
 {
   "database_id": 1,
-  "sql": "SELECT * FROM table_name LIMIT 10"
+  "sql": "SELECT * FROM table_name LIMIT 10",
+  "full_results": false // Optional: Set true to disable truncation (useful for DDL)
+}
+```
+
+### sql_submit (Async Query)
+```json
+{
+  "database_id": 1,
+  "sql": "SELECT SLEEP(120)",
+  "timeout_seconds": 300
+}
+```
+
+### sql_status
+```json
+{
+  "job_id": "550e8400-e29b-41d4-a716-446655440000"
+}
+```
+
+### db_table_profile
+```json
+{
+  "database_id": 1,
+  "schema": "public",
+  "table": "dim_customers",
+  "sample_rows": 3
 }
 ```
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "metabase-ai-assistant",
-  "version": "3.4.0",
+  "version": "3.4.2",
   "mcpName": "io.github.enessari/metabase-ai-assistant",
   "description": "The most powerful MCP Server for Metabase - 111+ tools for AI-powered SQL generation, dashboard automation, user management & enterprise BI. Works with Claude, Cursor, and any MCP-compatible AI.",
   "main": "src/index.js",

package/src/mcp/server.js

@@ -258,6 +258,10 @@ class MetabaseMCPServer {
                   type: 'string',
                   description: 'SQL query to execute',
                 },
+                full_results: {
+                  type: 'boolean',
+                  description: 'Set to true to disable result truncation (useful for DDL/definitions)',
+                },
               },
               required: ['database_id', 'sql'],
             },
@@ -3237,7 +3241,7 @@ class MetabaseMCPServer {
 
           // SQL execution
           case 'sql_execute':
-            return await this.handleExecuteSQL(args.database_id, args.sql);
+            return await this.handleExecuteSQL(args);
           case 'sql_submit':
             return await this.handleSQLSubmit(args);
           case 'sql_status':
@@ -3667,9 +3671,13 @@ class MetabaseMCPServer {
     };
   }
 
-  async handleExecuteSQL(databaseId, sql) {
+  async handleExecuteSQL(args) {
     await this.ensureInitialized();
 
+    const databaseId = args.database_id;
+    const sql = args.sql;
+    const fullResults = args.full_results === true;
+
     if (this.initError) {
       throw new McpError(ErrorCode.InternalError, `Failed to initialize: ${this.initError.message}`);
     }
@@ -3728,8 +3736,21 @@ class MetabaseMCPServer {
         rows.slice(0, 5).forEach((row) => {
           const formattedRow = row.map(cell => {
             if (cell === null) return 'NULL';
-            if (typeof cell === 'string' && cell.length > 30) {
-              return cell.substring(0, 27) + '...';
+
+            // Smart truncation logic
+            let truncateLimit = 100; // Increased base limit from 30
+
+            // Disable truncation for small result sets (DDL/procedures) or explicit full_results
+            if (fullResults || rows.length <= 2) {
+              truncateLimit = 50000;
+            }
+            // Check specific DDL-related column names
+            else if (columns.some(c => /definition|ddl|source|create_statement|routine_definition/i.test(c.name))) {
+              truncateLimit = 10000;
+            }
+
+            if (typeof cell === 'string' && cell.length > truncateLimit) {
+              return cell.substring(0, truncateLimit - 3) + '...';
             }
             return String(cell);
           });
@@ -3922,7 +3943,13 @@ class MetabaseMCPServer {
       output += `⏱️ Elapsed: ${elapsedSeconds} seconds\\n`;
 
       if (job.status === 'running' || job.status === 'pending') {
-        output += `\\n💡 Query is still running. Check again later or use \`sql_cancel\` to stop.`;
+        let waitSeconds = 3;
+        if (elapsedSeconds > 60) waitSeconds = 30;
+        else if (elapsedSeconds > 30) waitSeconds = 10;
+        else if (elapsedSeconds > 10) waitSeconds = 5;
+
+        output += `\\n💡 Query is still running. Please wait **${waitSeconds} seconds** before checking again.\\n`;
+        output += `(Use \`sql_cancel\` to stop if needed)`;
       } else if (job.status === 'complete') {
         const rows = job.result?.data?.rows || [];
         const columns = job.result?.data?.cols || [];