from pydantic import Field
from openenv.core.env_server.types import Action, Observation, State

class MazeAction(Action):
    direction: str = Field(..., description="up, down, left, or right")

class MazeObservation(Observation):
    position: list = Field(default=[], description="Agent's [row, col]")
    valid_moves: list = Field(default=[], description="List of valid directions")
    grid_view: str = Field(default="", description="String view of the maze")

class MazeState(State):
    maze: list = Field(default=[], description="Grid (1=open, 0=wall)")
    agent_pos: list = Field(default=[], description="Agent's [row, col]")
    goal_pos: list = Field(default=[], description="Goal [row, col]")