yashppawar's picture
Initial DiskPanic OpenEnv submission
569c142 verified
Raw History Blame Contribute Delete
1.16 kB
"""Pydantic Action and Observation models for DiskPanic."""
from typing import Optional
from pydantic import Field
from openenv.core import Action, Observation
class DiskPanicAction(Action):
"""An SRE agent's action: a single bash-like command."""
command: str = Field(
...,
description=(
"A bash-like command. Supported: ls, df, du, cat, rm, find, "
"sha256sum, systemctl (status/start/is-active), and "
"'echo \"content\" > /path/to/file' for writing files."
),
)
class DiskPanicObservation(Observation):
"""State of the simulated server after the last command."""
stdout: str = Field(default="", description="Output of the last command")
df_output: str = Field(default="", description="Current simulated `df -h /`")
service_status: str = Field(default="unknown", description="State of app.service")
task_id: str = Field(default="easy", description="Current task id (easy/medium/hard)")
step: int = Field(default=0, description="Step count in current episode")
last_error: Optional[str] = Field(default=None, description="Last command error, if any")