Centralize sync and async agent task reads

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

Author: cursoragent

CONTRIBUTING.md

@@ -79,6 +79,7 @@ This runs lint, format checks, compile checks, tests, and package build.
   - `tests/test_agent_examples_coverage.py` (agent task example coverage enforcement),
   - `tests/test_agent_payload_helper_usage.py` (shared agent start-payload helper usage enforcement),
   - `tests/test_agent_stop_helper_usage.py` (shared agent stop-request helper usage enforcement),
+  - `tests/test_agent_task_read_helper_usage.py` (shared agent task read-helper usage enforcement),
   - `tests/test_agent_terminal_status_helper_usage.py` (shared agent terminal-status helper usage enforcement),
   - `tests/test_architecture_marker_usage.py` (architecture marker coverage across guard modules),
   - `tests/test_binary_file_open_helper_usage.py` (shared binary file open helper usage enforcement),

hyperbrowser/client/managers/agent_task_read_utils.py

@@ -0,0 +1,77 @@
+from typing import Any, Type, TypeVar
+
+from .response_utils import parse_response_model
+
+T = TypeVar("T")
+
+
+def get_agent_task(
+    *,
+    client: Any,
+    route_prefix: str,
+    job_id: str,
+    model: Type[T],
+    operation_name: str,
+) -> T:
+    response = client.transport.get(
+        client._build_url(f"{route_prefix}/{job_id}"),
+    )
+    return parse_response_model(
+        response.data,
+        model=model,
+        operation_name=operation_name,
+    )
+
+
+def get_agent_task_status(
+    *,
+    client: Any,
+    route_prefix: str,
+    job_id: str,
+    model: Type[T],
+    operation_name: str,
+) -> T:
+    response = client.transport.get(
+        client._build_url(f"{route_prefix}/{job_id}/status"),
+    )
+    return parse_response_model(
+        response.data,
+        model=model,
+        operation_name=operation_name,
+    )
+
+
+async def get_agent_task_async(
+    *,
+    client: Any,
+    route_prefix: str,
+    job_id: str,
+    model: Type[T],
+    operation_name: str,
+) -> T:
+    response = await client.transport.get(
+        client._build_url(f"{route_prefix}/{job_id}"),
+    )
+    return parse_response_model(
+        response.data,
+        model=model,
+        operation_name=operation_name,
+    )
+
+
+async def get_agent_task_status_async(
+    *,
+    client: Any,
+    route_prefix: str,
+    job_id: str,
+    model: Type[T],
+    operation_name: str,
+) -> T:
+    response = await client.transport.get(
+        client._build_url(f"{route_prefix}/{job_id}/status"),
+    )
+    return parse_response_model(
+        response.data,
+        model=model,
+        operation_name=operation_name,
+    )

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

@@ -3,6 +3,7 @@
 from ...agent_status_utils import is_agent_terminal_status
 from ...browser_use_payload_utils import build_browser_use_start_payload
 from ...agent_stop_utils import stop_agent_task_async
+from ...agent_task_read_utils import get_agent_task_async, get_agent_task_status_async
 from ...job_wait_utils import wait_for_job_result_with_defaults_async
 from ...response_utils import parse_response_model
 from ...start_job_utils import build_started_job_context
@@ -40,21 +41,19 @@ async def start(
         )
 
     async def get(self, job_id: str) -> BrowserUseTaskResponse:
