fix prediction terminal status handling

Author: chengzeyi

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "wavespeed",
-  "version": "0.3.1",
+  "version": "0.3.2",
   "description": "WaveSpeed Client SDK for Wavespeed API",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package.json

@@ -20,7 +20,7 @@ const RETRYABLE_STATUS_CODES = new Set([429, 500, 502, 503, 504]);
  * Statuses that mean the task has reached a terminal, non-successful state
  * and polling must stop.
  */
-const TERMINAL_FAILURE_STATUSES = new Set(['failed', 'cancelled', 'timeout']);
+const TERMINAL_FAILURE_STATUSES = new Set(['failed', 'cancelled', 'timeout', 'deleted']);
 
 /**
  * Minimal filename-extension to MIME type mapping used to populate the
@@ -554,7 +554,7 @@ export class Client {
       }
 
       if (TERMINAL_FAILURE_STATUSES.has(status)) {
-        // cancelled and timeout are terminal too: keep polling and the task
+        // cancelled, timeout, and deleted are terminal too: keep polling and the task
         // will never complete, so surface the API's error text and stop.
         const error = data.error || `Task ${status}`;
         throw new Error(`Prediction failed (task_id: ${requestId}): ${error}`);
@@ -598,10 +598,7 @@ export class Client {
   }
 
   private _resultUrlFromData(data: Record<string, any>): string | undefined {
-    const urls = data.urls;
-    return urls && typeof urls === 'object' && typeof urls.get === 'string'
-      ? urls.get
-      : undefined;
+    return data.id ? `${this.baseUrl}/api/v3/predictions/${data.id}/result` : undefined;
   }
 
   private _isSyncTimeoutData(data: Record<string, any>): boolean {

src/api/client.ts

@@ -241,6 +241,32 @@ describe('Client', () => {
     expect(global.fetch).toHaveBeenCalledTimes(2);
   });
 
+  test('run treats deleted status as terminal failure', async () => {
+    const mockSubmitResponse = {
+      ok: true,
+      status: 200,
+      json: async () => ({ data: { id: 'req-123' } }),
+    };
+    const mockGetResponse = {
+      ok: true,
+      status: 200,
+      json: async () => ({
+        data: { status: 'deleted', error: 'Task was deleted' }
+      }),
+    };
+
+    (global.fetch as jest.Mock)
+      .mockResolvedValueOnce(mockSubmitResponse)
+      .mockResolvedValueOnce(mockGetResponse);
+
+    const client = new Client('test-key');
+
+    await expect(
+      client.run('wavespeed-ai/z-image/turbo', { prompt: 'test' })
+    ).rejects.toThrow('Task was deleted');
+    expect(global.fetch).toHaveBeenCalledTimes(2);
+  });
+
   test('run timeout', async () => {
     const mockSubmitResponse = {
       ok: true,
@@ -324,7 +350,6 @@ describe('Client', () => {
           status: 'processing',
           code: 5004,
           error: 'Sync mode timed out after 90 seconds. The prediction is still processing asynchronously.',
-          urls: { get: resultUrl }
         }
       }),
     };
@@ -353,7 +378,6 @@ describe('Client', () => {
           status: 'processing',
           code: 5004,
           error: 'Sync mode timed out after 90 seconds. The prediction is still processing asynchronously.',
-          urls: { get: resultUrl }
         }
       }),
     };