diff --git a/docker/main/requirements-wheels.txt b/docker/main/requirements-wheels.txt index ce27dabc40..b5e5369701 100644 --- a/docker/main/requirements-wheels.txt +++ b/docker/main/requirements-wheels.txt @@ -47,7 +47,7 @@ onnxruntime == 1.31.* # Embeddings transformers == 5.21.* # Generative AI -google-genai == 2.32.* +google-genai == 2.34.* ollama == 0.6.* openai == 3.36.* # push notifications diff --git a/frigate/genai/plugins/gemini.py b/frigate/genai/plugins/gemini.py index b6a520095b..9956f0157d 100644 --- a/frigate/genai/plugins/gemini.py +++ b/frigate/genai/plugins/gemini.py @@ -386,7 +386,7 @@ class GeminiClient(GenAIClient): response = self.provider.models.generate_content( model=self.genai_config.model, - contents=gemini_messages, # type: ignore[arg-type] + contents=gemini_messages, config=types.GenerateContentConfig(**config_params), ) @@ -648,7 +648,7 @@ class GeminiClient(GenAIClient): stream = await self.provider.aio.models.generate_content_stream( model=self.genai_config.model, - contents=gemini_messages, # type: ignore[arg-type] + contents=gemini_messages, config=types.GenerateContentConfig(**config_params), )