Add trigger info to the pipeline run metadata (#4106)

Author: stefannica

src/zenml/models/v2/core/pipeline_run.py

@@ -271,6 +271,10 @@ class PipelineRunResponseMetadata(ProjectScopedResponseMetadata):
         default=False,
         description="Whether a template can be created from this run.",
     )
+    trigger_info: Optional[PipelineRunTriggerInfo] = Field(
+        default=None,
+        title="Trigger information for the pipeline run.",
+    )
 
 
 class PipelineRunResponseResources(ProjectScopedResponseResources):

src/zenml/zen_stores/schemas/pipeline_run_schemas.py

@@ -43,6 +43,7 @@
     PipelineRunResponse,
     PipelineRunResponseBody,
     PipelineRunResponseMetadata,
+    PipelineRunTriggerInfo,
     PipelineRunUpdate,
     RunMetadataEntry,
 )
@@ -565,6 +566,23 @@ def to_model(
                 client_environment.pop("python_packages", None)
                 orchestrator_environment.pop("python_packages", None)
 
+            trigger_info: Optional[PipelineRunTriggerInfo] = None
+            if self.triggered_by and self.triggered_by_type:
+                if (
+                    self.triggered_by_type
+                    == PipelineRunTriggeredByType.STEP_RUN.value
+                ):
+                    trigger_info = PipelineRunTriggerInfo(
+                        step_run_id=self.triggered_by,
+                    )
+                elif (
+                    self.triggered_by_type
+                    == PipelineRunTriggeredByType.DEPLOYMENT.value
+                ):
+                    trigger_info = PipelineRunTriggerInfo(
+                        deployment_id=self.triggered_by,
+                    )
+
             metadata = PipelineRunResponseMetadata(
                 run_metadata=self.fetch_metadata(
                     include_full_metadata=include_full_metadata
@@ -580,6 +598,7 @@ def to_model(
                 if self.snapshot
                 else None,
                 is_templatable=is_templatable,
+                trigger_info=trigger_info,
             )
 
         resources = None