We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ba6e85e commit 22f34e9Copy full SHA for 22f34e9
src/google/adk/telemetry.py
@@ -211,10 +211,11 @@ def trace_call_llm(
211
'gen_ai.usage.input_tokens',
212
llm_response.usage_metadata.prompt_token_count,
213
)
214
- span.set_attribute(
215
- 'gen_ai.usage.output_tokens',
216
- llm_response.usage_metadata.candidates_token_count,
217
- )
+ 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,
218
+ )
219
if llm_response.finish_reason:
220
span.set_attribute(
221
'gen_ai.response.finish_reasons',
0 commit comments