Skip to content

Commit 1364f44

Browse files
authored
fix Gemini token validation issue with LiteLLM (#735)
Fix for #734
1 parent 1992be3 commit 1364f44

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/agents/extensions/models/litellm_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -111,12 +111,12 @@ async def get_response(
111111
input_tokens_details=InputTokensDetails(
112112
cached_tokens=getattr(
113113
response_usage.prompt_tokens_details, "cached_tokens", 0
114-
)
114+
) or 0
115115
),
116116
output_tokens_details=OutputTokensDetails(
117117
reasoning_tokens=getattr(
118118
response_usage.completion_tokens_details, "reasoning_tokens", 0
119-
)
119+
) or 0
120120
),
121121
)
122122
if response.usage

0 commit comments

Comments
 (0)