From 49dcc95d26b7d0636201e43882312087e1bcc888 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Wed, 13 Aug 2025 08:46:07 -0600 Subject: [PATCH] Generalize --- frigate/config/camera/genai.py | 5 +++-- frigate/genai/ollama.py | 3 +-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/frigate/config/camera/genai.py b/frigate/config/camera/genai.py index 4e893ab77..3c6baeb15 100644 --- a/frigate/config/camera/genai.py +++ b/frigate/config/camera/genai.py @@ -23,5 +23,6 @@ class GenAIConfig(FrigateBaseModel): base_url: Optional[str] = Field(default=None, title="Provider base url.") model: str = Field(default="gpt-4o", title="GenAI model.") provider: GenAIProviderEnum | None = Field(default=None, title="GenAI provider.") - keep_alive: int | str | None = Field(default="1h", title="Ollama keep_alive.") - extra_options: dict[str, Any] = Field(default={}, title="Ollama extra options.") + provider_options: dict[str, Any] = Field( + default={}, title="GenAI Provider extra options." + ) diff --git a/frigate/genai/ollama.py b/frigate/genai/ollama.py index fe6f944b0..0fb44d785 100644 --- a/frigate/genai/ollama.py +++ b/frigate/genai/ollama.py @@ -48,8 +48,7 @@ class OllamaClient(GenAIClient): self.genai_config.model, prompt, images=images if images else None, - keep_alive=self.genai_config.keep_alive, - options=self.genai_config.extra_options, + **self.genai_config.provider_options, ) return result["response"].strip() except (TimeoutException, ResponseError) as e: