from typing import Annotated, Literal
from typing_extensions import TypedDict
from langgraph.graph.message import add_messages
from langchain_core.messages import AnyMessage


class OrchestratorState(TypedDict):
    messages: Annotated[list[AnyMessage], add_messages]
    task: str
    route: Literal[
        "implementer",
        "reviewer",
        "researcher",
        "cross_reviewer",
        "scanner",
        "fast_coder",
        "connector",
        "done",
    ]
    result: str