Function trace_inference_result

src/google/adk/telemetry/tracing.py:836–878

Trace result of the inference in generate_content span.

(
    invocation_context: InvocationContext | None,
    span: Span | None | GenerateContentSpan,
    llm_response: LlmResponse,
)

def trace_inference_result(
    invocation_context: InvocationContext | None,
    span: Span | None | GenerateContentSpan,
    llm_response: LlmResponse,
):
  """Trace result of the inference in generate_content span."""
  telemetry_config = _telemetry_config_from_invocation_context(
      invocation_context
  )
  gc_span = None
  if isinstance(span, GenerateContentSpan):
    gc_span = span
    span = gc_span.span

  if span is None:
    return

  if llm_response.partial:
    return

  if finish_reason := llm_response.finish_reason:
    span.set_attribute(GEN_AI_RESPONSE_FINISH_REASONS, [finish_reason.lower()])
  _set_usage_metadata_attributes(span, llm_response.usage_metadata)

  if telemetry_config.should_use_experimental_genai_semconv and isinstance(
      gc_span, GenerateContentSpan
  ):
    set_operation_details_attributes_from_response(
        llm_response,
        gc_span.operation_details_attributes,
        gc_span.operation_details_common_attributes,
    )

  else:
    otel_logger.emit(
        LogRecord(
            event_name=GEN_AI_CHOICE_EVENT,
            body=choice_body(
                llm_response, telemetry_config or TelemetryConfig()
            ),
            attributes={GEN_AI_SYSTEM: _guess_gemini_system_name()},
        )
    )


def _guess_gemini_system_name() -> str: