Implement per-system AI flags

drb-c2-core/app/routers/systems.py

@@ -16,6 +16,11 @@ class TenCodesBody(BaseModel):
     ten_codes: Dict[str, str]
 
 
+class AiFlagsBody(BaseModel):
+    stt_enabled: Optional[bool] = None
+    correlation_enabled: Optional[bool] = None
+
+
 @router.get("")
 async def list_systems():
     return await fstore.collection_list("systems")
@@ -54,6 +59,30 @@ async def delete_system(system_id: str):
     await fstore.doc_delete("systems", system_id)
 
 
+# ── Per-system AI flag overrides ──────────────────────────────────────────────
+
+@router.put("/{system_id}/ai-flags")
+async def update_system_ai_flags(system_id: str, body: AiFlagsBody):
+    """
+    Set per-system AI flag overrides.  Only fields included in the body are
+    written; omitted fields remain unchanged (or absent, meaning inherit global).
+    Pass null to clear an override and fall back to the global flag.
+    """
+    existing = await fstore.doc_get("systems", system_id)
+    if not existing:
+        raise HTTPException(404, f"System '{system_id}' not found.")
+
+    current: dict = existing.get("ai_flags") or {}
+    for field, value in body.model_dump(exclude_unset=False).items():
+        if value is None:
+            current.pop(field, None)   # clear override → inherit global
+        else:
+            current[field] = value
+
+    await fstore.doc_update("systems", system_id, {"ai_flags": current})
+    return {"ok": True, "ai_flags": current}
+
+
 # ── Ten-codes endpoints ────────────────────────────────────────────────────────
 
 @router.get("/{system_id}/ten-codes")

drb-c2-core/app/routers/upload.py

@@ -161,21 +161,34 @@ async def _run_intelligence_pipeline(
 
     flags = await get_flags()
 
+    # Resolve per-system overrides: system flag=False beats global flag=True,
+    # but global flag=False beats everything (master switch).
+    system_ai_flags: dict = {}
+    if system_id:
+        sys_doc = await fstore.doc_get("systems", system_id)
+        system_ai_flags = (sys_doc or {}).get("ai_flags") or {}
+
+    def _flag(name: str) -> bool:
+        if not flags[name]:          # global master off
+            return False
+        return system_ai_flags.get(name, True)  # system override, default inherit
+
     transcript: Optional[str] = None
     segments: list[dict] = []
 
     # Step 1: Transcription
     if gcs_uri:
-        if flags["stt_enabled"]:
+        if _flag("stt_enabled"):
             transcript, segments = await transcription.transcribe_call(
                 call_id, gcs_uri, talkgroup_name, system_id=system_id
             )
         else:
-            logger.info(f"STT disabled — skipping transcription for call {call_id}")
+            scope = "globally" if not flags["stt_enabled"] else f"system {system_id}"
+            logger.info(f"STT disabled ({scope}) — skipping transcription for call {call_id}")
 
     # Step 2: Scene detection + intelligence extraction
     scenes: list[dict] = []
-    if flags["correlation_enabled"]:
+    if _flag("correlation_enabled"):
         if transcript:
             scenes = await intelligence.extract_scenes(
                 call_id, transcript, talkgroup_name,
@@ -183,7 +196,8 @@ async def _run_intelligence_pipeline(
                 node_id=node_id,
             )
     else:
-        logger.info(f"Correlation disabled — skipping scene extraction and correlation for call {call_id}")
+        scope = "globally" if not flags["correlation_enabled"] else f"system {system_id}"
+        logger.info(f"Correlation disabled ({scope}) — skipping scene extraction and correlation for call {call_id}")
 
     # Step 3: Correlate each scene independently.
     # A single recording can produce multiple incidents on a busy channel.