agent.py

from livekit import agents
from livekit.agents import AgentSession, Agent, RoomInputOptions
from livekit.plugins import (
    openai,
    rime,
    assemblyai,
    noise_cancellation,
    silero,
)
from livekit.plugins.rime import TTS as RimeTTS
from dotenv import load_dotenv
import os

load_dotenv()

class LoggingRimeTTS(RimeTTS):
    def synthesize(self, text, *args, **kwargs):
        print("🔊 [Rime TTS] Outgoing Audio:", text)
        return super().synthesize(text, *args, **kwargs)

class Assistant(Agent):
    def __init__(self) -> None:
        super().__init__(instructions="You are a helpful voice AI assistant.")

    async def on_user_turn_completed(self, chat_ctx, new_message):
        if new_message.text_content:
            print(f"👶 [User] User Transcript: {new_message.text_content}")

async def entrypoint(ctx: agents.JobContext):
    session = AgentSession(
        stt=assemblyai.STT(
            end_of_turn_confidence_threshold=0.7,
            min_end_of_turn_silence_when_confident=160,
            max_turn_silence=2400,
            format_turns=True,
        ),
        llm=openai.LLM.with_cerebras(
            model="llama-3.3-70b",
        ),
        tts=LoggingRimeTTS(model="mist",
            speaker="rainforest",
            speed_alpha=0.9,
            reduce_latency=True,
        ),
        vad=silero.VAD.load(),
        turn_detection="stt"
    )

    await session.start(
        room=ctx.room,
        agent=Assistant(),
        room_input_options=RoomInputOptions(
            # LiveKit Cloud enhanced noise cancellation
            # - If self-hosting, omit this parameter
            # - For telephony applications, use `BVCTelephony` for best results
            # noise_cancellation=noise_cancellation.BVC(), 
        ),
    )

    await ctx.connect()

    await session.generate_reply(
        instructions="Greet the user and offer your assistance."
    )


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