diff --git a/frigate/app.py b/frigate/app.py index 517e2bdaf..3faccbcc3 100644 --- a/frigate/app.py +++ b/frigate/app.py @@ -343,7 +343,7 @@ class FrigateApp: migrate_db.close() def init_embeddings(self) -> None: - self.embeddings = None + self.embeddings: Optional[Embeddings] = None try: self.embeddings = Embeddings(self.config) except ValueError: diff --git a/frigate/embeddings/processor.py b/frigate/embeddings/processor.py index 62bb1e1ac..996113180 100644 --- a/frigate/embeddings/processor.py +++ b/frigate/embeddings/processor.py @@ -7,6 +7,7 @@ import queue import threading from multiprocessing import Queue from multiprocessing.synchronize import Event as MpEvent +from typing import Optional import google.generativeai as genai import numpy as np @@ -27,7 +28,7 @@ class EmbeddingProcessor(threading.Thread): def __init__( self, config: FrigateConfig, - embeddings: Embeddings, + embeddings: Optional[Embeddings], queue: Queue, stop_event: MpEvent, ) -> None: diff --git a/frigate/events/cleanup.py b/frigate/events/cleanup.py index 25b30690a..ae678f03d 100644 --- a/frigate/events/cleanup.py +++ b/frigate/events/cleanup.py @@ -7,6 +7,7 @@ import threading from enum import Enum from multiprocessing.synchronize import Event as MpEvent from pathlib import Path +from typing import Optional from frigate.config import FrigateConfig from frigate.const import CLIPS_DIR @@ -23,7 +24,10 @@ class EventCleanupType(str, Enum): class EventCleanup(threading.Thread): def __init__( - self, config: FrigateConfig, embeddings: Embeddings, stop_event: MpEvent + self, + config: FrigateConfig, + embeddings: Optional[Embeddings], + stop_event: MpEvent, ): threading.Thread.__init__(self) self.name = "event_cleanup" diff --git a/frigate/http.py b/frigate/http.py index 065340ee1..d063c920b 100644 --- a/frigate/http.py +++ b/frigate/http.py @@ -12,6 +12,7 @@ import traceback from datetime import datetime, timedelta, timezone from functools import reduce from pathlib import Path +from typing import Optional from urllib.parse import unquote import cv2 @@ -78,7 +79,7 @@ def create_app( onvif: OnvifController, external_processor: ExternalEventProcessor, plus_api: PlusApi, - embeddings: Embeddings, + embeddings: Optional[Embeddings], ): app = Flask(__name__)