Function main

examples/mcp_tool_runner.py:22–51

()

async def main() -> None:
    # Connect to a local MCP server via stdio
    # This example uses the MCP filesystem server; replace with your own server
    server_params = StdioServerParameters(
        command="npx",
        args=["-y", "@modelcontextprotocol/server-filesystem", "/tmp"],
    )

    async with stdio_client(server_params) as (read, write):
        async with ClientSession(read, write) as mcp_client:
            await mcp_client.initialize()

            # List available tools from the MCP server and convert them
            tools_result = await mcp_client.list_tools()
            tools = [async_mcp_tool(t, mcp_client) for t in tools_result.tools]

            print(f"Connected to MCP server with {len(tools)} tools:")
            for tool in tools:
                print(f"  - {tool.name}")
            print()

            # Run a conversation with tool_runner()
            runner = client.beta.messages.tool_runner(
                model="claude-sonnet-5",
                max_tokens=1024,
                tools=tools,
                messages=[{"role": "user", "content": "List the files in /tmp"}],
            )
            async for message in runner:
                rich.print(message)


asyncio.run(main())