wavespeed 0.0.5 → 0.0.7

package/dist/index.d.ts

@@ -11,6 +11,16 @@ 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
  */
@@ -18,6 +28,7 @@ export interface RequestOptions extends RequestInit {
     timeout?: number;
     maxRetries?: number;
     webhook?: string;
+    isUpload?: boolean;
 }
 /**
  * Prediction model representing an image generation job
@@ -94,5 +105,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

@@ -102,11 +102,19 @@ class WaveSpeed {
     async fetchWithTimeout(path, options = {}) {
         const { timeout = this.timeout * 1000, ...fetchOptions } = options;
         // Ensure headers exist
-        fetchOptions.headers = {
-            ...fetchOptions.headers,
-            'Authorization': `Bearer ${this.apiKey}`,
-            'Content-Type': 'application/json'
-        };
+        if (!options.isUpload) {
+            fetchOptions.headers = {
+                'Authorization': `Bearer ${this.apiKey}`,
+                ...(fetchOptions.headers || {}),
+            };
+        }
+        else {
+            fetchOptions.headers = {
+                'Authorization': `Bearer ${this.apiKey}`,
+                'content-type': 'application/json',
+                ...(fetchOptions.headers || {}),
+            };
+        }
         // Default retry options
         const maxRetries = options.maxRetries || 3;
         const initialBackoff = 1000; // 1 second
@@ -218,6 +226,35 @@ class WaveSpeed {
         }
         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
+        if (options == null) {
+            options = { isUpload: true };
+        }
+        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.5",
+  "version": "0.0.7",
   "description": "WaveSpeed Client SDK for Wavespeed API",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",