oai-agent_mcp.py

import os
import asyncio

import logfire
from dotenv import load_dotenv
from agents import Agent, Runner
from agents.mcp import MCPServerStdio

load_dotenv()

# Configure Logfire
logfire.configure(send_to_logfire="if-token-present")
logfire.instrument_mcp()
logfire.instrument_openai_agents()


async def main(query: str = "Greet Andrew and give him the current time") -> None:
    """
    Main function to run the agent

    Args:
        query (str): The query to run the agent with
    """
    # Create and use the MCP server in an async context
    async with MCPServerStdio(
        params={
            "command": "uv",
            "args": ["run", "run_server.py", "stdio"],
        }
    ) as server:
        # Initialise the agent with the server
        agent = Agent(
            name="MCP agent",
            model="o4-mini",
            mcp_servers=[server],
        )

        result = await Runner.run(
            starting_agent=agent,
            input=query,
        )

        print(result.final_output)


if __name__ == "__main__":
    asyncio.run(main())