Get Started with LiveKit

import logging
import os
from dotenv import load_dotenv
from livekit import agents, api
from livekit.agents import AgentSession, Agent, RoomInputOptions
from livekit.plugins import openai, silero

load_dotenv()

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger("voice-agent")

CEREBRAS_URL = "https://api.cerebras.ai/v1"
CEREBRAS_API_KEY = os.getenv("CEREBRAS_API_KEY")
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")

LIVEKIT_API_KEY = os.getenv("LIVEKIT_API_KEY")
LIVEKIT_API_SECRET = os.getenv("LIVEKIT_API_SECRET")
LIVEKIT_URL = os.environ["LIVEKIT_URL"]

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

# Initialize OpenAI STT (Speech-to-Text) using Whisper
stt = openai.STT(
    model="whisper-1",
    api_key=OPENAI_API_KEY
)

# Initialize Cerebras LLM for ultra-fast conversation
llm = openai.LLM(
    model="gpt-oss-120b",
    api_key=CEREBRAS_API_KEY,
    base_url=CEREBRAS_URL,
    extra_headers={"X-Cerebras-3rd-Party-Integration": "livekit"}
)

async def entrypoint(ctx: agents.JobContext):
    logger.info(f"Starting agent in room {ctx.room.name}")
    session = AgentSession(
        stt=stt,
        llm=llm,
        tts=openai.TTS(), # Use OpenAI TTS
        vad=silero.VAD.load(),
    )
    await session.start(
        room=ctx.room,
        agent=Assistant(),
    )
    await session.generate_reply(
        instructions="Greet the user and offer your assistance."
    )

if __name__ == "__main__":
    # Generate and display token
    if not LIVEKIT_API_KEY or not LIVEKIT_API_SECRET:
        print("Missing LIVEKIT_API_KEY or LIVEKIT_API_SECRET in .env file")
        print("Get these from your LiveKit Cloud dashboard: https://cloud.livekit.io/")
    else:
        token = api.AccessToken(LIVEKIT_API_KEY, LIVEKIT_API_SECRET) \
            .with_identity("test_user") \
            .with_grants(api.VideoGrants(
                room_join=True,
                room="test_room",
            ))
        
        jwt_token = token.to_jwt()
        
    print("\n\nLiveKit Agent Ready to Connect!\033[0m\n\033[94m" + "="*50 + "\033[0m")
    print(f"\033[94mConnect at: https://agents-playground.livekit.io/\033[0m\n\033[94m")
    print(f"\033[94mURL: {LIVEKIT_URL}\033[0m")
    print(f"\033[94mToken: {jwt_token}\033[0m" + "="*50 + "\033[0m")
    
    agents.cli.run_app(agents.WorkerOptions(entrypoint_fnc=entrypoint))