fix: key not found error

rudderlabs · Jul 30, 2024 · 7afb96a · 7afb96a
1 parent 9334b0e
commit 7afb96a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/alerta/models/alert.py b/alerta/models/alert.py
@@ -53,8 +53,8 @@ def get_rudder_resource_from_tags(tags):
 
 def add_custom_labels(labels: JSON):
     # for profiles alerts, add profile_id as a label
-    if labels['category'] == 'wht' and 'job_id' in labels:
-        labels['profile_id'] = labels['job_id']
+    if labels.get('category') == 'wht' and 'job_id' in labels:
+        labels['profile_id'] = labels.get('job_id')
     return labels;
 
 class Alert: