encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-03-30 10:57:20 -07:00
Code Issues Packages Projects Releases Wiki Activity

fix: avoid crash when there is no candidates_token_count, which is Optional

Browse Source 
PiperOrigin-RevId: 795643665
This commit is contained in:
Dan Liebling
2025-08-15 15:55:34 -07:00
committed by Copybara-Service
parent ba6e85eb3f
commit 22f34e9d2c
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/google/adk/telemetry.py
+5 -4
View File
@@ -211,10 +211,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,
)
if llm_response.finish_reason:
span.set_attribute(
'gen_ai.response.finish_reasons',