From a0fcc4ec53f357c8e36d8ccfc620108327f46dc2 Mon Sep 17 00:00:00 2001 From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com> Date: Sat, 9 Nov 2024 06:14:45 -0600 Subject: [PATCH] use genai config for class init --- frigate/genai/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/frigate/genai/__init__.py b/frigate/genai/__init__.py index 02377a5a4..ebbea7476 100644 --- a/frigate/genai/__init__.py +++ b/frigate/genai/__init__.py @@ -6,7 +6,7 @@ from typing import Optional from playhouse.shortcuts import model_to_dict -from frigate.config import CameraConfig, FrigateConfig, GenAIProviderEnum +from frigate.config import CameraConfig, FrigateConfig, GenAIConfig, GenAIProviderEnum from frigate.models import Event PROVIDERS = {} @@ -25,8 +25,8 @@ def register_genai_provider(key: GenAIProviderEnum): class GenAIClient: """Generative AI client for Frigate.""" - def __init__(self, config: FrigateConfig, timeout: int = 60) -> None: - self.config: FrigateConfig = config + def __init__(self, genai_config: GenAIConfig, timeout: int = 60) -> None: + self.genai_config: GenAIConfig = genai_config self.timeout = timeout self.provider = self._init_provider()