from pydantic import BaseModel
class Triage(BaseModel):
ambiguous: bool
follow_up_details: str
reply_to_user: str
def triage_handler(output: Triage):
if output.ambiguous:
return ("STOP", None)
return ("CONTINUE", output.follow_up_details)
wf = (
client.Workflow()
.add(triage_agent, response_model=Triage, handler=triage_handler)
.add(deep_dive_agent)
)