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:
878878 metadata_details = cast (list [_GeminiModalityTokenCount ], metadata_details )
879879 suffix = key .removesuffix ('_details' )
880880 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 )
882882
883883 return usage .Usage (
884884 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:
603603 if key .endswith ('_details' ) and metadata_details :
604604 suffix = key .removesuffix ('_details' )
605605 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 )
607607
608608 return usage .Usage (
609609 request_tokens = metadata .get ('prompt_token_count' , 0 ),
You can’t perform that action at this time.
0 commit comments