@aithr-ai/mcp-server 1.1.4 → 1.1.6

package/index.js

@@ -212,7 +212,7 @@ class MCPServer {
       },
       {
         name: 'aether_trigger_generation',
-        description: 'Trigger AI generation for an agent on the canvas. For orchestra mode, pass the orchestraContext from aether_orchestra_next to inject full context including upstream artifacts.',
+        description: 'Trigger AI generation for an agent on the canvas. For orchestra mode, pass the orchestraContext from aether_orchestra_next to inject full context including upstream artifacts. If generation times out (>300s), returns autoSplitAdvice with guidance on breaking the task into smaller pieces.',
         inputSchema: {
           type: 'object',
           properties: {
@@ -228,6 +228,10 @@ class MCPServer {
               type: 'object',
               description: 'Orchestra context from aether_orchestra_next - includes sessionId, taskId, phase info, upstream artifacts, and plan excerpt. Pass this directly from the task object returned by orchestra_next.',
             },
+            autoSplit: {
+              type: 'boolean',
+              description: 'If true (default), returns guidance on splitting task when timeout occurs. Set to false to get raw error.',
+            },
           },
           required: ['nodeId'],
         },
@@ -1453,7 +1457,7 @@ class MCPServer {
   }
 
   async triggerGeneration(args) {
-    const { nodeId, context, orchestraContext } = args;
+    const { nodeId, context, orchestraContext, autoSplit = true } = args;
     const projectId = config.projectId;
 
     if (!projectId) {
@@ -1501,7 +1505,36 @@ class MCPServer {
           : undefined,
       };
     } catch (e) {
-      return { success: false, error: e.message };
+      const errorMsg = e.message || '';
+      const isTimeout = errorMsg.includes('timed out') ||
+                        errorMsg.includes('timeout') ||
+                        errorMsg.includes('FUNCTION_INVOCATION_TIMEOUT') ||
+                        errorMsg.includes('504') ||
+                        errorMsg.includes('Task timed out');
+
+      // If timeout and autoSplit enabled, return guidance for splitting
+      if (isTimeout && autoSplit) {
+        return {
+          success: false,
+          error: 'Generation timed out (exceeded 300 second limit)',
+          timeout: true,
+          autoSplitAdvice: {
+            shouldSplit: true,
+            message: 'Task is too large. Split into smaller sub-tasks.',
+            originalContext: context?.substring(0, 500) || 'No context provided',
+            suggestedApproach: [
+              '1. Break the task into 3-5 smaller focused tasks',
+              '2. Create separate agents for each sub-task using aether_add_agent',
+              '3. Run each sub-task individually',
+              '4. Combine results at the end',
+            ],
+            example: 'Instead of "Create all types for the app", do "Create User and Session types", then "Create Transcript and Claim types", etc.',
+          },
+          hint: 'The task was too complex for a single generation. Break it into smaller pieces and try again.',
+        };
+      }
+
+      return { success: false, error: errorMsg };
     }
   }
 
@@ -2909,7 +2942,21 @@ class MCPServer {
     }
 
     const response = await fetch(url, options);
-    const responseData = await response.json();
+
+    // Handle non-JSON responses gracefully
+    let responseData;
+    const responseText = await response.text();
+
+    try {
+      responseData = JSON.parse(responseText);
+    } catch (parseError) {
+      // If response isn't valid JSON, wrap it in an error object
+      if (!response.ok) {
+        throw new Error(`API error ${response.status}: ${responseText.substring(0, 500)}`);
+      }
+      // Successful but non-JSON response (shouldn't happen but handle gracefully)
+      responseData = { rawResponse: responseText };
+    }
 
     if (!response.ok) {
       // Include hints and code in error message for better debugging

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aithr-ai/mcp-server",
-  "version": "1.1.4",
+  "version": "1.1.6",
   "description": "MCP server to connect Claude Code to Aether canvas - AI-powered team workspace for software development",
   "main": "index.js",
   "bin": {