config-debug-env / server /models.py
Deepikachintamreddy
fix: model defaults 0.01 not 0.0
c8197e5
Raw
History Blame Contribute Delete
1.21 kB
from typing import List, Optional
from openenv.core.env_server import Action, Observation, State
class ConfigDebugAction(Action):
"""Action: the agent submits a fixed configuration file"""
fixed_config: str # The corrected configuration file content
class ConfigDebugObservation(Observation):
"""Observation: what the agent sees.
Inherits done: bool, reward: Optional[float], metadata from Observation.
"""
broken_config: str = ""
ground_truth: str = ""
file_type: str = ""
error_message: str = ""
task_id: str = ""
task_description: str = ""
difficulty: str = ""
num_bugs: int = 0
bugs_found_so_far: int = 0
previous_reward: float = 0.01
class ConfigDebugState(State):
"""Full environment state.
Inherits episode_id: Optional[str], step_count: int from State.
"""
current_task_id: str = ""
current_step: int = 0
max_steps: int = 5
total_reward: float = 0.01
is_done: bool = False
tasks_completed: List[str] = []
tasks_remaining: List[str] = []
bugs_found_so_far: int = 0
current_error_message: Optional[str] = None
progress_ratio: float = 0.01
current_difficulty: Optional[str] = None