from livekit.agents import Agent, AgentSession, function_tool, RunContext
from livekit.plugins import xai

class SupportAgent(Agent):
    def __init__(self):
        super().__init__(instructions="You are a technical support specialist.")

class BillingAgent(Agent):
    def __init__(self):
        super().__init__(instructions="You are a billing specialist.")

class TriageAgent(Agent):
    def __init__(self):
        super().__init__(instructions="You are a triage agent using Grok voice. Route callers to the right team.")

    @function_tool()
    async def handoff_to_support(self, context: RunContext):
        """Return an Agent instance to trigger a handoff to Support."""
        return SupportAgent(), "Handoff: connecting you to technical support."

    @function_tool()
    async def handoff_to_billing(self, context: RunContext):
        """Return an Agent instance to trigger a handoff to Billing."""
        return BillingAgent(), "Handoff: connecting you to billing."

# Create the session with the Grok realtime voice model
session = AgentSession(
    llm=xai.realtime.RealtimeModel(voice="Ara"),  # Grok voice model via xai plugin
    agent=TriageAgent()
)

# Run the session according to your app's realtime/telephony wiring
# e.g., await session.run(connection)

xai.realtime.RealtimeModel()