administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-01 19:17:41 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

embeddings maintainer support for post processor

Browse Source
This commit is contained in:
Josh Hawkins 2025-05-26 07:22:14 -05:00
parent c5dfc36171
commit cc21ea0761
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
frigate/embeddings/maintainer.py
View File

@ -37,6 +37,9 @@ from frigate.data_processing.common.license_plate.model import (
LicensePlateModelRunner,
)
from frigate.data_processing.post.api import PostProcessorApi
from frigate.data_processing.post.audio_transcription import (
AudioTranscriptionPostProcessor,
)
from frigate.data_processing.post.license_plate import (
LicensePlatePostProcessor,
)
@ -176,6 +179,18 @@ class EmbeddingMaintainer(threading.Thread):
)
)
audio_transcription_cameras = [
c
for c in self.config.cameras.values()
if c.enabled_in_config and c.audio_transcription.enabled
]
if audio_transcription_cameras:
self.post_processors.append(
AudioTranscriptionPostProcessor(
self.config, self.requestor, metrics, self.embeddings
)
)
self.stop_event = stop_event
self.tracked_events: dict[str, list[Any]] = {}
self.early_request_sent: dict[str, bool] = {}
@ -372,6 +387,8 @@ class EmbeddingMaintainer(threading.Thread):
},
PostProcessDataEnum.recording,
)
elif isinstance(processor, AudioTranscriptionPostProcessor):
continue
else:
processor.process_data(event_id, PostProcessDataEnum.event_id)