@aithr-ai/mcp-server 1.0.4 → 1.0.5

package/index.js

@@ -1213,77 +1213,41 @@ class MCPServer {
       return { error: 'No project ID configured. Set AETHER_PROJECT_ID environment variable.' };
     }
 
-    try {
-      // The generate endpoint returns SSE (Server-Sent Events), so we need to handle it specially
-      const url = `${config.aetherUrl}/api/workspaces/${config.workspaceSlug}/projects/${projectId}/agents/${nodeId}/generate`;
-      const headers = {
-        'Content-Type': 'application/json',
-      };
-
-      if (config.apiKey) {
-        headers['Authorization'] = `Bearer ${config.apiKey}`;
-      }
-
-      const response = await fetch(url, {
-        method: 'POST',
-        headers,
-        body: JSON.stringify({ context }),
-      });
-
-      if (!response.ok) {
-        const errorText = await response.text();
-        throw new Error(`API error: ${response.status} - ${errorText}`);
-      }
+    if (!nodeId) {
+      return { error: 'nodeId is required' };
+    }
 
-      // Read SSE stream and collect events
-      const reader = response.body.getReader();
-      const decoder = new TextDecoder();
-      let buffer = '';
-      let lastEvent = null;
-      let artifactId = null;
-      let sessionId = null;
-      let error = null;
-
-      while (true) {
-        const { done, value } = await reader.read();
-        if (done) break;
-
-        buffer += decoder.decode(value, { stream: true });
-
-        // Parse SSE events from buffer
-        const lines = buffer.split('\n');
-        buffer = lines.pop() || ''; // Keep incomplete line in buffer
-
-        for (const line of lines) {
-          if (line.startsWith('event: ')) {
-            lastEvent = line.slice(7).trim();
-          } else if (line.startsWith('data: ')) {
-            try {
-              const data = JSON.parse(line.slice(6));
-
-              if (lastEvent === 'generation_complete') {
-                artifactId = data.artifactId;
-                sessionId = data.sessionId;
-              } else if (lastEvent === 'error') {
-                error = data.error || 'Generation failed';
-              }
-            } catch {
-              // Ignore JSON parse errors for partial data
-            }
-          }
+    try {
+      // Use MCP endpoint with API key auth (non-streaming, waits for completion)
+      const result = await this.mcpApiCall(
+        `/mcp/projects/${projectId}/generate`,
+        'POST',
+        {
+          nodeId,
+          context: context || undefined,
+          previewLength: 500,
         }
-      }
-
-      if (error) {
-        return { success: false, error };
-      }
+      );
 
       return {
         success: true,
         message: 'Generation completed',
-        artifactId,
-        sessionId,
-        hint: artifactId
+        node: result.node,
+        generation: {
+          status: result.generation?.status,
+          preview: result.generation?.preview,
+          hasMore: result.generation?.hasMore,
+          duration: result.generation?.duration,
+        },
+        artifact: result.artifact
+          ? {
+              id: result.artifact.id,
+              title: result.artifact.title,
+              type: result.artifact.type,
+            }
+          : null,
+        usage: result.usage,
+        hint: result.artifact?.id
           ? 'Use aether_extract_files to extract code from the artifact, then aether_push_artifacts to push to GitHub.'
           : undefined,
       };

package/package.json

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