teoat commited on
Commit
c98cc5a
·
verified ·
1 Parent(s): f66e5e6

Upload app/routers/alerts.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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("risk_score", 0.0), # Assuming risk score is in metadata
92
- "created_at": alert.created_at.isoformat() if alert.created_at else None,
 
 
 
 
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 = getattr(current_user, "id", str(current_user)) if current_user else None
 
 
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()