Function _traced_call

src/python/persistent_wrapper.py:185–317

Call method with OpenTelemetry tracing if enabled. Extracts traceparent from context (3rd arg) and creates a child span that links to the parent trace from Promptfoo.

(method_callable, args, function_name)

def _traced_call(method_callable, args, function_name):
    """
    Call method with OpenTelemetry tracing if enabled.

    Extracts traceparent from context (3rd arg) and creates a child span
    that links to the parent trace from Promptfoo.
    """
    global _tracer, _tracing_enabled

    # Fast path: if tracing not enabled, just call the method
    if not _tracing_enabled or _tracer is None:
        return call_method(method_callable, args)

    # Extract traceparent from context (3rd argument for call_api)
    traceparent = None
    context_arg = None
    if len(args) >= 3:
        context_arg = args[2]
        if isinstance(context_arg, dict):
            traceparent = context_arg.get("traceparent")

    # If no traceparent, fall back to untraced call
    if not traceparent:
        return call_method(method_callable, args)

    try:
        from opentelemetry.propagate import extract
        from opentelemetry.trace import SpanKind, Status, StatusCode

        # Extract parent context from W3C traceparent header
        parent_ctx = extract({"traceparent": traceparent})

        # Determine span name following GenAI conventions
        span_name = f"python {function_name}"

        with _tracer.start_as_current_span(
            span_name, context=parent_ctx, kind=SpanKind.CLIENT
        ) as span:
            # Set GenAI semantic convention attributes
            span.set_attribute("gen_ai.system", "python")
            span.set_attribute("gen_ai.operation.name", function_name)

            # Set request attributes from prompt (1st arg)
            if len(args) >= 1:
                prompt = args[0]
                if isinstance(prompt, str):
                    span.set_attribute("promptfoo.request.body", _truncate_body(prompt))

            # Set model from config if available (2nd arg)
            if len(args) >= 2:
                options = args[1]
                if isinstance(options, dict):
                    config = options.get("config", {})
                    if config.get("model"):
                        span.set_attribute("gen_ai.request.model", config["model"])
                    # Also check for provider id
                    if options.get("id"):
                        span.set_attribute("promptfoo.provider.id", options["id"])