wavespeed 0.1.2__tar.gz → 1.0.4__tar.gz

{wavespeed-0.1.2 → wavespeed-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wavespeed
-Version: 0.1.2
+Version: 1.0.4
 Summary: WaveSpeedAI Python Client — Official Python SDK for WaveSpeedAI inference platform
 Author-email: WaveSpeedAI <support@wavespeed.ai>
 License: MIT
@@ -62,7 +62,7 @@ import wavespeed
 
 output = wavespeed.run(
     "wavespeed-ai/z-image/turbo",
-    input={"prompt": "Cat"},
+    {"prompt": "Cat"},
 )
 
 print(output["outputs"][0])  # Output URL
@@ -82,7 +82,7 @@ Or pass it directly:
 from wavespeed import Client
 
 client = Client(api_key="your-api-key")
-output = client.run("wavespeed-ai/z-image/turbo", input={"prompt": "Cat"})
+output = client.run("wavespeed-ai/z-image/turbo", {"prompt": "Cat"})
 ```
 
 ### Options
@@ -90,7 +90,7 @@ output = client.run("wavespeed-ai/z-image/turbo", input={"prompt": "Cat"})
 ```python
 output = wavespeed.run(
     "wavespeed-ai/z-image/turbo",
-    input={"prompt": "Cat"},
+    {"prompt": "Cat"},
     timeout=36000.0,       # Max wait time in seconds (default: 36000.0)
     poll_interval=1.0,     # Status check interval (default: 1.0)
     enable_sync_mode=False, # Single request mode, no polling (default: False)
@@ -106,7 +106,7 @@ Use `enable_sync_mode=True` for a single request that waits for the result (no p
 ```python
 output = wavespeed.run(
     "wavespeed-ai/z-image/turbo",
-    input={"prompt": "Cat"},
+    {"prompt": "Cat"},
     enable_sync_mode=True,
 )
 ```

{wavespeed-0.1.2 → wavespeed-1.0.4}/README.md

@@ -33,7 +33,7 @@ import wavespeed
 
 output = wavespeed.run(
     "wavespeed-ai/z-image/turbo",
-    input={"prompt": "Cat"},
+    {"prompt": "Cat"},
 )
 
 print(output["outputs"][0])  # Output URL
@@ -53,7 +53,7 @@ Or pass it directly:
 from wavespeed import Client
 
 client = Client(api_key="your-api-key")
-output = client.run("wavespeed-ai/z-image/turbo", input={"prompt": "Cat"})
+output = client.run("wavespeed-ai/z-image/turbo", {"prompt": "Cat"})
 ```
 
 ### Options
@@ -61,7 +61,7 @@ output = client.run("wavespeed-ai/z-image/turbo", input={"prompt": "Cat"})
 ```python
 output = wavespeed.run(
     "wavespeed-ai/z-image/turbo",
-    input={"prompt": "Cat"},
+    {"prompt": "Cat"},
     timeout=36000.0,       # Max wait time in seconds (default: 36000.0)
     poll_interval=1.0,     # Status check interval (default: 1.0)
     enable_sync_mode=False, # Single request mode, no polling (default: False)
@@ -77,7 +77,7 @@ Use `enable_sync_mode=True` for a single request that waits for the result (no p
 ```python
 output = wavespeed.run(
     "wavespeed-ai/z-image/turbo",
-    input={"prompt": "Cat"},
+    {"prompt": "Cat"},
     enable_sync_mode=True,
 )
 ```

{wavespeed-0.1.2 → wavespeed-1.0.4}/src/wavespeed/__init__.py

@@ -10,7 +10,7 @@ Example usage:
 
     output = wavespeed.run(
         "wavespeed-ai/z-image/turbo",
-        input={"prompt": "A beautiful sunset"}
+        {"prompt": "A beautiful sunset"}
     )
     print(output["outputs"][0])
 """

{wavespeed-0.1.2 → wavespeed-1.0.4}/src/wavespeed/_version.py

@@ -1,2 +1,2 @@
 # This file is automatically generated by setuptools_scm
-__version__ = "0.1.2"
+__version__ = "1.0.4"

{wavespeed-0.1.2 → wavespeed-1.0.4}/src/wavespeed/api/__init__.py

@@ -7,7 +7,7 @@ Example usage:
 
     output = wavespeed.run(
         "wavespeed-ai/z-image/turbo",
-        input={"prompt": "A beautiful sunset over mountains"}
+        {"prompt": "A beautiful sunset over mountains"}
     )
 
     print(output["outputs"][0])  # First output URL
@@ -65,27 +65,27 @@ def run(
     Example:
         output = wavespeed.run(
             "wavespeed-ai/z-image/turbo",
-            input={"prompt": "A cat sitting on a windowsill"}
+            {"prompt": "A cat sitting on a windowsill"}
         )
         print(output["outputs"][0])  # First output URL
 
         # With sync mode
         output = wavespeed.run(
             "wavespeed-ai/z-image/turbo",
-            input={"prompt": "A cat"},
+            {"prompt": "A cat"},
             enable_sync_mode=True
         )
 
         # With retry
         output = wavespeed.run(
             "wavespeed-ai/z-image/turbo",
-            input={"prompt": "A cat"},
+            {"prompt": "A cat"},
             max_retries=3
         )
     """
     return _get_default_client().run(
         model,
-        input=input,
+        input,
         timeout=timeout,
         poll_interval=poll_interval,
         enable_sync_mode=enable_sync_mode,

{wavespeed-0.1.2 → wavespeed-1.0.4}/src/wavespeed/api/client.py

@@ -23,13 +23,13 @@ class Client:
 
     Example:
         client = Client(api_key="your-api-key")
-        output = client.run("wavespeed-ai/z-image/turbo", input={"prompt": "Cat"})
+        output = client.run("wavespeed-ai/z-image/turbo", {"prompt": "Cat"})
 
         # With sync mode (single request, waits for result)
-        output = client.run("wavespeed-ai/z-image/turbo", input={"prompt": "Cat"}, enable_sync_mode=True)
+        output = client.run("wavespeed-ai/z-image/turbo", {"prompt": "Cat"}, enable_sync_mode=True)
 
         # With retry
-        output = client.run("wavespeed-ai/z-image/turbo", input={"prompt": "Cat"}, max_retries=3)
+        output = client.run("wavespeed-ai/z-image/turbo", {"prompt": "Cat"}, max_retries=3)
     """
 
     def __init__(
@@ -179,7 +179,8 @@ class Client:
 
                 if response.status_code != 200:
                     raise RuntimeError(
-                        f"Failed to get result: HTTP {response.status_code}: {response.text}"
+                        f"Failed to get result for task {request_id}: "
+                        f"HTTP {response.status_code}: {response.text}"
                     )
 
                 return response.json()
@@ -199,7 +200,8 @@ class Client:
                     time.sleep(delay)
                 else:
                     raise RuntimeError(
-                        f"Failed to get result after {self.max_connection_retries + 1} attempts"
+                        f"Failed to get result for task {request_id} "
+                        f"after {self.max_connection_retries + 1} attempts"
                     ) from e
 
     def _wait(
@@ -229,7 +231,9 @@ class Client:
             if timeout is not None:
                 elapsed = time.time() - start_time
                 if elapsed >= timeout:
-                    raise TimeoutError(f"Prediction timed out after {timeout} seconds")
+                    raise TimeoutError(
+                        f"Prediction timed out after {timeout} seconds (task_id: {request_id})"
+                    )
 
             result = self._get_result(request_id, timeout=timeout)
             data = result.get("data", {})
@@ -240,7 +244,9 @@ class Client:
 
             if status == "failed":
                 error = data.get("error") or "Unknown error"
-                raise RuntimeError(f"Prediction failed: {error}")
+                raise RuntimeError(
+                    f"Prediction failed (task_id: {request_id}): {error}"
+                )
 
             time.sleep(poll_interval)
 

{wavespeed-0.1.2 → wavespeed-1.0.4}/src/wavespeed/serverless/modules/logger.py

@@ -2,7 +2,7 @@
 
 import json
 import sys
-from datetime import datetime
+from datetime import datetime, timezone
 from enum import IntEnum
 from typing import Optional
 
@@ -113,7 +113,9 @@ class WaverlessLogger:
                 "requestId": job_id or "N/A",
                 "message": message,
                 "level": level.name,
-                "timestamp": datetime.utcnow().isoformat() + "Z",
+                "timestamp": datetime.now(timezone.utc)
+                .isoformat()
+                .replace("+00:00", "Z"),
             }
             return json.dumps(log_data)
         else:

{wavespeed-0.1.2 → wavespeed-1.0.4}/src/wavespeed.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wavespeed
-Version: 0.1.2
+Version: 1.0.4
 Summary: WaveSpeedAI Python Client — Official Python SDK for WaveSpeedAI inference platform
 Author-email: WaveSpeedAI <support@wavespeed.ai>
 License: MIT
@@ -62,7 +62,7 @@ import wavespeed
 
 output = wavespeed.run(
     "wavespeed-ai/z-image/turbo",
-    input={"prompt": "Cat"},
+    {"prompt": "Cat"},
 )
 
 print(output["outputs"][0])  # Output URL
@@ -82,7 +82,7 @@ Or pass it directly:
 from wavespeed import Client
 
 client = Client(api_key="your-api-key")
-output = client.run("wavespeed-ai/z-image/turbo", input={"prompt": "Cat"})
+output = client.run("wavespeed-ai/z-image/turbo", {"prompt": "Cat"})
 ```
 
 ### Options
@@ -90,7 +90,7 @@ output = client.run("wavespeed-ai/z-image/turbo", input={"prompt": "Cat"})
 ```python
 output = wavespeed.run(
     "wavespeed-ai/z-image/turbo",
-    input={"prompt": "Cat"},
+    {"prompt": "Cat"},
     timeout=36000.0,       # Max wait time in seconds (default: 36000.0)
     poll_interval=1.0,     # Status check interval (default: 1.0)
     enable_sync_mode=False, # Single request mode, no polling (default: False)
@@ -106,7 +106,7 @@ Use `enable_sync_mode=True` for a single request that waits for the result (no p
 ```python
 output = wavespeed.run(
     "wavespeed-ai/z-image/turbo",
-    input={"prompt": "Cat"},
+    {"prompt": "Cat"},
     enable_sync_mode=True,
 )
 ```

{wavespeed-0.1.2 → wavespeed-1.0.4}/tests/test_api.py

@@ -118,7 +118,7 @@ class TestClient(unittest.TestCase):
         mock_get.return_value = mock_get_response
 
         client = Client(api_key="test-key")
-        result = client.run("wavespeed-ai/z-image/turbo", input={"prompt": "test"})
+        result = client.run("wavespeed-ai/z-image/turbo", {"prompt": "test"})
 
         self.assertEqual(result["outputs"], ["https://example.com/out.png"])
 
@@ -142,7 +142,7 @@ class TestClient(unittest.TestCase):
 
         client = Client(api_key="test-key")
         with self.assertRaises(RuntimeError) as ctx:
-            client.run("wavespeed-ai/z-image/turbo", input={"prompt": "test"})
+            client.run("wavespeed-ai/z-image/turbo", {"prompt": "test"})
         self.assertIn("Model error", str(ctx.exception))
 
     @patch("wavespeed.api.client.time.time")
@@ -168,9 +168,7 @@ class TestClient(unittest.TestCase):
 
         client = Client(api_key="test-key")
         with self.assertRaises(TimeoutError) as ctx:
-            client.run(
-                "wavespeed-ai/z-image/turbo", input={"prompt": "test"}, timeout=10
-            )
+            client.run("wavespeed-ai/z-image/turbo", {"prompt": "test"}, timeout=10)
         self.assertIn("timed out", str(ctx.exception))
 
 
@@ -208,7 +206,7 @@ class TestModuleLevelRun(unittest.TestCase):
         }
         mock_get.return_value = mock_get_response
 
-        result = wavespeed.run("wavespeed-ai/z-image/turbo", input={"prompt": "test"})
+        result = wavespeed.run("wavespeed-ai/z-image/turbo", {"prompt": "test"})
         self.assertEqual(result["outputs"], ["https://example.com/out.png"])
 
 
@@ -342,7 +340,7 @@ class TestRealAPI(unittest.TestCase):
 
         output = wavespeed.run(
             "wavespeed-ai/z-image/turbo",
-            input={"prompt": "A simple red circle on white background"},
+            {"prompt": "A simple red circle on white background"},
         )
 
         self.assertIn("outputs", output)

{wavespeed-0.1.2 → wavespeed-1.0.4}/tests/test_serverless/test_modules/test_http.py

@@ -132,6 +132,7 @@ class TestFetchJobs(IsolatedAsyncioTestCase):
         mock_response.content_type = "application/json"
         mock_response.content_length = 100
         mock_response.json = AsyncMock(return_value={"id": "job_1", "input": {"n": 1}})
+        mock_response.raise_for_status = MagicMock()
         mock_response.__aenter__.return_value = mock_response
         mock_response.__aexit__.return_value = None
 
@@ -163,6 +164,7 @@ class TestFetchJobs(IsolatedAsyncioTestCase):
                 {"id": "job_2", "input": {"n": 2}},
             ]
         )
+        mock_response.raise_for_status = MagicMock()
         mock_response.__aenter__.return_value = mock_response
         mock_response.__aexit__.return_value = None
 
@@ -287,6 +289,7 @@ class TestFetchJobs(IsolatedAsyncioTestCase):
         mock_response.content_type = "application/json"
         mock_response.content_length = 10
         mock_response.json = AsyncMock(return_value=[])
+        mock_response.raise_for_status = MagicMock()
         mock_response.__aenter__.return_value = mock_response
         mock_response.__aexit__.return_value = None
 

{wavespeed-0.1.2 → wavespeed-1.0.4}/tests/test_serverless/test_modules/test_scaler.py

@@ -108,7 +108,7 @@ class TestJobScalerAsync(IsolatedAsyncioTestCase):
 
             scaler = JobScaler(config)
             scaler.job_progress = mock_progress_instance
-            scaler.jobs_queue = AsyncMock()
+            scaler.jobs_queue = MagicMock()
 
             # Override the jobs_handler directly since __init__ already captured handle_job
             mock_handle = AsyncMock()
@@ -132,7 +132,7 @@ class TestJobScalerAsync(IsolatedAsyncioTestCase):
 
             scaler = JobScaler(config)
             scaler.job_progress = mock_progress_instance
-            scaler.jobs_queue = AsyncMock()
+            scaler.jobs_queue = MagicMock()
 
             # Override the jobs_handler directly since __init__ already captured handle_job
             mock_handle = AsyncMock()