@aithr-ai/mcp-server 1.0.11 → 1.0.12

package/index.js

@@ -483,6 +483,10 @@ class MCPServer {
               type: 'string',
               description: 'Directory to clone the repository into (default: system temp directory)',
             },
+            useGeneratedFiles: {
+              type: 'boolean',
+              description: 'Use files from generatedFiles table (saved by aether_extract_batch). Recommended for large artifact sets (700+ files) to avoid buffer overflow.',
+            },
           },
         },
       },
@@ -1793,6 +1797,7 @@ class MCPServer {
       pushToMain = false, // Push directly to main/master instead of feature branch
       autoClone = true,   // Auto-clone repo if not found locally
       cloneDir,           // Directory to clone into (default: temp dir)
+      useGeneratedFiles = false, // Use files from generatedFiles table (saved by extract_batch)
     } = args;
 
     const projectId = config.projectId;
@@ -1800,30 +1805,76 @@ class MCPServer {
       return { error: 'No project ID configured. Set AETHER_PROJECT_ID environment variable.' };
     }
 
-    const { execSync } = require('child_process');
+    const { execSync, spawnSync } = require('child_process');
     const fs = require('fs');
     const path = require('path');
     const os = require('os');
 
     try {
-      // Export files from backend (uses robust parsing, returns full content)
-      // This also gives us the repository info
-      const exportData = await this.mcpApiCall(
-        `/mcp/projects/${projectId}/push-artifacts`,
-        'POST',
-        {
-          artifactIds,
-          branchName: pushToMain ? undefined : branchName,
-          exportFiles: true, // Get full file contents for local git
+      let files = [];
+      let repoInfo = null;
+
+      if (useGeneratedFiles) {
+        // Use files from generatedFiles table (populated by aether_extract_batch)
+        // This avoids the large response from push-artifacts
+        const generatedData = await this.mcpApiCall(
+          `/mcp/projects/${projectId}/push-artifacts`,
+          'POST',
+          {
+            artifactIds,
+            branchName: pushToMain ? undefined : branchName,
+            useGeneratedFiles: true, // Tell API to use generatedFiles table
+            exportFiles: false, // Don't export file contents in response
+          }
+        );
+
+        repoInfo = generatedData.repository;
+
+        // Fetch files in batches from generatedFiles API
+        let batch = 0;
+        const batchSize = 50;
+        let hasMore = true;
+
+        while (hasMore) {
+          const batchData = await this.mcpApiCall(
+            `/mcp/projects/${projectId}/generated-files?batch=${batch}&batchSize=${batchSize}`,
+            'GET'
+          );
+
+          if (batchData.files && batchData.files.length > 0) {
+            files = files.concat(batchData.files);
+          }
+
+          hasMore = batchData.hasMore || false;
+          batch++;
+
+          // Safety limit
+          if (batch > 100) break;
         }
-      );
+      } else {
+        // Original behavior: Export files from backend (may cause buffer overflow for large sets)
+        const exportData = await this.mcpApiCall(
+          `/mcp/projects/${projectId}/push-artifacts`,
+          'POST',
+          {
+            artifactIds,
+            branchName: pushToMain ? undefined : branchName,
+            exportFiles: true, // Get full file contents for local git
+          }
+        );
+
+        if (exportData.files) {
+          files = exportData.files;
+        }
+        repoInfo = exportData.repository;
+      }
 
-      if (!exportData.files || exportData.files.length === 0) {
-        return { error: 'No files to push', details: exportData };
+      if (!files || files.length === 0) {
+        return { error: 'No files to push. Try running aether_extract_batch first with saveToDatabase=true, then use useGeneratedFiles=true' };
       }
 
-      const files = exportData.files;
-      const repoInfo = exportData.repository;
+      // Limit response size by not including all files in response
+      const filesForResponse = files.slice(0, 20).map(f => f.path);
 
       if (!repoInfo || !repoInfo.fullName) {
         return { error: 'No repository linked to this project. Link a GitHub repository in Project Settings.' };

package/package.json

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