@ateam-ai/mcp 0.3.0 → 0.3.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ateam-ai/mcp",
-  "version": "0.3.0",
+  "version": "0.3.2",
   "mcpName": "io.github.ariekogan/ateam-mcp",
   "description": "A-Team MCP Server — build, validate, and deploy multi-agent solutions from any AI environment",
   "type": "module",

package/src/api.js

@@ -353,50 +353,74 @@ function formatError(method, path, status, body) {
  */
 async function request(method, path, body, sessionId, opts = {}) {
   const timeoutMs = opts.timeoutMs || REQUEST_TIMEOUT_MS;
-  const controller = new AbortController();
-  const timeout = setTimeout(() => controller.abort(), timeoutMs);
+  const maxRetries = opts.retries ?? 0;
   const baseUrl = getBaseUrl(sessionId);
 
-  try {
-    const fetchOpts = {
-      method,
-      headers: headers(sessionId),
-      signal: controller.signal,
-    };
-    if (body !== undefined) {
-      fetchOpts.body = JSON.stringify(body);
-    }
-
-    const res = await fetch(`${baseUrl}${path}`, fetchOpts);
-
-    if (!res.ok) {
-      const text = await res.text().catch(() => "");
-      throw new Error(formatError(method, path, res.status, text));
-    }
-
-    return res.json();
-  } catch (err) {
-    if (err.name === "AbortError") {
-      throw new Error(
-        `A-Team API timeout: ${method} ${path} did not respond within ${timeoutMs / 1000}s.\n` +
-        `Hint: The A-Team API at ${baseUrl} may be down. Check ${baseUrl}/health`
-      );
-    }
-    if (err.cause?.code === "ECONNREFUSED") {
-      throw new Error(
-        `Cannot connect to A-Team API at ${baseUrl}.\n` +
-        `Hint: The service may be down. Check ${baseUrl}/health`
-      );
+  for (let attempt = 0; attempt <= maxRetries; attempt++) {
+    const controller = new AbortController();
+    const timeout = setTimeout(() => controller.abort(), timeoutMs);
+
+    try {
+      const fetchOpts = {
+        method,
+        headers: headers(sessionId),
+        signal: controller.signal,
+      };
+      if (body !== undefined) {
+        fetchOpts.body = JSON.stringify(body);
+      }
+
+      const res = await fetch(`${baseUrl}${path}`, fetchOpts);
+
+      // Auto-retry on 502/504 (proxy timeout during long deploys)
+      if ((res.status === 502 || res.status === 504) && attempt < maxRetries) {
+        const wait = Math.min(5000 * (attempt + 1), 15000);
+        console.error(`[MCP] ${method} ${path} returned ${res.status}, retrying in ${wait / 1000}s (attempt ${attempt + 1}/${maxRetries})...`);
+        await new Promise(r => setTimeout(r, wait));
+        continue;
+      }
+
+      if (!res.ok) {
+        const text = await res.text().catch(() => "");
+        throw new Error(formatError(method, path, res.status, text));
+      }
+
+      return res.json();
+    } catch (err) {
+      if (err.name === "AbortError") {
+        if (attempt < maxRetries) {
+          const wait = Math.min(5000 * (attempt + 1), 15000);
+          console.error(`[MCP] ${method} ${path} timed out, retrying in ${wait / 1000}s (attempt ${attempt + 1}/${maxRetries})...`);
+          await new Promise(r => setTimeout(r, wait));
+          continue;
+        }
+        throw new Error(
+          `A-Team API timeout: ${method} ${path} did not respond within ${timeoutMs / 1000}s.\n` +
+          `Hint: The A-Team API at ${baseUrl} may be down. Check ${baseUrl}/health`
+        );
+      }
+      if (err.cause?.code === "ECONNREFUSED") {
+        if (attempt < maxRetries) {
+          const wait = Math.min(5000 * (attempt + 1), 15000);
+          console.error(`[MCP] ${method} ${path} connection refused, retrying in ${wait / 1000}s (attempt ${attempt + 1}/${maxRetries})...`);
+          await new Promise(r => setTimeout(r, wait));
+          continue;
+        }
+        throw new Error(
+          `Cannot connect to A-Team API at ${baseUrl}.\n` +
+          `Hint: The service may be down. Check ${baseUrl}/health`
+        );
+      }
+      if (err.cause?.code === "ENOTFOUND") {
+        throw new Error(
+          `Cannot resolve A-Team API host: ${baseUrl}.\n` +
+          `Hint: Check your internet connection and ADAS_API_URL setting.`
+        );
+      }
+      throw err;
+    } finally {
+      clearTimeout(timeout);
     }
-    if (err.cause?.code === "ENOTFOUND") {
-      throw new Error(
-        `Cannot resolve A-Team API host: ${baseUrl}.\n` +
-        `Hint: Check your internet connection and ADAS_API_URL setting.`
-      );
-    }
-    throw err;
-  } finally {
-    clearTimeout(timeout);
   }
 }
 

package/src/tools.js

@@ -367,6 +367,46 @@ export const tools = [
     },
   },
 
+  {
+    name: "ateam_upload_connector",
+    core: true,
+    description:
+      "Upload connector code to Core and restart — WITHOUT redeploying skills. " +
+      "Use this to update connector source code (server.js, UI assets, plugins) quickly. " +
+      "Set github=true to pull files from the solution's GitHub repo, or pass files directly. " +
+      "Much faster than ateam_build_and_run for connector-only changes.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        solution_id: {
+          type: "string",
+          description: "The solution ID",
+        },
+        connector_id: {
+          type: "string",
+          description: "The connector ID to upload (e.g. 'personal-assistant-ui-mcp')",
+        },
+        github: {
+          type: "boolean",
+          description: "If true, pull connector files from GitHub repo. Default: false.",
+        },
+        files: {
+          type: "array",
+          items: {
+            type: "object",
+            properties: {
+              path: { type: "string", description: "Relative file path (e.g. 'server.js', 'ui-dist/panel/1.0.0/index.html')" },
+              content: { type: "string", description: "File content" },
+            },
+            required: ["path", "content"],
+          },
+          description: "Files to upload. Alternative to github=true.",
+        },
+      },
+      required: ["solution_id", "connector_id"],
+    },
+  },
+
   // ═══════════════════════════════════════════════════════════════════
   // ADVANCED TOOLS — hidden from tools/list, still callable by name
   // Use these for manual lifecycle control, debugging, and diagnostics
