FIX pre-commit

cua2-core/src/cua2_core/app.py

@@ -21,7 +21,7 @@ async def lifespan(app: FastAPI):
     if not os.getenv("HF_TOKEN"):
         raise ValueError("HF_TOKEN is not set")
 
-    num_workers = int(os.getenv("NUM_WORKERS", "1"))
+    # num_workers = int(os.getenv("NUM_WORKERS", "1"))
     # max_sandboxes = int(600 / num_workers)
     max_sandboxes = 600
 

cua2-core/src/cua2_core/services/agent_service.py

@@ -204,7 +204,13 @@ class AgentService:
                         sandbox = response.sandbox
                         break
                     elif response.state == "max_sandboxes_reached":
-                        raise Exception("No sandbox available: pool limit reached")
+                        (
+                            available_count,
+                            non_available_count,
+                        ) = await self.sandbox_service.get_sandbox_counts()
+                        raise Exception(
+                            f"No sandbox available: pool limit reached (available: {available_count}, non-available: {non_available_count}, max: {self.max_sandboxes})"
+                        )
                 # Log progress every 10 attempts
                 if attempt > 0 and attempt % 10 == 0:
                     logger.info(
@@ -215,7 +221,13 @@ class AgentService:
                 # Final cleanup attempt before raising error
                 await self.sandbox_service.cleanup_stuck_creating_sandboxes()
                 await self.sandbox_service.cleanup_expired_ready_sandboxes()
-                raise Exception("No sandbox available: pool limit reached")
+                (
+                    available_count,
+                    non_available_count,
+                ) = await self.sandbox_service.get_sandbox_counts()
+                raise Exception(
+                    f"No sandbox available: pool limit reached (available: {available_count}, non-available: {non_available_count}, max: {self.max_sandboxes})"
+                )
 
             data_dir = self.active_tasks[message_id].trace_path
             user_content = self.active_tasks[message_id].instruction

cua2-core/src/cua2_core/services/sandbox_service.py

@@ -363,6 +363,22 @@ class SandboxService:
 
         return len(expired_sandboxes_to_kill)
 
+    async def get_sandbox_counts(self) -> tuple[int, int]:
+        """
+        Get the count of available (ready) and non-available (creating) sandboxes.
+
+        Returns:
+            Tuple of (available_count, non_available_count)
+        """
+        async with self.sandbox_lock:
+            ready_count = len(self.sandboxes)
+            creating_count = sum(
+                1
+                for meta in self.sandbox_metadata.values()
+                if meta.get("state") == "creating"
+            )
+            return (ready_count, creating_count)
+
     async def cleanup_sandboxes(self):
         sandboxes_to_kill = []