wavespeed 1.0.4__tar.gz → 1.0.6__tar.gz

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

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

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

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

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

@@ -312,6 +312,15 @@ class Client:
 
                 if enable_sync_mode:
                     # In sync mode, extract outputs from the result
+                    status = sync_result.get("data", {}).get("status")
+                    if status != "completed":
+                        error = (
+                            sync_result.get("data", {}).get("error") or "Unknown error"
+                        )
+                        request_id = sync_result.get("data", {}).get("id", "unknown")
+                        raise RuntimeError(
+                            f"Prediction failed (task_id: {request_id}): {error}"
+                        )
                     data = sync_result.get("data", {})
                     return {"outputs": data.get("outputs", [])}
 

{wavespeed-1.0.4 → wavespeed-1.0.6}/src/wavespeed/serverless/modules/scaler.py

@@ -62,6 +62,12 @@ class JobScaler:
         if concurrency_modifier := config.get("concurrency_modifier"):
             self.concurrency_modifier = concurrency_modifier
 
+        # Apply concurrency modifier immediately
+        old_concurrency = self.current_concurrency
+        self.current_concurrency = self.concurrency_modifier(self.current_concurrency)
+        if self.current_concurrency != old_concurrency:
+            self.jobs_queue = asyncio.Queue(maxsize=self.current_concurrency)
+
         # Allow overriding jobs_fetcher and handler in local test mode
         if not IS_LOCAL_TEST:
             return
@@ -207,20 +213,33 @@ class JobScaler:
         tasks: list[asyncio.Task[None]] = []
 
         while self.is_alive() or not self.jobs_queue.empty():
+            num_tasks = len(tasks)
+
             while len(tasks) < self.current_concurrency and not self.jobs_queue.empty():
                 job = await self.jobs_queue.get()
                 task = asyncio.create_task(self.handle_job(session, job))
                 tasks.append(task)
 
-            if tasks:
-                log.info(f"Jobs in progress: {len(tasks)}")
+            if len(tasks) > num_tasks:
+                log.info(
+                    f"Started {len(tasks) - num_tasks} new job(s), "
+                    f"total jobs in progress: {len(tasks)}"
+                )
 
+            if tasks:
                 done, pending = await asyncio.wait(
-                    tasks, return_when=asyncio.FIRST_COMPLETED
+                    tasks, timeout=0.1, return_when=asyncio.FIRST_COMPLETED
                 )
 
+                num_tasks = len(tasks)
                 tasks = [t for t in tasks if t not in done]
 
+                if len(tasks) < num_tasks:
+                    log.info(
+                        f"Completed {num_tasks - len(tasks)} job(s), "
+                        f"total jobs in progress: {len(tasks)}"
+                    )
+
             await asyncio.sleep(0)
 
         await asyncio.gather(*tasks)

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

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