Function _set_usage_metadata_attributes

src/google/adk/telemetry/tracing.py:304–311

Records usage metadata attributes on the given span.

(
    span: Span,
    usage_metadata: types.GenerateContentResponseUsageMetadata | None,
)

def _set_usage_metadata_attributes(
    span: Span,
    usage_metadata: types.GenerateContentResponseUsageMetadata | None,
) -> None:
  """Records usage metadata attributes on the given span."""
  if usage_metadata is None:
    return
  span.set_attributes(TokenUsage(usage_metadata).to_attributes())


def trace_call_llm(