introduce triggers in annotations (#1569)

* introduce events annotations * introduce triggers in annotations
Netflix · Oct 11, 2023 · f9d8f85 · f9d8f85
1 parent 5f373fe
commit f9d8f85
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/metaflow/plugins/argo/argo_workflows.py b/metaflow/plugins/argo/argo_workflows.py
@@ -609,6 +609,17 @@ def _compile_workflow_template(self):
         # Some more annotations to populate the Argo UI nicely
         if self.tags:
             annotations.update({"metaflow/tags": json.dumps(self.tags)})
+        if self.triggers:
+            annotations.update(
+                {
+                    "metaflow/triggers": json.dumps(
+                        [
+                            {key: trigger.get(key) for key in ["name", "type"]}
+                            for trigger in self.triggers
+                        ]
+                    )
+                }
+            )
         if self.notify_on_error:
             annotations.update(
                 {
@@ -1738,6 +1749,16 @@ def _compile_sensor(self):
                 }
             )
 
+        # Useful to paint the UI
+        trigger_annotations = {
+            "metaflow/triggered_by": json.dumps(
+                [
+                    {key: trigger.get(key) for key in ["name", "type"]}
+                    for trigger in self.triggers
+                ]
+            )
+        }
+
         return (
             Sensor()
             .metadata(
@@ -1805,6 +1826,17 @@ def _compile_sensor(self):
                                         "metadata": {
                                             "generateName": "%s-" % self.name,
                                             "namespace": KUBERNETES_NAMESPACE,
+                                            "annotations": {
+                                                "metaflow/triggered_by": json.dumps(
+                                                    [
+                                                        {
+                                                            key: trigger.get(key)
+                                                            for key in ["name", "type"]
+                                                        }
+                                                        for trigger in self.triggers
+                                                    ]
+                                                )
+                                            },
                                         },
                                         "spec": {
                                             "arguments": {