Add to type enum

frigate/data_processing/common/license_plate/mixin.py

@@ -24,6 +24,7 @@ from frigate.comms.event_metadata_updater import (
 )
 from frigate.const import CLIPS_DIR
 from frigate.embeddings.onnx.lpr_embedding import LPR_EMBEDDING_SIZE
+from frigate.types import TrackedObjectUpdateTypesEnum
 from frigate.util.builtin import EventsPerSecond
 from frigate.util.image import area
 
@@ -1514,7 +1515,7 @@ class LicensePlateProcessingMixin:
         self.requestor.send_data(
             "tracked_object_update",
             json.dumps({
-                "type": "lpr",
+                "type": TrackedObjectUpdateTypesEnum.lpr,
                 "name": sub_label,
                 "plate": top_plate,
                 "score": avg_confidence,

frigate/data_processing/real_time/face.py

@@ -26,6 +26,7 @@ from frigate.data_processing.common.face.model import (
     FaceNetRecognizer,
     FaceRecognizer,
 )
+from frigate.types import TrackedObjectUpdateTypesEnum
 from frigate.util.builtin import EventsPerSecond
 from frigate.util.image import area
 
@@ -295,7 +296,7 @@ class FaceRealTimeProcessor(RealTimeProcessorApi):
         self.requestor.send_data(
             "tracked_object_update",
             json.dumps({
-                "type": "face",
+                "type": TrackedObjectUpdateTypesEnum.face,
                 "name": weighted_sub_label,
                 "score": weighted_score,
                 "id": id,

frigate/types.py

@@ -25,3 +25,5 @@ class ModelStatusTypesEnum(str, Enum):
 
 class TrackedObjectUpdateTypesEnum(str, Enum):
     description = "description"
+    face = "face"
+    lpr = "lpr"