Skip to content

Commit dc85a98

Browse files
author
amiyapatanaik
committed
double counting bug in gemini yield
1 parent c0afc0d commit dc85a98

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pydantic_ai_slim/pydantic_ai/models/gemini.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -434,13 +434,13 @@ async def _get_gemini_responses(self) -> AsyncIterator[_GeminiResponse]:
434434
responses_to_yield = gemini_responses[:-1]
435435
for r in responses_to_yield[current_gemini_response_index:]:
436436
current_gemini_response_index += 1
437-
self._usage += _metadata_as_usage(r)
437+
self._usage = _metadata_as_usage(r)
438438
yield r
439439

440440
# Now yield the final response, which should be complete
441441
if gemini_responses:
442442
r = gemini_responses[-1]
443-
self._usage += _metadata_as_usage(r)
443+
self._usage = _metadata_as_usage(r)
444444
yield r
445445

446446
@property

0 commit comments

Comments
 (0)