app/state.py

"""
app/state.py — EpisodeState dataclass.
Holds the full state of one RL episode. Serialized to/from Redis.
"""
from dataclasses import dataclass, field
from typing import Any, Dict


@dataclass
class EpisodeState:
    task: Dict[str, Any]
    graph: Any          # CodeGraph instance
    step: int
    done: bool
    difficulty: str = "medium"