opentelemetry_mcp_client_instrumentation_quickstart_with_console_tracing.py

1from mcp import ClientSession, StdioServerParameters
2from opentelemetry.instrumentation.mcp import MCPInstrumentor
3from opentelemetry import trace
4from opentelemetry.sdk.trace import TracerProvider
5from opentelemetry.sdk.trace.export import (
6    BatchSpanProcessor,
7    ConsoleSpanExporter,
8)
9
10# 1. Setup OpenTelemetry (Standard Boilerplate)
11provider = TracerProvider()
12processor = BatchSpanProcessor(ConsoleSpanExporter())
13provider.add_span_processor(processor)
14trace.set_tracer_provider(provider)
15
16# 2. Initialize MCP Instrumentation
17# This will automatically instrument mcp.ClientSession and mcp.Server
18MCPInstrumentor().instrument()
19
20# 3. Use the MCP library as usual
21async def main():
22    server_params = StdioServerParameters(command="python", args=["server.py"])
23    async with ClientSession(server_params) as session:
24        # These calls will now generate OpenTelemetry spans
25        await session.initialize()
26        results = await session.list_tools()
27        print(results)
28
29if __name__ == "__main__":
30    import asyncio
31    asyncio.run(main())