From d95fffcecafd3634f36412f40e5f2b2c7427f684 Mon Sep 17 00:00:00 2001
From: "liuzhao@meicai.cn" <tolz@qq.com>
Date: Tue, 28 Apr 2026 14:58:32 +0800
Subject: [PATCH] fix:504 and 429 related errors handled

---
 src/wavespeed/serverless/modules/health.py | 4 +++-
 tests/test_config.py                       | 4 +---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/wavespeed/serverless/modules/health.py b/src/wavespeed/serverless/modules/health.py
index b63e686..78c0240 100644
--- a/src/wavespeed/serverless/modules/health.py
+++ b/src/wavespeed/serverless/modules/health.py
@@ -45,7 +45,9 @@ def start(self) -> None:
         """Start the health server in a background thread."""
         try:
             self._server = HTTPServer((self.host, self.port), HealthHandler)
-            self._thread = threading.Thread(target=self._server.serve_forever, daemon=True)
+            self._thread = threading.Thread(
+                target=self._server.serve_forever, daemon=True
+            )
             self._thread.start()
             log.info(f"Health server started at http://{self.host}:{self.port}/health")
         except Exception as e:
diff --git a/tests/test_config.py b/tests/test_config.py
index 6bf07d8..26846e2 100644
--- a/tests/test_config.py
+++ b/tests/test_config.py
@@ -43,9 +43,7 @@ def test_replaces_waverless_pod_id_placeholder(self):
         """Test that $WAVERLESS_POD_ID is replaced with pod_id."""
         template = "https://api.wavespeed.ai/v2/test/job-take/$WAVERLESS_POD_ID"
         result = _resolve_waverless_url(template, "my-pod-123")
-        self.assertEqual(
-            result, "https://api.wavespeed.ai/v2/test/job-take/my-pod-123"
-        )
+        self.assertEqual(result, "https://api.wavespeed.ai/v2/test/job-take/my-pod-123")
 
     def test_preserves_id_placeholder(self):
         """Test that $ID is NOT replaced - it's for job/worker ID at runtime."""