administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-17 08:35:21 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

use genai config for class init

Browse Source
This commit is contained in:
Josh Hawkins 2024-11-09 06:14:45 -06:00
parent 6234121699
commit a0fcc4ec53
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/genai/__init__.py
View File

@ -6,7 +6,7 @@ from typing import Optional
from playhouse.shortcuts import model_to_dict 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 from frigate.models import Event
PROVIDERS = {} PROVIDERS = {}
@ -25,8 +25,8 @@ def register_genai_provider(key: GenAIProviderEnum):
class GenAIClient: class GenAIClient:
"""Generative AI client for Frigate.""" """Generative AI client for Frigate."""
def __init__(self, config: FrigateConfig, timeout: int = 60) -> None: def __init__(self, genai_config: GenAIConfig, timeout: int = 60) -> None:
self.config: FrigateConfig = config self.genai_config: GenAIConfig = genai_config
self.timeout = timeout self.timeout = timeout
self.provider = self._init_provider() self.provider = self._init_provider()