idZddlmZddlmZddlmZmZddlm Z ddl m Z m Z m Z mZddl mZdZd Zd Zdd ed zd efdZ ddeeefd zd eeefd zfdZde ded d fdZdeeefd zd e fdZgdZy )aOpenTelemetry instrumentation for FastMCP. This module provides native OpenTelemetry integration for FastMCP servers and clients. It uses only the opentelemetry-api package, so telemetry is a no-op unless the user installs an OpenTelemetry SDK and configures exporters. Example usage with SDK: ```python from opentelemetry import trace from opentelemetry.sdk.trace import TracerProvider from opentelemetry.sdk.trace.export import ConsoleSpanExporter, SimpleSpanProcessor # Configure the SDK (user responsibility) provider = TracerProvider() provider.add_span_processor(SimpleSpanProcessor(ConsoleSpanExporter())) trace.set_tracer_provider(provider) # Now FastMCP will emit traces from fastmcp import FastMCP mcp = FastMCP("my-server") ``` )Any)context) propagatetrace)Context)SpanStatus StatusCodeTracer) get_tracerfastmcp traceparent tracestateNversionreturnc"tt|S)zGet the FastMCP tracer for creating spans. Args: version: Optional version string for the instrumentation Returns: A tracer instance. Returns a no-op tracer if no SDK is configured. )otel_get_tracerINSTRUMENTATION_NAME)rs M/mnt/e/genesis-system/.venv/lib/python3.12/site-packages/fastmcp/telemetry.pyr r &s / 99metaci}tj|i}d|vr |d|t<d|vr |d|t<|r i|xsi|S|S)a@Inject current trace context into a meta dict for MCP request propagation. Args: meta: Optional existing meta dict to merge with trace context Returns: A new dict containing the original meta (if any) plus trace context keys, or None if no trace context to inject and meta was None rr)rinjectTRACE_PARENT_KEYTRACE_STATE_KEY)rcarrier trace_metas rinject_trace_contextr2sg!G W!#J'.}'= #$w&-l&; ?#-4:2-*-- Krspan exceptioncv|j||jttjy)z3Record an exception on a span and set error status.N)record_exception set_statusr r ERROR)rr s rrecord_span_errorr%Ls')$OOF:++,-rctj}|jjrt j S|st j Si}t |vrt|t |d<t|vrt|t|d<|rtj|St j S)aExtract trace context from an MCP request meta dict. If already in a valid trace (e.g., from HTTP propagation), the existing trace context is preserved and meta is not used. Args: meta: The meta dict from an MCP request (ctx.request_context.meta) Returns: An OpenTelemetry Context with the extracted trace context, or the current context if no trace context found or already in a trace rr) rget_current_spanget_span_contextis_valid otel_context get_currentrstrrrextract)r current_spanrs rextract_trace_contextr/Rs))+L$$&//'')) '')) G4!$T*:%;!< $ #D$9 :   ))  # # %%r)rrrr/r rr%)N)__doc__typingr opentelemetryrr*rropentelemetry.contextropentelemetry.tracerr r r r rrrrr,dictr BaseExceptionr%r/__all__rrr9s.1*)@@=   :d :f :#' sCx.4  #s(^d4.D.].t. &S#X 5&'&@ r