@wavespeed/cli 0.4.10 → 0.4.11

package/dist/commands/auth.js

@@ -6,6 +6,7 @@ import clipboard from 'clipboardy';
 import { userConfig, getBaseUrl, getApiKey, loadProjectConfig, PROJECT_FILE } from '../lib/config.js';
 import { LINKS } from '../lib/links.js';
 import { clientAttributionHeaders } from '../lib/client-headers.js';
+import { GET_TIMEOUT_MS } from '../lib/api.js';
 const KEY_URL = 'https://wavespeed.ai/accesskey';
 function mask(key) {
     if (key.length < 12)
@@ -42,6 +43,7 @@ async function validateKey(apiKey, baseUrl) {
     try {
         res = await fetch(`${baseUrl}/api/v3/balance`, {
             headers: { Authorization: `Bearer ${apiKey}`, ...clientAttributionHeaders() },
+            signal: AbortSignal.timeout(GET_TIMEOUT_MS),
         });
     }
     catch (err) {

package/dist/lib/api.js

@@ -46,6 +46,7 @@ export async function fetchModels(opts = {}) {
         throw new Error('No API key configured. Run `wavespeed login`.');
     const res = await fetch(`${baseUrl}/api/v3/models`, {
         headers: { Authorization: `Bearer ${apiKey}`, ...clientAttributionHeaders() },
+        signal: AbortSignal.timeout(GET_TIMEOUT_MS),
     });
     if (!res.ok) {
         throw await httpError(res, 'GET', '/api/v3/models');
@@ -66,6 +67,12 @@ export function clearModelsCache() {
         /* nothing to clear */
     }
 }
+// Per-request timeouts (no overall cap). A hung socket would otherwise stall
+// the CLI forever. Read-only GETs are far below 30s in practice (result p99.9
+// < 1s); the async submit gets more headroom because a timeout there is an
+// ambiguous outcome, not a clean failure.
+export const GET_TIMEOUT_MS = 30_000;
+export const SUBMIT_TIMEOUT_MS = 60_000;
 // --- Account + prediction endpoints (not exposed by the wavespeed SDK) ---
 function authHeaders() {
     const apiKey = getApiKey();
@@ -94,10 +101,10 @@ async function httpError(res, method, path) {
     err.status = res.status;
     return err;
 }
-async function apiGet(path, opts = {}) {
+async function apiGet(path) {
     const res = await fetch(`${getBaseUrl()}${path}`, {
         headers: authHeaders(),
-        signal: opts.timeoutMs ? AbortSignal.timeout(opts.timeoutMs) : undefined,
+        signal: AbortSignal.timeout(GET_TIMEOUT_MS),
     });
     if (!res.ok)
         throw await httpError(res, 'GET', path);
@@ -106,11 +113,12 @@ async function apiGet(path, opts = {}) {
         throw new Error(json.message || `API returned code ${json.code}`);
     return json.data;
 }
-async function apiPost(path, body) {
+async function apiPost(path, body, opts = {}) {
     const res = await fetch(`${getBaseUrl()}${path}`, {
         method: 'POST',
         headers: { ...authHeaders(), 'Content-Type': 'application/json' },
         body: JSON.stringify(body),
+        signal: opts.timeoutMs ? AbortSignal.timeout(opts.timeoutMs) : undefined,
     });
     if (!res.ok)
         throw await httpError(res, 'POST', path);
@@ -142,13 +150,8 @@ export async function fetchHistory(query = {}) {
         body.status = query.status;
     return apiPost('/api/v3/predictions', body);
 }
-// A hung socket would otherwise stall the CLI forever. 30s is far above a
-// healthy status read, so it only trips on a dead connection.
-const POLL_REQUEST_TIMEOUT_MS = 30_000;
 export async function fetchPrediction(id) {
-    return apiGet(`/api/v3/predictions/${id}/result`, {
-        timeoutMs: POLL_REQUEST_TIMEOUT_MS,
-    });
+    return apiGet(`/api/v3/predictions/${id}/result`);
 }
 // Same set as wavespeed-python. A thrown error with no HTTP status is a
 // network failure or timeout; envelope/JSON errors carry no status either but
@@ -200,7 +203,20 @@ export async function submitPrediction(model, input, opts = {}) {
     const body = { ...input };
     if (opts.sync)
         body.enable_sync_mode = true;
-    return apiPost(`/api/v3/${model}`, body);
+    // sync mode legitimately holds the request until the task finishes, so only
+    // async submits get the per-request timeout.
+    try {
+        return await apiPost(`/api/v3/${model}`, body, {
+            timeoutMs: opts.sync ? undefined : SUBMIT_TIMEOUT_MS,
+        });
+    }
+    catch (err) {
+        // Never retried: the server may already have created (and billed) the task.
+        if (err?.name === 'TimeoutError') {
+            throw new Error(`Submission outcome unknown: no response within ${SUBMIT_TIMEOUT_MS / 1000}s. The task may have been created — run \`wavespeed history\` to check before retrying.`);
+        }
+        throw err;
+    }
 }
 /**
  * Poll a prediction until it reaches a terminal status. `onTick` fires each

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wavespeed/cli",
-  "version": "0.4.10",
+  "version": "0.4.11",
   "description": "WaveSpeed CLI — one command to run any WaveSpeed AI model from your terminal.",
   "publishConfig": {
     "access": "public"
@@ -56,7 +56,7 @@
     "conf": "^13.0.0",
     "open": "^10.1.0",
     "ora": "^8.1.0",
-    "wavespeed": "^0.3.0"
+    "wavespeed": "^0.3.4"
   },
   "devDependencies": {
     "@types/node": "^20.14.0",