[Notifications] Fix sensitive data appearing in api logs (#3663)

mlrun · May 31, 2023 · 6f4856e · 6f4856e
1 parent 2efe330
commit 6f4856e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/mlrun/utils/notifications/notification_pusher.py b/mlrun/utils/notifications/notification_pusher.py
@@ -156,7 +156,7 @@ async def _push_notification(
         )
         logger.debug(
             "Pushing notification",
-            notification=notification_object.to_dict(),
+            notification=_sanitize_notification(notification_object),
             run_uid=run.metadata.uid,
         )
         try:
@@ -346,3 +346,9 @@ def push_pipeline_run_results(
         if state:
             text += f", state={state}"
         self.push(text, "info", runs=runs_list)
+
+
+def _sanitize_notification(notification: mlrun.model.Notification):
+    notification_dict = notification.to_dict()
+    notification_dict.pop("params", None)
+    return notification_dict