Function get_trace_context

python/copilot/_telemetry.py:9–23

Get the current W3C Trace Context (traceparent/tracestate) if OpenTelemetry is available.

()

def get_trace_context() -> dict[str, str]:
    """Get the current W3C Trace Context (traceparent/tracestate) if OpenTelemetry is available."""
    try:
        from opentelemetry import context, propagate
    except ImportError:
        return {}

    carrier: dict[str, str] = {}
    propagate.inject(carrier, context=context.get_current())
    result: dict[str, str] = {}
    if "traceparent" in carrier:
        result["traceparent"] = carrier["traceparent"]
    if "tracestate" in carrier:
        result["tracestate"] = carrier["tracestate"]
    return result


@contextmanager