From e15924606b4f0a2e146641bf1e9468bbb9cc0fd7 Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Mon, 28 Apr 2025 11:46:36 -0600
Subject: [PATCH] Add to type enum

---
 frigate/data_processing/common/license_plate/mixin.py | 3 ++-
 frigate/data_processing/real_time/face.py             | 3 ++-
 frigate/types.py                                      | 2 ++
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/frigate/data_processing/common/license_plate/mixin.py b/frigate/data_processing/common/license_plate/mixin.py
index a49b7b50e..a1605c10e 100644
--- a/frigate/data_processing/common/license_plate/mixin.py
+++ b/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,
diff --git a/frigate/data_processing/real_time/face.py b/frigate/data_processing/real_time/face.py
index eb85c5336..0f960e0fa 100644
--- a/frigate/data_processing/real_time/face.py
+++ b/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,
diff --git a/frigate/types.py b/frigate/types.py
index 2422c5551..ee48cc02b 100644
--- a/frigate/types.py
+++ b/frigate/types.py
@@ -25,3 +25,5 @@ class ModelStatusTypesEnum(str, Enum):
 
 class TrackedObjectUpdateTypesEnum(str, Enum):
     description = "description"
+    face = "face"
+    lpr = "lpr"