Fix GrokComputerUseStepResponse xAI-style step field types

Align created_at, completed_at, and error with CuaStepResponse so
completed Grok task payloads parse without Pydantic validation errors.
Reuse CUA step helper models for structured error, incomplete_details,
and reasoning fields.

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

Author: cursoragent

hyperbrowser/models/agents/grok_computer_use.py

@@ -2,6 +2,7 @@
 
 from pydantic import BaseModel, ConfigDict, Field
 
+from .cua import CuaStepIncompleteDetails, CuaStepReasoning, CuaStepResponseError
 from ..session import CreateSessionParams
 from ..consts import GrokComputerUseLlm, GrokReasoningEffort
 
@@ -90,18 +91,22 @@ class GrokComputerUseStepResponse(BaseModel):
         populate_by_alias=True,
     )
 
-    created_at: Optional[str] = Field(default=None, serialization_alias="created_at")
-    completed_at: Optional[str] = Field(
+    created_at: Optional[int] = Field(default=None, serialization_alias="created_at")
+    completed_at: Optional[int] = Field(
         default=None, serialization_alias="completed_at"
     )
     output_text: Optional[str] = Field(default=None, serialization_alias="output_text")
-    error: Optional[str] = Field(default=None, serialization_alias="error")
-    incomplete_details: Optional[Any] = Field(
+    error: Optional[CuaStepResponseError] = Field(
+        default=None, serialization_alias="error"
+    )
+    incomplete_details: Optional[CuaStepIncompleteDetails] = Field(
         default=None, serialization_alias="incomplete_details"
     )
     model: Optional[str] = Field(default=None, serialization_alias="model")
     output: Optional[list[Any]] = Field(default=None, serialization_alias="output")
-    reasoning: Optional[Any] = Field(default=None, serialization_alias="reasoning")
+    reasoning: Optional[CuaStepReasoning] = Field(
+        default=None, serialization_alias="reasoning"
+    )
     status: Optional[str] = Field(default=None, serialization_alias="status")
 
 

tests/test_grok_computer_use_models.py

@@ -0,0 +1,37 @@
+from hyperbrowser.models.agents.grok_computer_use import GrokComputerUseTaskResponse
+
+
+def test_grok_computer_use_task_response_parses_xai_style_step_payload() -> None:
+    payload = {
+        "jobId": "job-123",
+        "status": "completed",
+        "data": {
+            "steps": [
+                {
+                    "created_at": 1710000000,
+                    "completed_at": 1710000001,
+                    "output_text": "done",
+                    "error": {"code": "ERR", "message": "something failed"},
+                    "incomplete_details": {"reason": "timeout"},
+                    "model": "grok-4.5",
+                    "output": [],
+                    "reasoning": {"effort": "high"},
+                    "status": "completed",
+                }
+            ],
+            "finalResult": "result",
+        },
+    }
+
+    result = GrokComputerUseTaskResponse(**payload)
+
+    step = result.data.steps[0]
+    assert step.created_at == 1710000000
+    assert step.completed_at == 1710000001
+    assert step.error is not None
+    assert step.error.code == "ERR"
+    assert step.error.message == "something failed"
+    assert step.incomplete_details is not None
+    assert step.incomplete_details.reason == "timeout"
+    assert step.reasoning is not None
+    assert step.reasoning.effort == "high"