Update cua2-core/src/cua2_core/models/models.py

cua2-core/src/cua2_core/models/models.py

@@ -137,7 +137,7 @@ class AgentTraceMetadata(BaseModel):
     maxSteps: int = 0
     completed: bool = False
     final_state: (
-        Literal["success", "stopped", "max_steps_reached", "error", "sandbox_timeout"]
+        Literal["success", "stopped", "max_steps_reached", "error", "sandbox_timeout", "invalid_prompt"]
         | None
     ) = None
     user_evaluation: Literal["success", "failed", "not_evaluated"] = "not_evaluated"
@@ -189,7 +189,7 @@ class AgentCompleteEvent(BaseModel):
     type: Literal["agent_complete"] = "agent_complete"
     traceMetadata: AgentTraceMetadata
     final_state: Literal[
-        "success", "stopped", "max_steps_reached", "error", "sandbox_timeout"
+        "success", "stopped", "max_steps_reached", "error", "sandbox_timeout", "invalid_prompt"
     ]
 
 
@@ -328,7 +328,7 @@ class ActiveTask(BaseModel):
         step_numberOfSteps: int | None = None,
         completed: bool | None = None,
         final_state: Literal[
-            "success", "stopped", "max_steps_reached", "error", "sandbox_timeout"
+            "success", "stopped", "max_steps_reached", "error", "sandbox_timeout", "invalid_prompt"
         ]
         | None = None,
         user_evaluation: Literal["success", "failed", "not_evaluated"] | None = None,