feat: only send critical alerts on endpoint

src/codegate/api/v1.py

@@ -2,6 +2,7 @@
 from uuid import UUID
 
 import requests
+from codegate.pipeline.base import AlertSeverity
 import structlog
 from fastapi import APIRouter, Depends, HTTPException, Response
 from fastapi.responses import StreamingResponse
@@ -379,7 +380,7 @@ async def get_workspace_alerts(workspace_name: str) -> List[Optional[v1_models.A
         raise HTTPException(status_code=500, detail="Internal server error")
 
     try:
-        alerts = await dbreader.get_alerts_by_workspace(ws.id)
+        alerts = await dbreader.get_alerts_by_workspace(ws.id, AlertSeverity.CRITICAL.value)
         prompts_outputs = await dbreader.get_prompts_with_output(ws.id)
         return await v1_processing.parse_get_alert_conversation(alerts, prompts_outputs)
     except Exception:

src/codegate/db/connection.py

@@ -580,7 +580,9 @@ async def get_prompts_with_output(self, workpace_id: str) -> List[GetPromptWithO
         )
         return prompts
 
-    async def get_alerts_by_workspace(self, workspace_id: str) -> List[Alert]:
+    async def get_alerts_by_workspace(
+        self, workspace_id: str, trigger_category: Optional[str] = None
+    ) -> List[Alert]:
         sql = text(
             """
             SELECT
@@ -594,10 +596,16 @@ async def get_alerts_by_workspace(self, workspace_id: str) -> List[Alert]:
             FROM alerts a
             INNER JOIN prompts p ON p.id = a.prompt_id
             WHERE p.workspace_id = :workspace_id
-            ORDER BY a.timestamp DESC
-            """
+        """
         )
         conditions = {"workspace_id": workspace_id}
+
+        if trigger_category:
+            sql = text(sql.text + " AND a.trigger_category = :trigger_category")
+            conditions["trigger_category"] = trigger_category
+
+        sql = text(sql.text + " ORDER BY a.timestamp DESC")
+
         prompts = await self._exec_select_conditions_to_pydantic(
             Alert, sql, conditions, should_raise=True
         )