from typing import TypedDict, Optional
from src.models import SandboxResult


class SandboxState(TypedDict):
    user_prompt: str
    script: Optional[str]
    requirements: list[str]
    explanation: Optional[str]
    sandbox_result: Optional[SandboxResult]
    error_analysis: Optional[str]
    correction_strategy: Optional[str]
    retry_count: int
    max_retries: int
    trace: list[dict]
    final_output: Optional[str]
    final_error: Optional[str]
    files: dict[str, str]