Spaces:
Paused
Paused
Upload app/routers/alerts.py with huggingface_hub
Browse files- app/routers/alerts.py +9 -3
app/routers/alerts.py
CHANGED
|
@@ -88,8 +88,12 @@ async def get_alerts(
|
|
| 88 |
"description": alert.description,
|
| 89 |
"severity": alert.severity,
|
| 90 |
"status": getattr(alert, "status", meta.get("status", "pending")),
|
| 91 |
-
"risk_score": meta.get(
|
| 92 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 93 |
}
|
| 94 |
)
|
| 95 |
|
|
@@ -126,7 +130,9 @@ async def update_alert(
|
|
| 126 |
if status in ["approved", "rejected", "escalated"]:
|
| 127 |
alert.is_acknowledged = True
|
| 128 |
alert.acknowledged_at = datetime.now(UTC)
|
| 129 |
-
user_id =
|
|
|
|
|
|
|
| 130 |
alert.acknowledged_by = user_id
|
| 131 |
|
| 132 |
db.commit()
|
|
|
|
| 88 |
"description": alert.description,
|
| 89 |
"severity": alert.severity,
|
| 90 |
"status": getattr(alert, "status", meta.get("status", "pending")),
|
| 91 |
+
"risk_score": meta.get(
|
| 92 |
+
"risk_score", 0.0
|
| 93 |
+
), # Assuming risk score is in metadata
|
| 94 |
+
"created_at": (
|
| 95 |
+
alert.created_at.isoformat() if alert.created_at else None
|
| 96 |
+
),
|
| 97 |
}
|
| 98 |
)
|
| 99 |
|
|
|
|
| 130 |
if status in ["approved", "rejected", "escalated"]:
|
| 131 |
alert.is_acknowledged = True
|
| 132 |
alert.acknowledged_at = datetime.now(UTC)
|
| 133 |
+
user_id = (
|
| 134 |
+
getattr(current_user, "id", str(current_user)) if current_user else None
|
| 135 |
+
)
|
| 136 |
alert.acknowledged_by = user_id
|
| 137 |
|
| 138 |
db.commit()
|