@getmegabrain/cli 0.1.10 → 0.1.11

package/dist/science/agent/model.js

@@ -4,11 +4,17 @@
  * `ModelClient` so the transport is swappable; in production it's the Gateway
  * (`/api/gateway/v1/messages`) authenticated with the signed-in account token.
  */
+/** Transient HTTP statuses worth retrying: overload (429/529), gateway/upstream (500/502/503/504),
+ *  and request timeout (408). A 503 "temporarily_unavailable" from the Gateway lands here. */
+const RETRYABLE_STATUSES = new Set([408, 429, 500, 502, 503, 504, 529]);
 export class GatewayModel {
     url;
     apiKey;
     model;
     maxTokens;
+    maxRetries;
+    retryBaseMs;
+    sleep;
     constructor(options) {
         this.url =
             options.url ??
@@ -17,30 +23,69 @@ export class GatewayModel {
         this.apiKey = options.apiKey;
         this.model = options.model ?? process.env.MB_SCIENCE_MODEL ?? 'claude-opus-4-8';
         this.maxTokens = options.maxTokens ?? 4096;
+        this.maxRetries = options.maxRetries ?? 4;
+        this.retryBaseMs = options.retryBaseMs ?? 800;
+        this.sleep = options.sleep ?? (ms => new Promise(resolve => setTimeout(resolve, ms)));
+    }
+    /** Backoff for a given attempt (0-indexed): exponential + jitter, honoring Retry-After. */
+    backoffMs(attempt, retryAfter) {
+        const headerSeconds = retryAfter ? Number(retryAfter) : NaN;
+        if (Number.isFinite(headerSeconds) && headerSeconds >= 0) {
+            return Math.min(headerSeconds * 1000, 30_000);
+        }
+        const base = this.retryBaseMs * 2 ** attempt;
+        return Math.min(base, 15_000) + Math.floor(Math.random() * 250);
     }
     async complete(request) {
         if (!this.apiKey)
             throw new Error('Sign in to MegaBrain to use the agent.');
-        const response = await fetch(this.url, {
-            method: 'POST',
-            headers: {
-                'content-type': 'application/json',
-                authorization: `Bearer ${this.apiKey}`,
-                'anthropic-version': '2023-06-01',
-            },
-            body: JSON.stringify({
-                model: this.model,
-                max_tokens: this.maxTokens,
-                system: request.system,
-                messages: request.messages,
-                tools: request.tools,
-            }),
+        const body = JSON.stringify({
+            model: this.model,
+            max_tokens: this.maxTokens,
+            system: request.system,
+            messages: request.messages,
+            tools: request.tools,
         });
-        if (!response.ok) {
-            const detail = await response.text();
-            throw new Error(`Gateway request failed (${response.status}): ${detail.slice(0, 500)}`);
+        let lastError = '';
+        for (let attempt = 0; attempt <= this.maxRetries; attempt += 1) {
+            let response;
+            try {
+                response = await fetch(this.url, {
+                    method: 'POST',
+                    headers: {
+                        'content-type': 'application/json',
+                        authorization: `Bearer ${this.apiKey}`,
+                        'anthropic-version': '2023-06-01',
+                    },
+                    body,
+                });
+            }
+            catch (error) {
+                // Network-level failure (DNS, connection reset) — treat as transient.
+                lastError = `network error: ${error instanceof Error ? error.message : String(error)}`;
+                if (attempt < this.maxRetries) {
+                    await this.sleep(this.backoffMs(attempt));
+                    continue;
+                }
+                throw new Error(`Gateway request failed (${lastError}) after ${attempt + 1} attempts`);
+            }
+            if (response.ok) {
+                const data = (await response.json());
+                return { content: data.content ?? [], stopReason: data.stop_reason ?? 'end_turn' };
+            }
+            const detail = (await response.text()).slice(0, 500);
+            if (RETRYABLE_STATUSES.has(response.status)) {
+                lastError = `${response.status}: ${detail}`;
+                if (attempt < this.maxRetries) {
+                    await this.sleep(this.backoffMs(attempt, response.headers.get('retry-after')));
+                    continue;
+                }
+                // Transient, but out of retries.
+                throw new Error(`Gateway request failed (${lastError}) after ${attempt + 1} attempts`);
+            }
+            throw new Error(`Gateway request failed (${response.status}): ${detail}`);
         }
-        const data = (await response.json());
-        return { content: data.content ?? [], stopReason: data.stop_reason ?? 'end_turn' };
+        // Unreachable: the loop always returns or throws. Present for exhaustiveness.
+        throw new Error(`Gateway request failed (${lastError})`);
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@getmegabrain/cli",
-  "version": "0.1.10",
+  "version": "0.1.11",
   "description": "MegaBrain CLI — sign in once, then code from your terminal via OpenCode wired to the MegaBrain Gateway, or run `megabrain science` for the local research workbench.",
   "license": "MIT",
   "type": "module",