fix Gemini token validation issue with LiteLLM (#735)

handrew · web-flow · commit 1364f4408e8e · 2025-05-21T17:59:47.000-04:00
Fix for #734
diff --git a/src/agents/extensions/models/litellm_model.py b/src/agents/extensions/models/litellm_model.py
@@ -111,12 +111,12 @@ async def get_response(
                         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
