File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
pydantic_ai_slim/pydantic_ai/models Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -878,7 +878,7 @@ def _metadata_as_usage(response: _GeminiResponse) -> usage.Usage:
878
878
metadata_details = cast (list [_GeminiModalityTokenCount ], metadata_details )
879
879
suffix = key .removesuffix ('_details' )
880
880
for detail in metadata_details :
881
- details [f'{ detail ["modality" ].lower ()} _{ suffix } ' ] = detail [ 'token_count' ]
881
+ details [f'{ detail ["modality" ].lower ()} _{ suffix } ' ] = detail . get ( 'token_count' , 0 )
882
882
883
883
return usage .Usage (
884
884
request_tokens = metadata .get ('prompt_token_count' , 0 ),
Original file line number Diff line number Diff line change @@ -603,7 +603,7 @@ def _metadata_as_usage(response: GenerateContentResponse) -> usage.Usage:
603
603
if key .endswith ('_details' ) and metadata_details :
604
604
suffix = key .removesuffix ('_details' )
605
605
for detail in metadata_details :
606
- details [f'{ detail ["modality" ].lower ()} _{ suffix } ' ] = detail [ 'token_count' ]
606
+ details [f'{ detail ["modality" ].lower ()} _{ suffix } ' ] = detail . get ( 'token_count' , 0 )
607
607
608
608
return usage .Usage (
609
609
request_tokens = metadata .get ('prompt_token_count' , 0 ),
You can’t perform that action at this time.
0 commit comments