Spaces:
Running
Running
File size: 711 Bytes
2bf281f 96e57e5 37f9abc 0707f61 96e57e5 0707f61 2bf281f 96e57e5 37f9abc b40eaae | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | from typing import Literal
from langchain_core.messages import AIMessage
from multi_agent_sdlc.models import CoderStatus
from multi_agent_sdlc.state import DevState
def route_after_coder(
state: DevState,
) -> Literal["coder_tools", "prepare_tester", "coder"]:
coder_status = state.get("coder_status")
if coder_status is CoderStatus.COMPLETED:
return "prepare_tester"
# if coder_status in {CoderStatus.BLOCKED, CoderStatus.FAILED}:
# return "human_intervention" # future
messages = state.get("coder_messages", [])
last_message = messages[-1]
if isinstance(last_message, AIMessage) and last_message.tool_calls:
return "coder_tools"
return "coder"
|