-        response = await self._client.transport.get(
-            self._client._build_url(f"/task/browser-use/{job_id}")
-        )
-        return parse_response_model(
-            response.data,
+        return await get_agent_task_async(
+            client=self._client,
+            route_prefix="/task/browser-use",
+            job_id=job_id,
             model=BrowserUseTaskResponse,
             operation_name="browser-use task",
         )
 
     async def get_status(self, job_id: str) -> BrowserUseTaskStatusResponse:
-        response = await self._client.transport.get(
-            self._client._build_url(f"/task/browser-use/{job_id}/status")
-        )
-        return parse_response_model(
-            response.data,
+        return await get_agent_task_status_async(
+            client=self._client,
+            route_prefix="/task/browser-use",
+            job_id=job_id,
             model=BrowserUseTaskStatusResponse,
             operation_name="browser-use task status",
         )

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

@@ -3,6 +3,7 @@
 from ...agent_payload_utils import build_agent_start_payload
 from ...agent_status_utils import is_agent_terminal_status
 from ...agent_stop_utils import stop_agent_task_async
+from ...agent_task_read_utils import get_agent_task_async, get_agent_task_status_async
 from ...job_wait_utils import wait_for_job_result_with_defaults_async
 from ...response_utils import parse_response_model
 from ...start_job_utils import build_started_job_context
@@ -43,21 +44,19 @@ async def start(
         )
 
     async def get(self, job_id: str) -> ClaudeComputerUseTaskResponse:
-        response = await self._client.transport.get(
-            self._client._build_url(f"/task/claude-computer-use/{job_id}")
-        )
-        return parse_response_model(
-            response.data,
+        return await get_agent_task_async(
+            client=self._client,
+            route_prefix="/task/claude-computer-use",
+            job_id=job_id,
             model=ClaudeComputerUseTaskResponse,
             operation_name="claude computer use task",
         )
 
     async def get_status(self, job_id: str) -> ClaudeComputerUseTaskStatusResponse:
-        response = await self._client.transport.get(
-            self._client._build_url(f"/task/claude-computer-use/{job_id}/status")
-        )
-        return parse_response_model(
-            response.data,
+        return await get_agent_task_status_async(
+            client=self._client,
+            route_prefix="/task/claude-computer-use",
+            job_id=job_id,
             model=ClaudeComputerUseTaskStatusResponse,
             operation_name="claude computer use task status",
         )

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

@@ -3,6 +3,7 @@
 from ...agent_payload_utils import build_agent_start_payload
 from ...agent_status_utils import is_agent_terminal_status
 from ...agent_stop_utils import stop_agent_task_async
+from ...agent_task_read_utils import get_agent_task_async, get_agent_task_status_async
 from ...job_wait_utils import wait_for_job_result_with_defaults_async
 from ...response_utils import parse_response_model
 from ...start_job_utils import build_started_job_context
@@ -41,21 +42,19 @@ async def start(self, params: StartCuaTaskParams) -> StartCuaTaskResponse:
         )
 
     async def get(self, job_id: str) -> CuaTaskResponse:
-        response = await self._client.transport.get(
-            self._client._build_url(f"/task/cua/{job_id}")
-        )
-        return parse_response_model(
-            response.data,
+        return await get_agent_task_async(
+            client=self._client,
+            route_prefix="/task/cua",
+            job_id=job_id,
             model=CuaTaskResponse,
             operation_name="cua task",
         )
 
     async def get_status(self, job_id: str) -> CuaTaskStatusResponse:
-        response = await self._client.transport.get(
-            self._client._build_url(f"/task/cua/{job_id}/status")
-        )
-        return parse_response_model(
-            response.data,
+        return await get_agent_task_status_async(
+            client=self._client,
+            route_prefix="/task/cua",
+            job_id=job_id,
             model=CuaTaskStatusResponse,
             operation_name="cua task status",
         )

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

@@ -3,6 +3,7 @@
 from ...agent_payload_utils import build_agent_start_payload
 from ...agent_status_utils import is_agent_terminal_status
 from ...agent_stop_utils import stop_agent_task_async
+from ...agent_task_read_utils import get_agent_task_async, get_agent_task_status_async
 from ...job_wait_utils import wait_for_job_result_with_defaults_async
 from ...response_utils import parse_response_model
 from ...start_job_utils import build_started_job_context
@@ -43,21 +44,19 @@ async def start(
         )
 
     async def get(self, job_id: str) -> GeminiComputerUseTaskResponse:
-        response = await self._client.transport.get(
-            self._client._build_url(f"/task/gemini-computer-use/{job_id}")
-        )
-        return parse_response_model(
-            response.data,
+        return await get_agent_task_async(
+            client=self._client,
+            route_prefix="/task/gemini-computer-use",
+            job_id=job_id,
             model=GeminiComputerUseTaskResponse,
             operation_name="gemini computer use task",
         )
 
     async def get_status(self, job_id: str) -> GeminiComputerUseTaskStatusResponse:
-        response = await self._client.transport.get(
-            self._client._build_url(f"/task/gemini-computer-use/{job_id}/status")
-        )
-        return parse_response_model(
-            response.data,
+        return await get_agent_task_status_async(
+            client=self._client,
+            route_prefix="/task/gemini-computer-use",
+            job_id=job_id,
             model=GeminiComputerUseTaskStatusResponse,
             operation_name="gemini computer use task status",
         )

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

@@ -3,6 +3,7 @@
 from ...agent_payload_utils import build_agent_start_payload
 from ...agent_status_utils import is_agent_terminal_status
 from ...agent_stop_utils import stop_agent_task_async
+from ...agent_task_read_utils import get_agent_task_async, get_agent_task_status_async
 from ...job_wait_utils import wait_for_job_result_with_defaults_async
 from ...response_utils import parse_response_model
 from ...start_job_utils import build_started_job_context
@@ -43,21 +44,19 @@ async def start(
         )
 
     async def get(self, job_id: str) -> HyperAgentTaskResponse:
-        response = await self._client.transport.get(
-            self._client._build_url(f"/task/hyper-agent/{job_id}")
-        )
-        return parse_response_model(
-            response.data,
+        return await get_agent_task_async(
+            client=self._client,
+            route_prefix="/task/hyper-agent",
+            job_id=job_id,
             model=HyperAgentTaskResponse,
             operation_name="hyper agent task",
         )
 
     async def get_status(self, job_id: str) -> HyperAgentTaskStatusResponse:
-        response = await self._client.transport.get(
-            self._client._build_url(f"/task/hyper-agent/{job_id}/status")
-        )
-        return parse_response_model(
-            response.data,
+        return await get_agent_task_status_async(
+            client=self._client,
+            route_prefix="/task/hyper-agent",
+            job_id=job_id,
             model=HyperAgentTaskStatusResponse,
             operation_name="hyper agent task status",
         )

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

@@ -3,6 +3,7 @@
 from ...agent_status_utils import is_agent_terminal_status
 from ...browser_use_payload_utils import build_browser_use_start_payload
 from ...agent_stop_utils import stop_agent_task
+from ...agent_task_read_utils import get_agent_task, get_agent_task_status
 from ...job_wait_utils import wait_for_job_result_with_defaults
 from ...response_utils import parse_response_model
 from ...start_job_utils import build_started_job_context
@@ -38,21 +39,19 @@ def start(self, params: StartBrowserUseTaskParams) -> StartBrowserUseTaskRespons
         )
 
     def get(self, job_id: str) -> BrowserUseTaskResponse:
-        response = self._client.transport.get(
-            self._client._build_url(f"/task/browser-use/{job_id}")
-        )
-        return parse_response_model(
-            response.data,
+        return get_agent_task(
+            client=self._client,
+            route_prefix="/task/browser-use",
+            job_id=job_id,
             model=BrowserUseTaskResponse,
             operation_name="browser-use task",
         )
 
     def get_status(self, job_id: str) -> BrowserUseTaskStatusResponse:
-        response = self._client.transport.get(
-            self._client._build_url(f"/task/browser-use/{job_id}/status")
-        )
-        return parse_response_model(
-            response.data,
+        return get_agent_task_status(
+            client=self._client,
+            route_prefix="/task/browser-use",
+            job_id=job_id,
             model=BrowserUseTaskStatusResponse,
             operation_name="browser-use task status",
         )

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

@@ -3,6 +3,7 @@
 from ...agent_payload_utils import build_agent_start_payload
 from ...agent_status_utils import is_agent_terminal_status
 from ...agent_stop_utils import stop_agent_task
+from ...agent_task_read_utils import get_agent_task, get_agent_task_status
 from ...job_wait_utils import wait_for_job_result_with_defaults
 from ...response_utils import parse_response_model
 from ...start_job_utils import build_started_job_context
@@ -43,21 +44,19 @@ def start(
         )
 
     def get(self, job_id: str) -> ClaudeComputerUseTaskResponse:
-        response = self._client.transport.get(
-            self._client._build_url(f"/task/claude-computer-use/{job_id}")
-        )
-        return parse_response_model(
-            response.data,
+        return get_agent_task(
+            client=self._client,
+            route_prefix="/task/claude-computer-use",
+            job_id=job_id,
             model=ClaudeComputerUseTaskResponse,
             operation_name="claude computer use task",
         )
 
     def get_status(self, job_id: str) -> ClaudeComputerUseTaskStatusResponse:
-        response = self._client.transport.get(
-            self._client._build_url(f"/task/claude-computer-use/{job_id}/status")
-        )
-        return parse_response_model(
-            response.data,
+        return get_agent_task_status(
+            client=self._client,
+            route_prefix="/task/claude-computer-use",
+            job_id=job_id,
             model=ClaudeComputerUseTaskStatusResponse,
             operation_name="claude computer use task status",
         )

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

@@ -3,6 +3,7 @@
 from ...agent_payload_utils import build_agent_start_payload
 from ...agent_status_utils import is_agent_terminal_status
 from ...agent_stop_utils import stop_agent_task
+from ...agent_task_read_utils import get_agent_task, get_agent_task_status
 from ...job_wait_utils import wait_for_job_result_with_defaults
 from ...response_utils import parse_response_model
 from ...start_job_utils import build_started_job_context
@@ -41,21 +42,19 @@ def start(self, params: StartCuaTaskParams) -> StartCuaTaskResponse:
         )
 
     def get(self, job_id: str) -> CuaTaskResponse:
-        response = self._client.transport.get(
-            self._client._build_url(f"/task/cua/{job_id}")
-        )
-        return parse_response_model(
-            response.data,
+        return get_agent_task(
+            client=self._client,
+            route_prefix="/task/cua",
+            job_id=job_id,
             model=CuaTaskResponse,
             operation_name="cua task",
         )
 
     def get_status(self, job_id: str) -> CuaTaskStatusResponse:
-        response = self._client.transport.get(
-            self._client._build_url(f"/task/cua/{job_id}/status")
-        )
-        return parse_response_model(
-            response.data,
+        return get_agent_task_status(
+            client=self._client,
+            route_prefix="/task/cua",
+            job_id=job_id,
             model=CuaTaskStatusResponse,
             operation_name="cua task status",
         )