diff --git a/frigate/embeddings/__init__.py b/frigate/embeddings/__init__.py index d75d88500..dd05fb0ca 100644 --- a/frigate/embeddings/__init__.py +++ b/frigate/embeddings/__init__.py @@ -15,19 +15,19 @@ from setproctitle import setproctitle from frigate.comms.embeddings_updater import EmbeddingsRequestEnum, EmbeddingsRequestor from frigate.config import FrigateConfig from frigate.const import CONFIG_DIR, FACE_DIR +from frigate.data_processing.types import DataProcessorMetrics from frigate.db.sqlitevecq import SqliteVecQueueDatabase from frigate.models import Event from frigate.util.builtin import serialize from frigate.util.services import listen -from ..postprocessing.types import PostProcessingMetrics from .maintainer import EmbeddingMaintainer from .util import ZScoreNormalization logger = logging.getLogger(__name__) -def manage_embeddings(config: FrigateConfig, metrics: PostProcessingMetrics) -> None: +def manage_embeddings(config: FrigateConfig, metrics: DataProcessorMetrics) -> None: # Only initialize embeddings if semantic search is enabled if not config.semantic_search.enabled: return diff --git a/frigate/embeddings/embeddings.py b/frigate/embeddings/embeddings.py index 3f046d0c3..852806a8d 100644 --- a/frigate/embeddings/embeddings.py +++ b/frigate/embeddings/embeddings.py @@ -16,12 +16,12 @@ from frigate.const import ( UPDATE_EMBEDDINGS_REINDEX_PROGRESS, UPDATE_MODEL_STATE, ) +from frigate.data_processing.types import DataProcessorMetrics from frigate.db.sqlitevecq import SqliteVecQueueDatabase from frigate.models import Event from frigate.types import ModelStatusTypesEnum from frigate.util.builtin import serialize -from ..postprocessing.types import PostProcessingMetrics from .functions.onnx import GenericONNXEmbedding, ModelTypeEnum logger = logging.getLogger(__name__) @@ -65,7 +65,7 @@ class Embeddings: self, config: FrigateConfig, db: SqliteVecQueueDatabase, - metrics: PostProcessingMetrics, + metrics: DataProcessorMetrics, ) -> None: self.config = config self.db = db