fix Gemini token validation issue with LiteLLM (#735)

Fix for #734
2025-05-21 14:59:47 -07:00 · 2025-05-21 14:59:47 -07:00 · 1364f4408e
commit 1364f4408e
parent 1992be3e8d
1 changed files with 2 additions and 2 deletions
--- a/src/agents/extensions/models/litellm_model.py
+++ b/src/agents/extensions/models/litellm_model.py
@ -111,12 +111,12 @@ class LitellmModel(Model):
                        input_tokens_details=InputTokensDetails(
                            cached_tokens=getattr(
                                response_usage.prompt_tokens_details, "cached_tokens", 0
-                            )
+                            ) or 0
                        ),
                        output_tokens_details=OutputTokensDetails(
                            reasoning_tokens=getattr(
                                response_usage.completion_tokens_details, "reasoning_tokens", 0
-                            )
+                            ) or 0
                        ),
                    )
                    if response.usage