Function trace_context

python/copilot/_telemetry.py:27–48

Context manager that sets the trace context from W3C headers for the block's duration.

(traceparent: str | None, tracestate: str | None)

@contextmanager
def trace_context(traceparent: str | None, tracestate: str | None) -> Generator[None, None, None]:
    """Context manager that sets the trace context from W3C headers for the block's duration."""
    try:
        from opentelemetry import context, propagate
    except ImportError:
        yield
        return

    if not traceparent:
        yield
        return

    carrier: dict[str, str] = {"traceparent": traceparent}
    if tracestate:
        carrier["tracestate"] = tracestate

    ctx = propagate.extract(carrier, context=context.get_current())
    token = context.attach(ctx)
    try:
        yield
    finally:
        context.detach(token)