From 87e0ff9fac1571c2a21f33772ec42729276308ff Mon Sep 17 00:00:00 2001 From: amiyapatanaik Date: Tue, 13 May 2025 19:19:27 +0800 Subject: [PATCH] double counting bug in gemini yield --- pydantic_ai_slim/pydantic_ai/models/gemini.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pydantic_ai_slim/pydantic_ai/models/gemini.py b/pydantic_ai_slim/pydantic_ai/models/gemini.py index 314757967..673470e6d 100644 --- a/pydantic_ai_slim/pydantic_ai/models/gemini.py +++ b/pydantic_ai_slim/pydantic_ai/models/gemini.py @@ -434,13 +434,13 @@ async def _get_gemini_responses(self) -> AsyncIterator[_GeminiResponse]: responses_to_yield = gemini_responses[:-1] for r in responses_to_yield[current_gemini_response_index:]: current_gemini_response_index += 1 - self._usage += _metadata_as_usage(r) + self._usage = _metadata_as_usage(r) yield r # Now yield the final response, which should be complete if gemini_responses: r = gemini_responses[-1] - self._usage += _metadata_as_usage(r) + self._usage = _metadata_as_usage(r) yield r @property