Update models.py (#14)

- Update models.py (abda8799ab189bfc10663219a2c24ccfa6a9b6b3)


Co-authored-by: tarun saroya <Tarun-sar0ya@users.noreply.huggingface.co>

models.py

@@ -0,0 +1,31 @@
+"""Models for the API Triage Agent environment.
+
+Defines the Action, Observation, and State data structures
+used by the OpenEnv client-server interface.
+"""
+
+from pydantic import BaseModel
+from typing import List
+
+
+class Action(BaseModel):
+    """An action that the agent can take in the environment."""
+    action_name: str
+
+
+class Observation(BaseModel):
+    """What the agent observes at each step."""
+    step: int
+    max_steps: int
+    incident_summary: str
+    logs: List[str]
+    response_code: int
+    fix_applied: bool
+    is_resolved: bool
+
+
+class State(BaseModel):
+    """Internal episode state tracking."""
+    episode_id: str = ""
+    step_count: int = 0
+    is_done: bool = False