Dynamically enable/disable GenAI (#19139)

* config

* dispatcher and mqtt

* docs

* use config updater

* add switch to frontend

frigate/comms/dispatcher.py

@@ -75,6 +75,7 @@ class Dispatcher:
             "birdseye_mode": self._on_birdseye_mode_command,
             "review_alerts": self._on_alerts_command,
             "review_detections": self._on_detections_command,
+            "genai": self._on_genai_command,
         }
         self._global_settings_handlers: dict[str, Callable] = {
             "notifications": self._on_global_notification_command,
@@ -207,6 +208,7 @@ class Dispatcher:
                     ].onvif.autotracking.enabled,
                     "alerts": self.config.cameras[camera].review.alerts.enabled,
                     "detections": self.config.cameras[camera].review.detections.enabled,
+                    "genai": self.config.cameras[camera].genai.enabled,
                 }
 
             self.publish("camera_activity", json.dumps(camera_status))
@@ -737,3 +739,28 @@ class Dispatcher:
             review_settings,
         )
         self.publish(f"{camera_name}/review_detections/state", payload, retain=True)
+
+    def _on_genai_command(self, camera_name: str, payload: str) -> None:
+        """Callback for GenAI topic."""
+        genai_settings = self.config.cameras[camera_name].genai
+
+        if payload == "ON":
+            if not self.config.cameras[camera_name].genai.enabled_in_config:
+                logger.error(
+                    "GenAI must be enabled in the config to be turned on via MQTT."
+                )
+                return
+
+            if not genai_settings.enabled:
+                logger.info(f"Turning on GenAI for {camera_name}")
+                genai_settings.enabled = True
+        elif payload == "OFF":
+            if genai_settings.enabled:
+                logger.info(f"Turning off GenAI for {camera_name}")
+                genai_settings.enabled = False
+
+        self.config_updater.publish_update(
+            CameraConfigUpdateTopic(CameraConfigUpdateEnum.genai, camera_name),
+            genai_settings,
+        )
+        self.publish(f"{camera_name}/genai/state", payload, retain=True)

frigate/comms/mqtt.py

@@ -122,6 +122,11 @@ class MqttClient(Communicator):  # type: ignore[misc]
                 "ON" if camera.review.detections.enabled_in_config else "OFF",
                 retain=True,
             )
+            self.publish(
+                f"{camera_name}/genai/state",
+                "ON" if camera.genai.enabled_in_config else "OFF",
+                retain=True,
+            )
 
         if self.config.notifications.enabled_in_config:
             self.publish(
@@ -215,6 +220,7 @@ class MqttClient(Communicator):  # type: ignore[misc]
             "birdseye_mode",
             "review_alerts",
             "review_detections",
+            "genai",
         ]
 
         for name in self.config.cameras.keys():

frigate/config/camera/genai.py

@@ -58,6 +58,10 @@ class GenAICameraConfig(BaseModel):
         title="What triggers to use to send frames to generative AI for a tracked object.",
     )
 
+    enabled_in_config: Optional[bool] = Field(
+        default=None, title="Keep track of original state of generative AI."
+    )
+
     @field_validator("required_zones", mode="before")
     @classmethod
     def validate_required_zones(cls, v):

frigate/config/camera/updater.py

@@ -17,6 +17,7 @@ class CameraConfigUpdateEnum(str, Enum):
     birdseye = "birdseye"
     detect = "detect"
     enabled = "enabled"
+    genai = "genai"
     motion = "motion"  # includes motion and motion masks
     notifications = "notifications"
     objects = "objects"
@@ -97,6 +98,8 @@ class CameraConfigUpdateSubscriber:
             config.detect = updated_config
         elif update_type == CameraConfigUpdateEnum.enabled:
             config.enabled = updated_config
+        elif update_type == CameraConfigUpdateEnum.genai:
+            config.genai = updated_config
         elif update_type == CameraConfigUpdateEnum.motion:
             config.motion = updated_config
         elif update_type == CameraConfigUpdateEnum.notifications:

frigate/config/config.py

@@ -606,6 +606,7 @@ class FrigateConfig(FrigateBaseModel):
             camera_config.review.detections.enabled_in_config = (
                 camera_config.review.detections.enabled
             )
+            camera_config.genai.enabled_in_config = camera_config.genai.enabled
 
             # Add default filters
             object_keys = camera_config.objects.track

frigate/embeddings/maintainer.py

@@ -100,6 +100,7 @@ class EmbeddingMaintainer(threading.Thread):
             [
                 CameraConfigUpdateEnum.add,
                 CameraConfigUpdateEnum.remove,
+                CameraConfigUpdateEnum.genai,
                 CameraConfigUpdateEnum.semantic_search,
             ],
         )