wavespeed 0.0.4 → 0.0.6

package/dist/index.d.ts

@@ -11,12 +11,23 @@ export type PredictionStatus = 'created' | 'processing' | 'completed' | 'failed'
 export interface PredictionUrls {
     get: string;
 }
+export interface UploadFileResp {
+    code: number;
+    message: string;
+    data: {
+        type: string;
+        download_url: string;
+        filename: string;
+        size: number;
+    };
+}
 /**
  * Request options for fetch
  */
 export interface RequestOptions extends RequestInit {
     timeout?: number;
     maxRetries?: number;
+    webhook?: string;
 }
 /**
  * Prediction model representing an image generation job
@@ -93,5 +104,16 @@ export declare class WaveSpeed {
      * @param options Additional fetch options
      */
     create(modelId: string, input: Record<string, any>, options?: RequestOptions): Promise<Prediction>;
+    /**
+     * Upload a file (binary) to the /media/upload/binary endpoint
+     * @param filePath Absolute path to the file to upload
+     * @returns The API response JSON
+     */
+    /**
+     * Upload a file (binary) to the /media/upload/binary endpoint (browser Blob version)
+     * @param file Blob to upload
+     * @returns The API response JSON
+     */
+    upload(file: Blob, options?: RequestOptions): Promise<string>;
 }
 export default WaveSpeed;

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,8 +213,37 @@ 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);
     }
+    /**
+     * Upload a file (binary) to the /media/upload/binary endpoint
+     * @param filePath Absolute path to the file to upload
+     * @returns The API response JSON
+     */
+    /**
+     * Upload a file (binary) to the /media/upload/binary endpoint (browser Blob version)
+     * @param file Blob to upload
+     * @returns The API response JSON
+     */
+    async upload(file, options) {
+        const form = new FormData();
+        form.append('file', file);
+        // Only set Authorization header; browser will set Content-Type
+        const response = await this.fetchWithTimeout('media/upload/binary', {
+            method: 'POST',
+            body: form,
+            ...options
+        });
+        if (!response.ok) {
+            const errorText = await response.text();
+            throw new Error(`Failed to upload file: ${response.status} ${errorText}`);
+        }
+        const resp = await response.json();
+        return resp.data.download_url;
+    }
 }
 exports.WaveSpeed = WaveSpeed;
 // Export default and named exports for different import styles

package/package.json

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