From 81d4b26b0189e8a23e220fff7144fbad4eee9254 Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Fri, 18 Jul 2025 07:42:46 -0600
Subject: [PATCH] Add enum for type of classification for objects

---
 frigate/config/classification.py | 6 ++++++
 frigate/types.py                 | 5 +++++
 2 files changed, 11 insertions(+)

diff --git a/frigate/config/classification.py b/frigate/config/classification.py
index 6b6e0cf52..c4aaabd2d 100644
--- a/frigate/config/classification.py
+++ b/frigate/config/classification.py
@@ -3,6 +3,8 @@ from typing import Dict, List, Optional
 
 from pydantic import ConfigDict, Field
 
+from frigate.types import ObjectClassificationType
+
 from .base import FrigateBaseModel
 
 __all__ = [
@@ -88,6 +90,10 @@ class CustomClassificationStateConfig(FrigateBaseModel):
 
 class CustomClassificationObjectConfig(FrigateBaseModel):
     objects: list[str] = Field(title="Object types to classify.")
+    classification_type: ObjectClassificationType = Field(
+        default=ObjectClassificationType.sub_label,
+        title="Type of classification that is applied.",
+    )
 
 
 class CustomClassificationConfig(FrigateBaseModel):
diff --git a/frigate/types.py b/frigate/types.py
index a9e27ba90..db5d927bb 100644
--- a/frigate/types.py
+++ b/frigate/types.py
@@ -25,6 +25,11 @@ class ModelStatusTypesEnum(str, Enum):
     complete = "complete"
 
 
+class ObjectClassificationType(str, Enum):
+    sub_label = "sub_label"
+    attribute = "attribute"
+
+
 class TrackedObjectUpdateTypesEnum(str, Enum):
     description = "description"
     face = "face"