Skip to content

Commit 22f34e9

Browse files
danyoelcopybara-github
authored andcommitted
fix: avoid crash when there is no candidates_token_count, which is Optional
PiperOrigin-RevId: 795643665
1 parent ba6e85e commit 22f34e9

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/google/adk/telemetry.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -211,10 +211,11 @@ def trace_call_llm(
211211
'gen_ai.usage.input_tokens',
212212
llm_response.usage_metadata.prompt_token_count,
213213
)
214-
span.set_attribute(
215-
'gen_ai.usage.output_tokens',
216-
llm_response.usage_metadata.candidates_token_count,
217-
)
214+
if llm_response.usage_metadata.candidates_token_count is not None:
215+
span.set_attribute(
216+
'gen_ai.usage.output_tokens',
217+
llm_response.usage_metadata.candidates_token_count,
218+
)
218219
if llm_response.finish_reason:
219220
span.set_attribute(
220221
'gen_ai.response.finish_reasons',

0 commit comments

Comments
 (0)