Generalize

This commit is contained in:
Nicolas Mowen 2025-08-13 08:46:07 -06:00
parent 69c7d5047e
commit 49dcc95d26
2 changed files with 4 additions and 4 deletions

View File

@ -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."
)

View File

@ -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: