wavespeed 0.0.4 → 0.0.5

package/dist/index.d.ts

@@ -17,6 +17,7 @@ export interface PredictionUrls {
 export interface RequestOptions extends RequestInit {
     timeout?: number;
     maxRetries?: number;
+    webhook?: string;
 }
 /**
  * Prediction model representing an image generation job

package/dist/index.js

@@ -90,8 +90,8 @@ class WaveSpeed {
         if (options.baseUrl) {
             this.baseUrl = options.baseUrl;
         }
-        this.pollInterval = options.pollInterval || Number(getEnvVar('WAVESPEED_POLL_INTERVAL')) || 1;
-        this.timeout = options.timeout || Number(getEnvVar('WAVESPEED_TIMEOUT')) || 60;
+        this.pollInterval = options.pollInterval || Number(getEnvVar('WAVESPEED_POLL_INTERVAL')) || 0.5;
+        this.timeout = options.timeout || Number(getEnvVar('WAVESPEED_TIMEOUT')) || 120;
     }
     /**
      * Fetch with timeout support
@@ -198,7 +198,12 @@ class WaveSpeed {
      * @param options Additional fetch options
      */
     async create(modelId, input, options) {
-        const response = await this.fetchWithTimeout(`${modelId}`, {
+        // Build URL with webhook if provided in options
+        let url = `${modelId}`;
+        if (options === null || options === void 0 ? void 0 : options.webhook) {
+            url += `?webhook=${options.webhook}`;
+        }
+        const response = await this.fetchWithTimeout(url, {
             method: 'POST',
             body: JSON.stringify(input),
             ...options
@@ -208,6 +213,9 @@ class WaveSpeed {
             throw new Error(`Failed to create prediction: ${response.status} ${errorText}`);
         }
         const data = await response.json();
+        if (data.code !== 200) {
+            throw new Error(`Failed to create prediction: ${data.code} ${data}`);
+        }
         return new Prediction(data.data, this);
     }
 }

package/package.json

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