From e9f422ce09b6291f7984389906346e06e2282035 Mon Sep 17 00:00:00 2001
From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com>
Date: Sat, 7 Mar 2026 16:53:41 -0600
Subject: [PATCH] enrichment updater and enum

---
 frigate/config/enrichment_updater.py | 49 ++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)
 create mode 100644 frigate/config/enrichment_updater.py

diff --git a/frigate/config/enrichment_updater.py b/frigate/config/enrichment_updater.py
new file mode 100644
index 000000000..f570fd49f
--- /dev/null
+++ b/frigate/config/enrichment_updater.py
@@ -0,0 +1,49 @@
+"""Convenience classes for updating enrichment configurations dynamically."""
+
+from enum import Enum
+from typing import Any
+
+from frigate.comms.config_updater import ConfigSubscriber
+
+
+class EnrichmentConfigEnum(str, Enum):
+    """Supported enrichment config update topics."""
+
+    classification = "config/classification"
+    classification_custom = "config/classification/custom/"
+    face_recognition = "config/face_recognition"
+    lpr = "config/lpr"
+
+
+class EnrichmentConfigSubscriber:
+    """Subscribes to all enrichment config updates under ``config/``."""
+
+    def __init__(self) -> None:
+        self.subscriber = ConfigSubscriber("config/")
+
+    def check_for_update(
+        self,
+    ) -> tuple[EnrichmentConfigEnum, str, Any] | tuple[None, None, None]:
+        """Check for an enrichment config update.
+
+        Returns:
+            A tuple of (update_type, topic, payload) or (None, None, None)
+            if no update is available. For custom classification topics the
+            full topic string is returned so the model name can be extracted.
+        """
+        topic, payload = self.subscriber.check_for_update()
+
+        if topic is None:
+            return (None, None, None)
+
+        for member in EnrichmentConfigEnum:
+            if topic == member.value or (
+                member == EnrichmentConfigEnum.classification_custom
+                and topic.startswith(member.value)
+            ):
+                return (member, topic, payload)
+
+        return (None, None, None)
+
+    def stop(self) -> None:
+        self.subscriber.stop()