Harden Claude/Gemini/HyperAgent start param serialization

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

Author: cursoragent

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

@@ -6,6 +6,7 @@
     wait_for_job_result_async,
 )
 from ...response_utils import parse_response_model
+from .....exceptions import HyperbrowserError
 
 from .....models import (
     POLLING_ATTEMPTS,
@@ -24,9 +25,22 @@ def __init__(self, client):
     async def start(
         self, params: StartClaudeComputerUseTaskParams
     ) -> StartClaudeComputerUseTaskResponse:
+        try:
+            payload = params.model_dump(exclude_none=True, by_alias=True)
+        except HyperbrowserError:
+            raise
+        except Exception as exc:
+            raise HyperbrowserError(
+                "Failed to serialize Claude Computer Use start params",
+                original_error=exc,
+            ) from exc
+        if type(payload) is not dict:
+            raise HyperbrowserError(
+                "Failed to serialize Claude Computer Use start params"
+            )
         response = await self._client.transport.post(
             self._client._build_url("/task/claude-computer-use"),
-            data=params.model_dump(exclude_none=True, by_alias=True),
+            data=payload,
         )
         return parse_response_model(
             response.data,

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

@@ -6,6 +6,7 @@
     wait_for_job_result_async,
 )
 from ...response_utils import parse_response_model
+from .....exceptions import HyperbrowserError
 
 from .....models import (
     POLLING_ATTEMPTS,
@@ -24,9 +25,22 @@ def __init__(self, client):
     async def start(
         self, params: StartGeminiComputerUseTaskParams
     ) -> StartGeminiComputerUseTaskResponse:
+        try:
+            payload = params.model_dump(exclude_none=True, by_alias=True)
+        except HyperbrowserError:
+            raise
+        except Exception as exc:
+            raise HyperbrowserError(
+                "Failed to serialize Gemini Computer Use start params",
+                original_error=exc,
+            ) from exc
+        if type(payload) is not dict:
+            raise HyperbrowserError(
+                "Failed to serialize Gemini Computer Use start params"
+            )
         response = await self._client.transport.post(
             self._client._build_url("/task/gemini-computer-use"),
-            data=params.model_dump(exclude_none=True, by_alias=True),
+            data=payload,
         )
         return parse_response_model(
             response.data,

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

@@ -6,6 +6,7 @@
     wait_for_job_result_async,
 )
 from ...response_utils import parse_response_model
+from .....exceptions import HyperbrowserError
 
 from .....models import (
     POLLING_ATTEMPTS,
@@ -24,9 +25,20 @@ def __init__(self, client):
     async def start(
         self, params: StartHyperAgentTaskParams
     ) -> StartHyperAgentTaskResponse:
+        try:
+            payload = params.model_dump(exclude_none=True, by_alias=True)
+        except HyperbrowserError:
+            raise
+        except Exception as exc:
+            raise HyperbrowserError(
+                "Failed to serialize HyperAgent start params",
+                original_error=exc,
+            ) from exc
+        if type(payload) is not dict:
+            raise HyperbrowserError("Failed to serialize HyperAgent start params")
         response = await self._client.transport.post(
             self._client._build_url("/task/hyper-agent"),
-            data=params.model_dump(exclude_none=True, by_alias=True),
+            data=payload,
         )
         return parse_response_model(
             response.data,

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

@@ -2,6 +2,7 @@
 
 from ....polling import build_operation_name, ensure_started_job_id, wait_for_job_result
 from ...response_utils import parse_response_model
+from .....exceptions import HyperbrowserError
 
 from .....models import (
     POLLING_ATTEMPTS,
@@ -20,9 +21,22 @@ def __init__(self, client):
     def start(
         self, params: StartClaudeComputerUseTaskParams
     ) -> StartClaudeComputerUseTaskResponse:
+        try:
+            payload = params.model_dump(exclude_none=True, by_alias=True)
+        except HyperbrowserError:
+            raise
+        except Exception as exc:
+            raise HyperbrowserError(
+                "Failed to serialize Claude Computer Use start params",
+                original_error=exc,
+            ) from exc
+        if type(payload) is not dict:
+            raise HyperbrowserError(
+                "Failed to serialize Claude Computer Use start params"
+            )
         response = self._client.transport.post(
             self._client._build_url("/task/claude-computer-use"),
-            data=params.model_dump(exclude_none=True, by_alias=True),
+            data=payload,
         )
         return parse_response_model(
             response.data,

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

@@ -2,6 +2,7 @@
 
 from ....polling import build_operation_name, ensure_started_job_id, wait_for_job_result
 from ...response_utils import parse_response_model
+from .....exceptions import HyperbrowserError
 
 from .....models import (
     POLLING_ATTEMPTS,
@@ -20,9 +21,22 @@ def __init__(self, client):
     def start(
         self, params: StartGeminiComputerUseTaskParams
     ) -> StartGeminiComputerUseTaskResponse:
+        try:
+            payload = params.model_dump(exclude_none=True, by_alias=True)
+        except HyperbrowserError:
+            raise
+        except Exception as exc:
+            raise HyperbrowserError(
+                "Failed to serialize Gemini Computer Use start params",
+                original_error=exc,
+            ) from exc
+        if type(payload) is not dict:
+            raise HyperbrowserError(
+                "Failed to serialize Gemini Computer Use start params"
+            )
         response = self._client.transport.post(
             self._client._build_url("/task/gemini-computer-use"),
-            data=params.model_dump(exclude_none=True, by_alias=True),
+            data=payload,
         )
         return parse_response_model(
             response.data,

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

@@ -2,6 +2,7 @@
 
 from ....polling import build_operation_name, ensure_started_job_id, wait_for_job_result
 from ...response_utils import parse_response_model
+from .....exceptions import HyperbrowserError
 
 from .....models import (
     POLLING_ATTEMPTS,
@@ -18,9 +19,20 @@ def __init__(self, client):
         self._client = client
 
     def start(self, params: StartHyperAgentTaskParams) -> StartHyperAgentTaskResponse:
+        try:
+            payload = params.model_dump(exclude_none=True, by_alias=True)
+        except HyperbrowserError:
+            raise
+        except Exception as exc:
+            raise HyperbrowserError(
+                "Failed to serialize HyperAgent start params",
+                original_error=exc,
+            ) from exc
+        if type(payload) is not dict:
+            raise HyperbrowserError("Failed to serialize HyperAgent start params")
         response = self._client.transport.post(
             self._client._build_url("/task/hyper-agent"),
-            data=params.model_dump(exclude_none=True, by_alias=True),
+            data=payload,
         )
         return parse_response_model(
             response.data,