This sample demonstrates how a "task mode" agent can act as a sub-agent to an LLM agent, effectively extracting structured data from a conversational flow.
The main agent (coordinator) delegates interactions to two sub-agents:
order_collector: A task agent that collects the user's food order (from a menu of Pizza, Burger, Salad) and returns a structured list of selected items as alist[OrderItem].payment_collector: A task agent that collects the user's credit card and CVV information, returning aPaymentInfoobject.
Once the tasks are completed, the coordinator automatically uses a place_order tool with the structured data returned by both agents.
I would like to order some food please.I want 2 pizzas and 1 salad.My credit card is 1234-5678-9012-3456 and my CVV is 123.
[ coordinator ] --(uses)--> [ place_order (tool) ]
/ \
v v
[ order_collector ] [ payment_collector ]
-
Define a sub-agent with
mode="task"and an output schema:order_collector = Agent( name="order_collector", mode="task", output_schema=list[OrderItem], ... )
-
Assign it to a parent agent and use it in the instruction to collect the information:
coordinator = Agent( sub_agents=[order_collector], instruction="Delegate using `order_collector`...", ... )