Fix Gemini Chat

This commit is contained in:
Nicolas Mowen 2026-03-17 20:23:04 -06:00
parent 2f05ff2a96
commit 996d2b15b1

View File

@ -397,13 +397,13 @@ class GeminiClient(GenAIClient):
tool_calls_by_index: dict[int, dict[str, Any]] = {} tool_calls_by_index: dict[int, dict[str, Any]] = {}
finish_reason = "stop" finish_reason = "stop"
response = self.provider.models.generate_content_stream( stream = await self.provider.aio.models.generate_content_stream(
model=self.genai_config.model, model=self.genai_config.model,
contents=gemini_messages, contents=gemini_messages,
config=types.GenerateContentConfig(**config_params), config=types.GenerateContentConfig(**config_params),
) )
async for chunk in response: async for chunk in stream:
if not chunk or not chunk.candidates: if not chunk or not chunk.candidates:
continue continue