wavespeed 2.0.3__tar.gz → 2.0.4__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wavespeed
-Version: 2.0.3
+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.3 → wavespeed-2.0.4}/src/wavespeed/_version.py

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

{wavespeed-2.0.3 → 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,

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wavespeed
-Version: 2.0.3
+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.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.3 → 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."""

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

@@ -1,8 +0,0 @@
-{
-  "tag": "2.0.3",
-  "distance": 0,
-  "node": "g3e926667ae3ef93fc11dc277b6f09b081750ad7d",
-  "dirty": false,
-  "branch": "HEAD",
-  "node_date": "2026-09-21"
-}