wavespeed 0.0.13 → 0.0.15

package/dist/index.d.ts

@@ -106,6 +106,7 @@ export declare class WaveSpeed {
      * @param options Additional fetch options
      */
     create(modelId: string, input: Record<string, any>, options?: RequestOptions): Promise<Prediction>;
+    getPrediction(predictionId: string): Promise<Prediction>;
     /**
      * Upload a file (binary) to the /media/upload/binary endpoint
      * @param filePath Absolute path to the file to upload
@@ -116,6 +117,6 @@ export declare class WaveSpeed {
      * @param file Blob to upload
      * @returns The API response JSON
      */
-    upload(file: Blob, options?: RequestOptions): Promise<string>;
+    upload(file: File | Blob, options?: RequestOptions): Promise<string>;
 }
 export default WaveSpeed;

package/dist/index.js

@@ -131,7 +131,8 @@ class WaveSpeed {
             const timeoutId = setTimeout(() => controller.abort(), timeout);
             try {
                 // Construct the full URL by joining baseUrl and path
-                const url = new URL(path.startsWith('/') ? path.substring(1) : path, this.baseUrl).toString();
+                const baseUrl = options.isUpload ? this.baseUrl.replace('v3', 'v2') : this.baseUrl;
+                const url = new URL(path.startsWith('/') ? path.substring(1) : path, baseUrl).toString();
                 const response = await fetch(url, {
                     ...fetchOptions,
                     signal: controller.signal
@@ -225,6 +226,18 @@ class WaveSpeed {
         }
         return new Prediction(data.data, this);
     }
+    async getPrediction(predictionId) {
+        const response = await this.fetchWithTimeout(`predictions/${predictionId}/result`);
+        if (!response.ok) {
+            const errorText = await response.text();
+            throw new Error(`Failed to get prediction: ${response.status} ${errorText}`);
+        }
+        const data = await response.json();
+        if (data.code !== 200) {
+            throw new Error(`Failed to get 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

package/package.json

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