@kolbo/mcp 1.12.0 → 1.12.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kolbo/mcp",
-  "version": "1.12.0",
+  "version": "1.12.1",
   "description": "Kolbo AI MCP Server - Generate images, videos, music, speech, and sound effects from Claude Code",
   "main": "src/index.js",
   "bin": {

package/src/polling.js

@@ -25,6 +25,25 @@ class GenerationFailedError extends Error {
   }
 }
 
+// HTTP status codes / fetch failure modes we treat as transient when
+// polling. The job is almost always still running on the server (or about
+// to come back up) — bailing out makes the agent give up on a generation
+// that completes 2 seconds later.
+const TRANSIENT_STATUS_CODES = new Set([0, 408, 425, 429, 500, 502, 503, 504, 522, 524]);
+
+function isTransientPollError(err) {
+  if (!err) return false;
+  // fetch() rejects with a TypeError on network failure (ECONNREFUSED,
+  // ECONNRESET, DNS failure, kolbo-api restart, etc.) — no `status` on
+  // the error object.
+  if (err.name === 'TypeError') return true;
+  if (err.code === 'ECONNRESET' || err.code === 'ECONNREFUSED' || err.code === 'ETIMEDOUT' || err.code === 'EPIPE') return true;
+  // KolboApiError tags status on the options object.
+  const status = err.status ?? err.options?.status;
+  if (typeof status === 'number' && TRANSIENT_STATUS_CODES.has(status)) return true;
+  return false;
+}
+
 async function pollUntilDone(client, generationId, options = {}) {
   const {
     interval = 5000,
@@ -34,13 +53,36 @@ async function pollUntilDone(client, generationId, options = {}) {
 
   const startTime = Date.now();
   const url = statusUrl || `/v1/generate/${encodeURIComponent(generationId)}/status`;
+  let transientFailures = 0;
 
   while (true) {
     if (Date.now() - startTime > timeout) {
       throw new PollingTimeoutError(generationId, timeout);
     }
 
-    const result = await client.get(url);
+    let result;
+    try {
+      result = await client.get(url);
+      transientFailures = 0; // reset on successful poll
+    } catch (err) {
+      // kolbo-api restart, transient network blip, rate-limit, 5xx —
+      // the job is still alive on the server (or coming back). Don't
+      // abandon the polling loop just because one status check failed:
+      // wait a bit longer and try again. After ~30 consecutive failures
+      // (~2.5 minutes at the 5s base interval) we still surface the
+      // last error so a truly dead backend doesn't loop forever.
+      if (isTransientPollError(err)) {
+        transientFailures++;
+        if (transientFailures > 30) {
+          throw err;
+        }
+        const backoff = Math.min(interval * Math.pow(1.5, Math.min(transientFailures - 1, 5)), 30000);
+        await new Promise((resolve) => setTimeout(resolve, backoff));
+        continue;
+      }
+      // Non-transient (auth, 4xx other than 408/425/429) — bubble up.
+      throw err;
+    }
 
     if (result.state === 'completed') {
       return result;