Subagents

from zeroruntime import SubAgent, function_tool


class Reception(SubAgent):
    name = "reception"
    instructions = "Find out what the caller needs, then hand them over."

    @function_tool
    async def to_billing(self, summary: str) -> dict:
        """Hand the caller to billing.

        Args:
            summary: What they need, in a few words.
        """
        return await self.transfer_to("billing", reason=summary)


class Billing(SubAgent):
    name = "billing"
    instructions = "You handle invoices."

import asyncio

from zeroruntime import Agent, Pipeline
from zeroruntime.inference import TurnDetector
from zeroruntime.plugins import CartesiaTTS, DeepgramSTT, GoogleLLM, SileroVAD


class Support(Agent):
    def __init__(self):
        super().__init__(
            instructions="You answer the phone at Nimbus Telecom.",
            pipeline=Pipeline(
                stt=DeepgramSTT(),
                llm=GoogleLLM(),
                tts=CartesiaTTS(),
                vad=SileroVAD(),
                turn_detector=TurnDetector(),
            ),
            subagents=[Reception(), Billing()],
        )

    @function_tool
    async def end_call(self) -> dict:
        """End the call once the caller has nothing else."""
        asyncio.create_task(self.hangup(reason="caller done"))
        return {"status": "ending_call"}

class Billing(SubAgent):
    name = "billing"
    instructions = "You handle invoices."

    async def on_enter(self, reason: str):
        return f"Say you can help with this: {reason}."

    async def on_exit(self):
        return "Thanks for calling Nimbus Telecom."