import logging
from dataclasses import dataclass, field
from typing import Dict, List
from livekit.agents import JobContext, WorkerOptions, cli
from livekit.agents.llm import function_tool
from livekit.agents.voice import Agent, AgentSession

@dataclass
class AgentData:
    user_name: str = ""
    responses: Dict[str, str] = field(default_factory=dict)

class SmallTalkAgent(Agent):
    def __init__(self, job_context: JobContext):
        super().__init__(instructions="Start with smalltalk and gather user info.")

    async def on_enter(self):
        await self.session.say("Hello! What's your name?")

    @function_tool
    async def gather_user_info(self, name: str):
        self.session.userdata.user_name = name
        return QuizAgent(self.session.job_context)

class QuizAgent(Agent):
    def __init__(self, job_context: JobContext):
        super().__init__(instructions="Create and display a quiz.")

    async def on_enter(self):
        await self.session.say("Let's start a quiz!")

    @function_tool
    async def quiz_completed(self):
        return StoryAgent(self.session.job_context, self.session.userdata.user_name)

class StoryAgent(Agent):
    def __init__(self, job_context: JobContext, user_name: str):
        super().__init__(instructions=f"Tell a story for {user_name}.")

    async def on_enter(self):
        await self.session.say(f"Once upon a time, {user_name}...")

async def entrypoint(ctx: JobContext):
    session = AgentSession()
    session.userdata = AgentData()
    await session.start(agent=SmallTalkAgent(ctx))

if __name__ == "__main__":
    cli.run_app(WorkerOptions(entrypoint_fnc=entrypoint))

output.audio_enabled

True

allow_interruptions

before_tts_cb