state.py

from typing import TypedDict, List, Optional, Any


class AgentState(TypedDict):
    user_input: str
    task: str
    steps: List[str]

    uploaded_files: List[str]
    current_image: Optional[str]

    result: Any
    history: List[dict]