"""OpenEnv wire models. Actions and observations are OpenEnv's MCP types;
only the state is specific to this environment."""

from __future__ import annotations

from openenv.core.env_server.types import State
from pydantic import Field


class DesignState(State):
    task_id: str | None = None
    split: str | None = None
    mode: str = "reference"
    surface: str = "tools"
    skin: str = "figma"
    n_elements: int = 0
    done: bool = False
    reward: float | None = None
    editor_url: str | None = None
    grade: dict | None = Field(default=None, description="Score breakdown after submit")