Expose max_status_failures across wait helpers

Co-authored-by: Shri Sukhani <shrisukhani@users.noreply.github.com>

Author: cursoragent

README.md

@@ -107,6 +107,7 @@ These methods now support explicit polling controls:
 
 - `poll_interval_seconds` (default `2.0`)
 - `max_wait_seconds` (default `600.0`)
+- `max_status_failures` (default `5`)
 
 Example:
 

hyperbrowser/client/managers/async_manager/agents/browser_use.py

@@ -55,6 +55,7 @@ async def start_and_wait(
         params: StartBrowserUseTaskParams,
         poll_interval_seconds: float = 2.0,
         max_wait_seconds: Optional[float] = 600.0,
+        max_status_failures: int = POLLING_ATTEMPTS,
     ) -> BrowserUseTaskResponse:
         job_start_resp = await self.start(params)
         job_id = job_start_resp.job_id
@@ -68,6 +69,7 @@ async def start_and_wait(
             in {"completed", "failed", "stopped"},
             poll_interval_seconds=poll_interval_seconds,
             max_wait_seconds=max_wait_seconds,
+            max_status_failures=max_status_failures,
         )
         return await retry_operation_async(
             operation_name=f"Fetching browser-use task job {job_id}",

hyperbrowser/client/managers/async_manager/agents/claude_computer_use.py

@@ -49,6 +49,7 @@ async def start_and_wait(
         params: StartClaudeComputerUseTaskParams,
         poll_interval_seconds: float = 2.0,
         max_wait_seconds: Optional[float] = 600.0,
+        max_status_failures: int = POLLING_ATTEMPTS,
     ) -> ClaudeComputerUseTaskResponse:
         job_start_resp = await self.start(params)
         job_id = job_start_resp.job_id
@@ -62,6 +63,7 @@ async def start_and_wait(
             in {"completed", "failed", "stopped"},
             poll_interval_seconds=poll_interval_seconds,
             max_wait_seconds=max_wait_seconds,
+            max_status_failures=max_status_failures,
         )
         return await retry_operation_async(
             operation_name=f"Fetching Claude Computer Use task job {job_id}",

hyperbrowser/client/managers/async_manager/agents/cua.py

@@ -47,6 +47,7 @@ async def start_and_wait(
         params: StartCuaTaskParams,
         poll_interval_seconds: float = 2.0,
         max_wait_seconds: Optional[float] = 600.0,
+        max_status_failures: int = POLLING_ATTEMPTS,
     ) -> CuaTaskResponse:
         job_start_resp = await self.start(params)
         job_id = job_start_resp.job_id
@@ -60,6 +61,7 @@ async def start_and_wait(
             in {"completed", "failed", "stopped"},
             poll_interval_seconds=poll_interval_seconds,
             max_wait_seconds=max_wait_seconds,
+            max_status_failures=max_status_failures,
         )
         return await retry_operation_async(
             operation_name=f"Fetching CUA task job {job_id}",

hyperbrowser/client/managers/async_manager/agents/gemini_computer_use.py

@@ -49,6 +49,7 @@ async def start_and_wait(
         params: StartGeminiComputerUseTaskParams,
         poll_interval_seconds: float = 2.0,
         max_wait_seconds: Optional[float] = 600.0,
+        max_status_failures: int = POLLING_ATTEMPTS,
     ) -> GeminiComputerUseTaskResponse:
         job_start_resp = await self.start(params)
         job_id = job_start_resp.job_id
@@ -62,6 +63,7 @@ async def start_and_wait(
             in {"completed", "failed", "stopped"},
             poll_interval_seconds=poll_interval_seconds,
             max_wait_seconds=max_wait_seconds,
+            max_status_failures=max_status_failures,
         )
         return await retry_operation_async(
             operation_name=f"Fetching Gemini Computer Use task job {job_id}",

hyperbrowser/client/managers/async_manager/agents/hyper_agent.py

@@ -49,6 +49,7 @@ async def start_and_wait(
         params: StartHyperAgentTaskParams,
         poll_interval_seconds: float = 2.0,
         max_wait_seconds: Optional[float] = 600.0,
+        max_status_failures: int = POLLING_ATTEMPTS,
     ) -> HyperAgentTaskResponse:
         job_start_resp = await self.start(params)
         job_id = job_start_resp.job_id
@@ -62,6 +63,7 @@ async def start_and_wait(
             in {"completed", "failed", "stopped"},
             poll_interval_seconds=poll_interval_seconds,
             max_wait_seconds=max_wait_seconds,
+            max_status_failures=max_status_failures,
         )
         return await retry_operation_async(
             operation_name=f"Fetching HyperAgent task {job_id}",

hyperbrowser/client/managers/async_manager/crawl.py

@@ -49,6 +49,7 @@ async def start_and_wait(
         return_all_pages: bool = True,
         poll_interval_seconds: float = 2.0,
         max_wait_seconds: Optional[float] = 600.0,
+        max_status_failures: int = POLLING_ATTEMPTS,
     ) -> CrawlJobResponse:
         job_start_resp = await self.start(params)
         job_id = job_start_resp.job_id
@@ -61,6 +62,7 @@ async def start_and_wait(
             is_terminal_status=lambda status: status in {"completed", "failed"},
             poll_interval_seconds=poll_interval_seconds,
             max_wait_seconds=max_wait_seconds,
+            max_status_failures=max_status_failures,
         )
 
         if not return_all_pages:

hyperbrowser/client/managers/async_manager/extract.py

@@ -47,6 +47,7 @@ async def start_and_wait(
         params: StartExtractJobParams,
         poll_interval_seconds: float = 2.0,
         max_wait_seconds: Optional[float] = 600.0,
+        max_status_failures: int = POLLING_ATTEMPTS,
     ) -> ExtractJobResponse:
         job_start_resp = await self.start(params)
         job_id = job_start_resp.job_id
@@ -59,6 +60,7 @@ async def start_and_wait(
             is_terminal_status=lambda status: status in {"completed", "failed"},
             poll_interval_seconds=poll_interval_seconds,
             max_wait_seconds=max_wait_seconds,
+            max_status_failures=max_status_failures,
         )
         return await retry_operation_async(
             operation_name=f"Fetching extract job {job_id}",

hyperbrowser/client/managers/async_manager/scrape.py

@@ -55,6 +55,7 @@ async def start_and_wait(
         return_all_pages: bool = True,
         poll_interval_seconds: float = 2.0,
         max_wait_seconds: Optional[float] = 600.0,
+        max_status_failures: int = POLLING_ATTEMPTS,
     ) -> BatchScrapeJobResponse:
         job_start_resp = await self.start(params)
         job_id = job_start_resp.job_id
@@ -67,6 +68,7 @@ async def start_and_wait(
             is_terminal_status=lambda status: status in {"completed", "failed"},
             poll_interval_seconds=poll_interval_seconds,
             max_wait_seconds=max_wait_seconds,
+            max_status_failures=max_status_failures,
         )
 
         if not return_all_pages:
@@ -142,6 +144,7 @@ async def start_and_wait(
         params: StartScrapeJobParams,
         poll_interval_seconds: float = 2.0,
         max_wait_seconds: Optional[float] = 600.0,
+        max_status_failures: int = POLLING_ATTEMPTS,
     ) -> ScrapeJobResponse:
         job_start_resp = await self.start(params)
         job_id = job_start_resp.job_id
@@ -154,6 +157,7 @@ async def start_and_wait(
             is_terminal_status=lambda status: status in {"completed", "failed"},
             poll_interval_seconds=poll_interval_seconds,
             max_wait_seconds=max_wait_seconds,
+            max_status_failures=max_status_failures,
         )
         return await retry_operation_async(
             operation_name=f"Fetching scrape job {job_id}",

hyperbrowser/client/managers/async_manager/web/batch_fetch.py

@@ -57,6 +57,7 @@ async def start_and_wait(
         return_all_pages: bool = True,
         poll_interval_seconds: float = 2.0,
         max_wait_seconds: Optional[float] = 600.0,
+        max_status_failures: int = POLLING_ATTEMPTS,
     ) -> BatchFetchJobResponse:
         job_start_resp = await self.start(params)
         job_id = job_start_resp.job_id
@@ -69,6 +70,7 @@ async def start_and_wait(
             is_terminal_status=lambda status: status in {"completed", "failed"},
             poll_interval_seconds=poll_interval_seconds,
             max_wait_seconds=max_wait_seconds,
+            max_status_failures=max_status_failures,
         )
 
         if not return_all_pages: