env/models.py

from dataclasses import dataclass, field
from typing import List


@dataclass
class Observation:
    code: str
    task: str
    history: List[str]
    task_id: int
    language: str = "javascript"
    difficulty: str = "easy"
    category: str = "syntax"


@dataclass
class Action:
    action_type: str  # "identify" or "fix"
    content: str      # explanation or corrected code


@dataclass
class EpisodeState:
    current_task_id: int
    current_task_difficulty: str
    current_task_category: str
    phase: str
    step: int
    total_reward: float
    done: bool
    history: List[str] = field(default_factory=list)