from praisonaiagents import Agent, AgentFlow
from praisonaiagents import route
# Classifier decides where to route
def classify(ctx):
text = ctx.input.lower()
if "api" in text or "code" in text:
return StepResult(output="technical")
elif "price" in text or "bill" in text:
return StepResult(output="billing")
return StepResult(output="general")
# Specialist agents
tech = Agent(name="Tech", instructions="Handle technical questions")
billing = Agent(name="Billing", instructions="Handle billing questions")
general = Agent(name="General", instructions="Handle general questions")
# Create routing workflow
flow = AgentFlow(steps=[
classify,
route({
"technical": [tech],
"billing": [billing],
"general": [general],
"default": [general]
})
])
result = flow.start("How do I integrate the API?")