21 lines
493 B
Python
21 lines
493 B
Python
|
|
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
|