Skip to content

Commit 87e0ff9

Browse files
amiyapatanaikamiyapatanaik
authored andcommitted
double counting bug in gemini yield
1 parent d1615df commit 87e0ff9

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)