mcp-agent.py

import logging
from datetime import datetime

from dotenv import load_dotenv

from livekit.agents import Agent, AgentServer, AgentSession, JobContext, cli, inference, mcp
from livekit.agents.llm import ToolFlag

logger = logging.getLogger("mcp-agent")

load_dotenv()


class MyAgent(Agent):
    def __init__(self) -> None:
        super().__init__(
            instructions=(
                "You are a friendly travel assistant that communicates via voice. "
                "Avoid emojis and markdown — speak naturally and concisely. "
                "Use the get_weather tool when the user asks about the weather somewhere. "
                "Use the book_flight tool when the user wants to book a flight; it takes "
                "a couple of minutes and will report progress along the way — narrate the "
                "updates naturally and don't make up flight details. "
                "Don't repeat information you have already said in the conversation. "
                f"Today is {datetime.now().strftime('%Y-%m-%d %A')}"
            ),
        )

    async def on_enter(self):
        # when the agent is added to the session, it'll generate a reply
        # according to its instructions
        self.session.generate_reply(instructions="greeting the user and introducing yourself")


server = AgentServer()


@server.rtc_session()
async def entrypoint(ctx: JobContext):
    session = AgentSession(
        stt=inference.STT("deepgram/nova-3", language="multi"),
        llm=inference.LLM("openai/gpt-4.1-mini"),
        tts=inference.TTS("cartesia/sonic-3"),
        tools=[
            mcp.MCPToolset(
                id="mcp_toolset_1",
                mcp_server=mcp.MCPServerHTTP(
                    url="http://localhost:8000/sse",
                    # book_flight takes ~70s; bump the per-request timeout above that.
                    client_session_timeout_seconds=120,
                ),
                # book_flight is long-running: forward progress and allow cancellation
                tool_options={
                    "book_flight": {
                        "flags": ToolFlag.CANCELLABLE,
                        "on_duplicate": "confirm",
                        "report_progress": True,
                    },
                },
            )
        ],
    )

    await session.start(agent=MyAgent(), room=ctx.room)


if __name__ == "__main__":
    cli.run_app(server)