@@ -987,6 +1027,7 @@ const TENANT_TOOLS = new Set([
   "ateam_redeploy",
   "ateam_delete_solution",
   "ateam_delete_connector",
+  "ateam_upload_connector",
   "ateam_solution_chat",
   // Read operations (tenant-specific data)
   "ateam_list_solutions",
@@ -1268,7 +1309,7 @@ const handlers = {
     // Phase 2: Deploy
     let deploy;
     try {
-      deploy = await post("/deploy/solution", { solution, skills, connectors, mcp_store: effectiveMcpStore }, sid, { timeoutMs: 300_000 });
+      deploy = await post("/deploy/solution", { solution, skills, connectors, mcp_store: effectiveMcpStore }, sid, { timeoutMs: 300_000, retries: 2 });
       phases.push({ phase: "deploy", status: deploy.ok ? "done" : "failed" });
     } catch (err) {
       return {
@@ -1558,7 +1599,7 @@ const handlers = {
     post(`/deploy/solutions/${solution_id}/github/push`, { message }, sid, { timeoutMs: 60_000 }),
 
   ateam_github_pull: async ({ solution_id }, sid) =>
-    post(`/deploy/solutions/${solution_id}/github/pull`, {}, sid, { timeoutMs: 300_000 }),
+    post(`/deploy/solutions/${solution_id}/github/pull`, {}, sid, { timeoutMs: 300_000, retries: 2 }),
 
   ateam_github_status: async ({ solution_id }, sid) =>
     get(`/deploy/solutions/${solution_id}/github/status`, sid),
@@ -1589,11 +1630,19 @@ const handlers = {
   ateam_delete_connector: async ({ solution_id, connector_id }, sid) =>
     del(`/deploy/solutions/${solution_id}/connectors/${connector_id}`, sid),
 
+  ateam_upload_connector: async ({ solution_id, connector_id, github, files }, sid) =>
+    post(
+      `/deploy/solutions/${solution_id}/connectors/${connector_id}/upload`,
+      { github, files },
+      sid,
+      { timeoutMs: 300_000, retries: 1 },
+    ),
+
   ateam_redeploy: async ({ solution_id, skill_id }, sid) => {
     const endpoint = skill_id
       ? `/deploy/solutions/${solution_id}/skills/${skill_id}/redeploy`
       : `/deploy/solutions/${solution_id}/redeploy`;
-    const result = await post(endpoint, {}, sid, { timeoutMs: 300_000 });
+    const result = await post(endpoint, {}, sid, { timeoutMs: 300_000, retries: 2 });
     return {
       ok: result.ok,
       solution_id,