wavespeed 2.0.2__tar.gz → 2.0.4__tar.gz

wavespeed-2.0.4/.github/dependabot.yml

@@ -0,0 +1,11 @@
+version: 2
+updates:
+  # Actions are pinned to commit SHAs; this keeps the pins moving.
+  - package-ecosystem: github-actions
+    directory: /
+    schedule:
+      interval: weekly
+    # One pull request per run instead of one per action.
+    groups:
+      actions:
+        patterns: ['*']

{wavespeed-2.0.2 → wavespeed-2.0.4}/.github/workflows/pre-commit.yml

@@ -24,6 +24,7 @@ jobs:
     steps:
     - uses: actions/checkout@f43a0e5ff2bd294095638e18286ca9a3d1956744 # v3
       with:
+        persist-credentials: false
         path: "."
     - name: Set up Python
       uses: actions/setup-python@3542bca2639a428e1796aaa6a2ffef0c0f575566 # v3

{wavespeed-2.0.2 → wavespeed-2.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wavespeed
-Version: 2.0.2
+Version: 2.0.4
 Summary: WaveSpeed Python Client — Official Python SDK for the WaveSpeed inference platform
 Author-email: WaveSpeedAI <support@wavespeed.ai>
 License: MIT

{wavespeed-2.0.2 → wavespeed-2.0.4}/src/wavespeed/_version.py

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

{wavespeed-2.0.2 → wavespeed-2.0.4}/src/wavespeed/api/client.py

@@ -44,6 +44,10 @@ class _SubmissionError(RuntimeError):
     """A task submission failed and must not be retried automatically."""
 
 
+class _TaskFailedError(RuntimeError):
+    """The task itself reported status ``failed``; a task-level re-run is allowed."""
+
+
 class Client:
     """WaveSpeed API client.
 
@@ -332,14 +336,13 @@ class Client:
 
             if status in ("failed", "cancelled", "timeout", "deleted"):
                 error = data.get("error") or "Unknown error"
-                raise RuntimeError(
-                    f"Prediction {status} (task_id: {request_id}): {error}"
-                )
+                error_cls = _TaskFailedError if status == "failed" else RuntimeError
+                raise error_cls(f"Prediction {status} (task_id: {request_id}): {error}")
 
             time.sleep(poll_interval)
 
     def _is_retryable_error(self, error: Exception) -> bool:
-        """Determine if an error is worth retrying at the task level.
+        """Determine if an error is worth re-running the task for.
 
         Args:
             error: The exception to check.
@@ -347,29 +350,17 @@ class Client:
         Returns:
             True if the error is retryable.
         """
-        # Submission errors are ambiguous: the server may already have created
-        # the task, so never turn them into another POST automatically.
-        if isinstance(error, _SubmissionError):
+        # Only a failure reported by the task itself is re-run, and only when it
+        # was a transient upstream error (HTTP 5xx / 429), as before: a
+        # deterministic failure such as a content-policy rejection would fail
+        # again. Submission errors are ambiguous (the task may already exist),
+        # and poll-side HTTP errors, poll retry exhaustion and client-side wait
+        # timeouts all happen after the task was created and may be charged, so
+        # none of them may turn into another POST.
+        if not isinstance(error, _TaskFailedError):
             return False
-
-        # Retry timeout and connection errors from result-query GETs.
-        if isinstance(
-            error,
-            (
-                requests.exceptions.Timeout,
-                requests.exceptions.ConnectionError,
-                TimeoutError,
-            ),
-        ):
-            return True
-
-        # Retry server errors (5xx) and rate limiting (429)
-        if isinstance(error, RuntimeError):
-            error_str = str(error)
-            if "HTTP 5" in error_str or "HTTP 429" in error_str:
-                return True
-
-        return False
+        error_str = str(error)
+        return "HTTP 5" in error_str or "HTTP 429" in error_str
 
     def _format_sync_mode_error(self, data: dict[str, Any]) -> str:
         """Build an actionable error for a non-completed sync-mode response."""
@@ -464,8 +455,10 @@ class Client:
     @staticmethod
     def _task_id_from_error(error: Exception) -> str:
         """Extract a task ID from an error message, mirroring the JS SDK."""
-        match = re.search(r"task_id:\s*([^)]+)", str(error))
-        return match.group(1).strip() if match else "unknown"
+        match = re.search(r"task_id:\s*([^)]+)|for task ([^\s:]+)", str(error))
+        if not match:
+            return "unknown"
+        return (match.group(1) or match.group(2)).strip()
 
     def run_no_throw(
         self,
@@ -587,7 +580,8 @@ class Client:
             URL of the uploaded file.
 
         Raises:
-            ValueError: If API key is not configured.
+            ValueError: If API key is not configured, or a non-seekable
+                stream exceeds the 200 MB upload limit.
             FileNotFoundError: If file path does not exist.
             RuntimeError: If upload fails.
 
@@ -627,7 +621,10 @@ class Client:
                 size = stream.tell() - start
                 stream.seek(start)
             except (AttributeError, OSError, io.UnsupportedOperation):
-                data = stream.read(200 * 1024 * 1024 + 1)
+                max_size = 200 * 1024 * 1024
+                data = stream.read(max_size + 1)
+                if len(data) > max_size:
+                    raise ValueError("File size exceeds the 200 MB limit.")
                 stream = io.BytesIO(data)
                 size = len(data)
 

{wavespeed-2.0.2 → wavespeed-2.0.4}/src/wavespeed.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wavespeed
-Version: 2.0.2
+Version: 2.0.4
 Summary: WaveSpeed Python Client — Official Python SDK for the WaveSpeed inference platform
 Author-email: WaveSpeedAI <support@wavespeed.ai>
 License: MIT

{wavespeed-2.0.2 → wavespeed-2.0.4}/src/wavespeed.egg-info/SOURCES.txt

@@ -9,6 +9,7 @@ VERSIONING.md
 pyproject.toml
 pytest.ini
 setup.cfg
+.github/dependabot.yml
 .github/workflows/claude.yml
 .github/workflows/pre-commit.yml
 .github/workflows/python-packages.yml

{wavespeed-2.0.2 → wavespeed-2.0.4}/src/wavespeed.egg-info/scm_file_list.json

@@ -1,5 +1,6 @@
 {
   "files": [
+    ".github/dependabot.yml",
     ".github/workflows/claude.yml",
     ".github/workflows/pre-commit.yml",
     ".github/workflows/python-packages.yml",

wavespeed-2.0.4/src/wavespeed.egg-info/scm_version.json

@@ -0,0 +1,8 @@
+{
+  "tag": "2.0.4",
+  "distance": 0,
+  "node": "g1bf08ac65f5019f75a190187231450423b203238",
+  "dirty": false,
+  "branch": "HEAD",
+  "node_date": "2026-10-17"
+}

{wavespeed-2.0.2 → wavespeed-2.0.4}/tests/test_api.py

@@ -149,6 +149,98 @@ class TestClient(unittest.TestCase):
 
         mock_post.assert_called_once()
 
+    @staticmethod
+    def _response(status_code, payload=None, text=""):
+        response = MagicMock()
+        response.status_code = status_code
+        response.text = text
+        response.json.return_value = payload
+        return response
+
+    @patch("wavespeed.api.client.time.sleep")
+    @patch("wavespeed.api.client.requests.get")
+    @patch("wavespeed.api.client.requests.post")
+    def test_task_retries_do_not_repost_after_poll_retry_exhaustion(
+        self, mock_post, mock_get, _mock_sleep
+    ):
+        """A poll that exhausts its GET retries must not create a second task."""
+        mock_post.return_value = self._response(200, {"data": {"id": "req-1"}})
+        mock_get.return_value = self._response(502, text="Bad Gateway")
+
+        client = Client(api_key="test-key", max_retries=2, max_connection_retries=1)
+        with self.assertRaises(RuntimeError) as ctx:
+            client.run("wavespeed-ai/z-image/turbo", {"prompt": "test"})
+
+        self.assertIn("req-1", str(ctx.exception))
+        self.assertIn("HTTP 502", str(ctx.exception))
+        self.assertEqual(mock_post.call_count, 1)
+        self.assertEqual(mock_get.call_count, 2)
+
+        result = client.run_no_throw("wavespeed-ai/z-image/turbo", {"prompt": "test"})
+        self.assertEqual(result["task_id"], "req-1")
+        self.assertEqual(mock_post.call_count, 2)  # one POST per run, never per poll
+
+    @patch("wavespeed.api.client.time.sleep")
+    @patch("wavespeed.api.client.requests.get")
+    @patch("wavespeed.api.client.requests.post")
+    def test_task_retries_do_not_repost_after_wait_timeout(
+        self, mock_post, mock_get, _mock_sleep
+    ):
+        """A client-side wait timeout leaves the task alive and must not re-POST."""
+        mock_post.return_value = self._response(200, {"data": {"id": "req-1"}})
+        mock_get.return_value = self._response(200, {"data": {"status": "processing"}})
+
+        client = Client(api_key="test-key", max_retries=2)
+        with self.assertRaises(TimeoutError) as ctx:
+            client.run("wavespeed-ai/z-image/turbo", {"prompt": "test"}, timeout=0)
+
+        self.assertIn("req-1", str(ctx.exception))
+        self.assertEqual(mock_post.call_count, 1)
+
+    @patch("wavespeed.api.client.time.sleep")
+    @patch("wavespeed.api.client.requests.get")
+    @patch("wavespeed.api.client.requests.post")
+    def test_task_retries_rerun_when_task_reports_failed(
+        self, mock_post, mock_get, _mock_sleep
+    ):
+        """A task that fails with a transient upstream error is re-run when max_retries > 0."""
+        mock_post.side_effect = [
+            self._response(200, {"data": {"id": "req-1"}}),
+            self._response(200, {"data": {"id": "req-2"}}),
+        ]
+        mock_get.side_effect = [
+            self._response(
+                200, {"data": {"status": "failed", "error": "HTTP 500 upstream error"}}
+            ),
+            self._response(200, {"data": {"status": "completed", "outputs": ["url"]}}),
+        ]
+
+        client = Client(api_key="test-key", max_retries=1)
+        result = client.run("wavespeed-ai/z-image/turbo", {"prompt": "test"})
+
+        self.assertEqual(result, {"outputs": ["url"]})
+        self.assertEqual(mock_post.call_count, 2)
+
+    @patch("wavespeed.api.client.time.sleep")
+    @patch("wavespeed.api.client.requests.get")
+    @patch("wavespeed.api.client.requests.post")
+    def test_task_retries_do_not_rerun_deterministic_failure(
+        self, mock_post, mock_get, _mock_sleep
+    ):
+        """A task that fails for a non-transient reason is not re-submitted."""
+        mock_post.side_effect = [self._response(200, {"data": {"id": "req-1"}})]
+        mock_get.side_effect = [
+            self._response(
+                200, {"data": {"status": "failed", "error": "content policy violation"}}
+            ),
+        ]
+
+        client = Client(api_key="test-key", max_retries=2)
+        with self.assertRaises(RuntimeError):
+            client.run("wavespeed-ai/z-image/turbo", {"prompt": "test"})
+
+        self.assertEqual(mock_post.call_count, 1)
+
     @patch("wavespeed.api.client.requests.get")
     def test_get_result_success(self, mock_get):
         """Test successful result retrieval."""
@@ -582,6 +674,22 @@ class TestUpload(unittest.TestCase):
         with self.assertRaises(FileNotFoundError):
             client.upload("/nonexistent/path/to/file.png")
 
+    @patch("wavespeed.api.client.requests.post")
+    def test_upload_oversize_stream_raises(self, mock_post):
+        """A non-seekable stream over the limit fails before any request."""
+
+        class Pipe:
+            name = "clip.mp4"
+
+            def read(self, n):
+                return bytes(n)
+
+        client = Client(api_key="test-key")
+        with self.assertRaises(ValueError) as ctx:
+            client.upload(Pipe())
+        self.assertIn("200 MB", str(ctx.exception))
+        mock_post.assert_not_called()
+
     def test_upload_raises_without_api_key(self):
         """Test that upload raises ValueError without API key."""
         client = Client()

wavespeed-2.0.2/src/wavespeed.egg-info/scm_version.json

@@ -1,8 +0,0 @@
-{
-  "tag": "2.0.2",
-  "distance": 0,
-  "node": "gac985323c284a8d135a6396b7dc2ac6f9f5e7fd4",
-  "dirty": false,
-  "branch": "HEAD",
-  "node_date": "2026-08-26"
-}