diff --git a/src/google/adk/telemetry.py b/src/google/adk/telemetry.py index 10ac58399..4a1b66153 100644 --- a/src/google/adk/telemetry.py +++ b/src/google/adk/telemetry.py @@ -200,10 +200,11 @@ def trace_call_llm( 'gen_ai.usage.input_tokens', llm_response.usage_metadata.prompt_token_count, ) - span.set_attribute( - 'gen_ai.usage.output_tokens', - llm_response.usage_metadata.candidates_token_count, - ) + if llm_response.usage_metadata.candidates_token_count is not None: + span.set_attribute( + 'gen_ai.usage.output_tokens', + llm_response.usage_metadata.candidates_token_count, + ) def trace_send_data(