models.py

"""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