admin-valentin/openai-agents-python
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix Gemini token validation issue with LiteLLM (#735)

Browse source 
Fix for #734
This commit is contained in:
Andrew Han 2025-05-21 14:59:47 -07:00 committed by GitHub
parent 1992be3e8d
commit 1364f4408e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/agents/extensions/models/litellm_model.py
View file

@ -111,12 +111,12 @@ class LitellmModel(Model):
input_tokens_details=InputTokensDetails( input_tokens_details=InputTokensDetails(
cached_tokens=getattr( cached_tokens=getattr(
response_usage.prompt_tokens_details, "cached_tokens", 0 response_usage.prompt_tokens_details, "cached_tokens", 0
) ) or 0
), ),
output_tokens_details=OutputTokensDetails( output_tokens_details=OutputTokensDetails(
reasoning_tokens=getattr( reasoning_tokens=getattr(
response_usage.completion_tokens_details, "reasoning_tokens", 0 response_usage.completion_tokens_details, "reasoning_tokens", 0
) ) or 0
), ),
) )
if response.usage if response.usage