wavespeed 0.0.1 → 0.0.3

package/README.md

@@ -73,6 +73,54 @@ async function generateImage(): Promise<void> {
 generateImage().catch(console.error);
 ```
 
+### Manual Status Polling Example
+
+If you need more control over the polling process, you can use the `create` method and manually poll for status updates:
+
+```typescript
+import WaveSpeed from 'wavespeed';
+
+// Initialize the client with your API key
+const client = new WaveSpeed('YOUR_API_KEY');
+
+async function generateWithManualPolling(): Promise<void> {
+  // Create a prediction without waiting
+  const prediction = await client.create('wavespeed-ai/flux-dev', {
+    prompt: 'A beautiful mountain landscape at sunset',
+    size: '1024*1024',
+    num_inference_steps: 28,
+    guidance_scale: 5.0,
+    num_images: 1
+  });
+
+  console.log(`Prediction created with ID: ${prediction.id}`);
+  console.log(`Initial status: ${prediction.status}`);
+
+  // Manually poll for status updates
+  let currentPrediction = prediction;
+  
+  while (currentPrediction.status !== 'completed' && currentPrediction.status !== 'failed') {
+    console.log('Prediction still processing, checking again in 2 seconds...');
+    await new Promise(resolve => setTimeout(resolve, 2000));
+    
+    // Reload the prediction to get the latest status
+    currentPrediction = await currentPrediction.reload();
+    console.log(`Updated status: ${currentPrediction.status}`);
+  }
+  
+  if (currentPrediction.status === 'completed') {
+    console.log('Prediction completed successfully!');
+    currentPrediction.outputs.forEach((imgUrl, i) => {
+      console.log(`Image ${i+1}: ${imgUrl}`);
+    });
+  } else {
+    console.error(`Prediction failed: ${currentPrediction.error}`);
+  }
+}
+
+generateWithManualPolling().catch(console.error);
+```
+
 ## API Reference
 
 ### WaveSpeed Client

package/dist/index.js

@@ -1,7 +1,9 @@
 "use strict";
+/**
+ * Input parameters for image generation
+ */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.WaveSpeed = exports.Prediction = void 0;
-const url_1 = require("url");
 /**
  * Prediction model representing an image generation job
  */
@@ -73,15 +75,23 @@ class WaveSpeed {
      */
     constructor(apiKey, options = {}) {
         this.baseUrl = 'https://api.wavespeed.ai/api/v2/';
-        this.apiKey = apiKey || process.env.WAVESPEED_API_KEY || '';
+        // Browser-friendly environment variable handling
+        const getEnvVar = (name) => {
+            // Try to get from process.env for Node.js environments
+            if (typeof process !== 'undefined' && process.env && process.env[name]) {
+                return process.env[name];
+            }
+            return undefined;
+        };
+        this.apiKey = apiKey || getEnvVar('WAVESPEED_API_KEY') || '';
         if (!this.apiKey) {
             throw new Error('API key is required. Provide it as a parameter or set the WAVESPEED_API_KEY environment variable.');
         }
         if (options.baseUrl) {
             this.baseUrl = options.baseUrl;
         }
-        this.pollInterval = options.pollInterval || Number(process.env.WAVESPEED_POLL_INTERVAL) || 1;
-        this.timeout = options.timeout || Number(process.env.WAVESPEED_TIMEOUT) || 60;
+        this.pollInterval = options.pollInterval || Number(getEnvVar('WAVESPEED_POLL_INTERVAL')) || 1;
+        this.timeout = options.timeout || Number(getEnvVar('WAVESPEED_TIMEOUT')) || 60;
     }
     /**
      * Fetch with timeout support
@@ -97,11 +107,13 @@ class WaveSpeed {
             'Authorization': `Bearer ${this.apiKey}`,
             'Content-Type': 'application/json'
         };
+        // Use AbortController for timeout (supported in modern browsers)
         const controller = new AbortController();
         const id = setTimeout(() => controller.abort(), timeout);
         try {
-            // Construct the full URL by joining baseUrl and path
-            const url = new url_1.URL(path, this.baseUrl).toString();
+            // Use browser's built-in URL API
+            const url = new URL(path, this.baseUrl).toString();
+            // Use the global fetch API available in browsers
             const response = await fetch(url, {
                 ...fetchOptions,
                 signal: controller.signal
@@ -147,3 +159,7 @@ class WaveSpeed {
 exports.WaveSpeed = WaveSpeed;
 // Export default and named exports for different import styles
 exports.default = WaveSpeed;
+// Add browser global for UMD-style usage
+if (typeof window !== 'undefined') {
+    window.WaveSpeed = WaveSpeed;
+}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "wavespeed",
-  "version": "0.0.1",
-  "description": "Client SDK for Wavespeed API",
+  "version": "0.0.3",
+  "description": "WaveSpeed Client SDK for Wavespeed API",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "scripts": {
@@ -34,4 +34,4 @@
   "files": [
     "dist"
   ]
-}
+}