vellum-ai 1.2.1__py3-none-any.whl → 1.2.3__py3-none-any.whl

vellum_ee/workflows/display/nodes/base_node_display.py

@@ -183,8 +183,6 @@ class BaseNodeDisplay(Generic[NodeType], metaclass=BaseNodeDisplayMeta):
             existing_adornments = adornments if adornments is not None else []
             return display_class().serialize(display_context, adornments=existing_adornments + [adornment])
 
-        ports = self.serialize_ports(display_context)
-
         outputs: JsonArray = []
         for output in node.Outputs:
             type = primitive_type_to_vellum_variable_type(output)
@@ -207,14 +205,7 @@ class BaseNodeDisplay(Generic[NodeType], metaclass=BaseNodeDisplayMeta):
             "id": str(node_id),
             "label": self.label,
             "type": "GENERIC",
-            "display_data": self.get_display_data().dict(),
-            "base": self.get_base().dict(),
-            "definition": self.get_definition().dict(),
-            "trigger": {
-                "id": str(self.get_target_handle_id()),
-                "merge_behavior": node.Trigger.merge_behavior.value,
-            },
-            "ports": ports,
+            **self.serialize_generic_fields(display_context),
             "adornments": adornments,
             "attributes": attributes,
             "outputs": outputs,
@@ -247,6 +238,14 @@ class BaseNodeDisplay(Generic[NodeType], metaclass=BaseNodeDisplayMeta):
 
         return ports
 
+    def serialize_trigger(self) -> JsonObject:
+        """Serialize the trigger of the node."""
+        node = self._node
+        return {
+            "id": str(self.get_target_handle_id()),
+            "merge_behavior": node.Trigger.merge_behavior.value,
+        }
+
     def _serialize_attributes(self, display_context: "WorkflowDisplayContext") -> JsonArray:
         """Serialize node attributes, skipping unserializable ones."""
         attributes: JsonArray = []
@@ -271,6 +270,16 @@ class BaseNodeDisplay(Generic[NodeType], metaclass=BaseNodeDisplayMeta):
 
         return attributes
 
+    def serialize_generic_fields(self, display_context: "WorkflowDisplayContext") -> JsonObject:
+        """Serialize generic fields that are common to all nodes."""
+        return {
+            "display_data": self.get_display_data().dict(),
+            "base": self.get_base().dict(),
+            "definition": self.get_definition().dict(),
+            "trigger": self.serialize_trigger(),
+            "ports": self.serialize_ports(display_context),
+        }
+
     def get_base(self) -> CodeResourceDefinition:
         node = self._node
 

vellum_ee/workflows/display/nodes/vellum/api_node.py

@@ -213,10 +213,7 @@ class BaseAPINodeDisplay(BaseNodeDisplay[_APINodeType], Generic[_APINodeType]):
                 "json_output_id": str(json_output_display.id),
                 "status_code_output_id": str(status_code_output_display.id),
             },
-            "display_data": self.get_display_data().dict(),
-            "base": self.get_base().dict(),
-            "definition": self.get_definition().dict(),
-            "ports": self.serialize_ports(display_context),
+            **self.serialize_generic_fields(display_context),
         }
 
         attributes = self._serialize_attributes(display_context)

vellum_ee/workflows/display/nodes/vellum/code_execution_node.py

@@ -103,8 +103,5 @@ class BaseCodeExecutionNodeDisplay(BaseNodeDisplay[_CodeExecutionNodeType], Gene
                 "output_id": str(self.output_id) if self.output_id else str(output_display.id),
                 "log_output_id": str(self.log_output_id) if self.log_output_id else str(log_output_display.id),
             },
-            "display_data": self.get_display_data().dict(),
-            "base": self.get_base().dict(),
-            "definition": self.get_definition().dict(),
-            "ports": self.serialize_ports(display_context),
+            **self.serialize_generic_fields(display_context),
         }

vellum_ee/workflows/display/nodes/vellum/conditional_node.py

@@ -217,10 +217,7 @@ but the defined conditions have length {len(condition_ids)}"""
                 "conditions": conditions,  # type: ignore
                 "version": "2",
             },
-            "display_data": self.get_display_data().dict(),
-            "base": self.get_base().dict(),
-            "definition": self.get_definition().dict(),
-            "ports": self.serialize_ports(display_context),
+            **self.serialize_generic_fields(display_context),
         }
 
     def get_nested_rule_details_by_path(

vellum_ee/workflows/display/nodes/vellum/error_node.py

@@ -47,9 +47,7 @@ class BaseErrorNodeDisplay(BaseNodeDisplay[_ErrorNodeType], Generic[_ErrorNodeTy
                 "target_handle_id": str(self.get_target_handle_id()),
                 "error_source_input_id": str(error_source_input_id),
             },
-            "display_data": self.get_display_data().dict(),
-            "base": self.get_base().dict(),
-            "definition": self.get_definition().dict(),
+            **self.serialize_generic_fields(display_context),
         }
 
         if self.name:

vellum_ee/workflows/display/nodes/vellum/final_output_node.py

@@ -45,9 +45,7 @@ class BaseFinalOutputNodeDisplay(BaseNodeDisplay[_FinalOutputNodeType], Generic[
                 "node_input_id": str(node_input.id),
             },
             "inputs": [node_input.dict()],
-            "display_data": self.get_display_data().dict(),
-            "base": self.get_base().dict(),
-            "definition": self.get_definition().dict(),
+            **self.serialize_generic_fields(display_context),
             "outputs": [
                 {
                     "id": str(self._get_output_id()),

vellum_ee/workflows/display/nodes/vellum/guardrail_node.py

@@ -45,8 +45,5 @@ class BaseGuardrailNodeDisplay(BaseNodeDisplay[_GuardrailNodeType], Generic[_Gua
                 "metric_definition_id": str(raise_if_descriptor(node.metric_definition)),
                 "release_tag": raise_if_descriptor(node.release_tag),
             },
-            "display_data": self.get_display_data().dict(),
-            "base": self.get_base().dict(),
-            "definition": self.get_definition().dict(),
-            "ports": self.serialize_ports(display_context),
+            **self.serialize_generic_fields(display_context),
         }

vellum_ee/workflows/display/nodes/vellum/inline_prompt_node.py

@@ -109,19 +109,12 @@ class BaseInlinePromptNodeDisplay(BaseNodeDisplay[_InlinePromptNodeType], Generi
                 },
                 "ml_model_name": ml_model,
             },
-            "display_data": self.get_display_data().dict(),
-            "base": self.get_base().dict(),
-            "definition": self.get_definition().dict(),
-            "trigger": {
-                "id": str(self.get_target_handle_id()),
-                "merge_behavior": node.Trigger.merge_behavior.value,
-            },
+            **self.serialize_generic_fields(display_context),
             "outputs": [
                 {"id": str(json_display.id), "name": "json", "type": "JSON", "value": None},
                 {"id": str(output_display.id), "name": "text", "type": "STRING", "value": None},
                 {"id": str(array_display.id), "name": "results", "type": "ARRAY", "value": None},
             ],
-            "ports": self.serialize_ports(display_context),
         }
         attributes = self._serialize_attributes(display_context)
         if attributes:

vellum_ee/workflows/display/nodes/vellum/inline_subworkflow_node.py

@@ -55,10 +55,7 @@ class BaseInlineSubworkflowNodeDisplay(
                 "input_variables": [workflow_input.dict() for workflow_input in workflow_inputs],
                 "output_variables": [workflow_output.dict() for workflow_output in workflow_outputs],
             },
-            "display_data": self.get_display_data().dict(),
-            "base": self.get_base().dict(),
-            "definition": self.get_definition().dict(),
-            "ports": self.serialize_ports(display_context),
+            **self.serialize_generic_fields(display_context),
         }
 
     def _generate_node_and_workflow_inputs(

vellum_ee/workflows/display/nodes/vellum/map_node.py

@@ -79,8 +79,5 @@ class BaseMapNodeDisplay(BaseAdornmentNodeDisplay[_MapNodeType], Generic[_MapNod
                 "item_input_id": item_workflow_input_id,
                 "index_input_id": index_workflow_input_id,
             },
-            "display_data": self.get_display_data().dict(),
-            "base": self.get_base().dict(),
-            "definition": self.get_definition().dict(),
-            "ports": self.serialize_ports(display_context),
+            **self.serialize_generic_fields(display_context),
         }

vellum_ee/workflows/display/nodes/vellum/merge_node.py

@@ -47,10 +47,7 @@ class BaseMergeNodeDisplay(BaseNodeDisplay[_MergeNodeType], Generic[_MergeNodeTy
                 "target_handles": [{"id": str(target_handle_id)} for target_handle_id in target_handle_ids],
                 "source_handle_id": str(self.get_source_handle_id(display_context.port_displays)),
             },
-            "display_data": self.get_display_data().dict(),
-            "base": self.get_base().dict(),
-            "definition": self.get_definition().dict(),
-            "ports": self.serialize_ports(display_context),
+            **self.serialize_generic_fields(display_context),
         }
 
     def get_target_handle_ids(self) -> Optional[List[UUID]]:

vellum_ee/workflows/display/nodes/vellum/note_node.py

@@ -13,7 +13,7 @@ class BaseNoteNodeDisplay(BaseNodeDisplay[_NoteNodeType], Generic[_NoteNodeType]
     style: ClassVar[Union[Dict[str, Any], None]] = None
 
     def serialize(self, display_context: WorkflowDisplayContext, **kwargs: Any) -> JsonObject:
-        del display_context, kwargs  # Unused parameters
+        del kwargs  # Unused parameters
         node_id = self.node_id
 
         return {
@@ -25,7 +25,5 @@ class BaseNoteNodeDisplay(BaseNodeDisplay[_NoteNodeType], Generic[_NoteNodeType]
                 "text": self.text,
                 "style": self.style,
             },
-            "display_data": self.get_display_data().dict(),
-            "base": self.get_base().dict(),
-            "definition": self.get_definition().dict(),
+            **self.serialize_generic_fields(display_context),
         }

vellum_ee/workflows/display/nodes/vellum/prompt_deployment_node.py

@@ -64,10 +64,7 @@ class BasePromptDeploymentNodeDisplay(BaseNodeDisplay[_PromptDeploymentNodeType]
                 "release_tag": raise_if_descriptor(node.release_tag),
                 "ml_model_fallbacks": list(ml_model_fallbacks) if ml_model_fallbacks else None,
             },
-            "display_data": self.get_display_data().dict(),
-            "base": self.get_base().dict(),
-            "definition": self.get_definition().dict(),
-            "ports": self.serialize_ports(display_context),
+            **self.serialize_generic_fields(display_context),
             "outputs": [
                 {"id": str(json_display.id), "name": "json", "type": "JSON", "value": None},
                 {"id": str(output_display.id), "name": "text", "type": "STRING", "value": None},

vellum_ee/workflows/display/nodes/vellum/search_node.py

@@ -74,10 +74,7 @@ class BaseSearchNodeDisplay(BaseNodeDisplay[_SearchNodeType], Generic[_SearchNod
                 "external_id_filters_node_input_id": str(node_inputs["external_id_filters"].id),
                 "metadata_filters_node_input_id": str(node_inputs["metadata_filters"].id),
             },
-            "display_data": self.get_display_data().dict(),
-            "base": self.get_base().dict(),
-            "definition": self.get_definition().dict(),
-            "ports": self.serialize_ports(display_context),
+            **self.serialize_generic_fields(display_context),
         }
 
     def _generate_search_node_inputs(

vellum_ee/workflows/display/nodes/vellum/subworkflow_deployment_node.py

@@ -61,8 +61,5 @@ class BaseSubworkflowDeploymentNodeDisplay(
                 "workflow_deployment_id": str(deployment.id),
                 "release_tag": raise_if_descriptor(node.release_tag),
             },
-            "display_data": self.get_display_data().dict(),
-            "base": self.get_base().dict(),
-            "definition": self.get_definition().dict(),
-            "ports": self.serialize_ports(display_context),
+            **self.serialize_generic_fields(display_context),
         }

vellum_ee/workflows/display/nodes/vellum/templating_node.py

@@ -66,8 +66,5 @@ class BaseTemplatingNodeDisplay(BaseNodeDisplay[_TemplatingNodeType], Generic[_T
                 "template_node_input_id": str(template_node_input.id),
                 "output_type": inferred_output_type,
             },
-            "display_data": self.get_display_data().dict(),
-            "base": self.get_base().dict(),
-            "definition": self.get_definition().dict(),
-            "ports": self.serialize_ports(display_context),
+            **self.serialize_generic_fields(display_context),
         }

vellum_ee/workflows/display/nodes/vellum/tests/test_code_execution_node.py

@@ -251,6 +251,7 @@ def test_serialize_node__with_non_exist_code_input_path_with_dry_run():
                         ],
                     },
                     "ports": [{"id": "7afa3858-f50c-4116-936a-a401e3b2c60f", "name": "default", "type": "DEFAULT"}],
+                    "trigger": {"id": "3a39ea63-9f86-4891-a902-0216a7190720", "merge_behavior": "AWAIT_ANY"},
                 },
             ],
             "edges": [

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_api_node_serialization.py

@@ -203,6 +203,10 @@ def test_serialize_workflow(vellum_client):
                 "name": "SimpleAPINode",
                 "module": ["tests", "workflows", "basic_api_node", "workflow"],
             },
+            "trigger": {
+                "id": "14b538a5-aedb-41f3-b579-039956b7c7ed",
+                "merge_behavior": "AWAIT_ANY",
+            },
             "ports": [{"id": "7c33b4d3-9204-4bd5-9371-80ee34f83073", "name": "default", "type": "DEFAULT"}],
         },
         api_node,

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_code_execution_node_serialization.py

@@ -110,6 +110,10 @@ def test_serialize_workflow_with_filepath():
             "module": ["tests", "workflows", "basic_code_execution_node", "workflow"],
             "name": "SimpleCodeExecutionNode",
         },
+        "trigger": {
+            "id": "e02a2701-22c0-4533-8b00-175998e7350a",
+            "merge_behavior": "AWAIT_ANY",
+        },
         "ports": [{"id": "832f81ec-427b-42a8-825c-e62c43c1f961", "name": "default", "type": "DEFAULT"}],
     }
     assert not DeepDiff(
@@ -342,6 +346,10 @@ def test_serialize_workflow_with_code():
             "name": "SimpleCodeExecutionNode",
             "module": ["tests", "workflows", "basic_code_execution_node", "workflow_with_code"],
         },
+        "trigger": {
+            "id": "e02a2701-22c0-4533-8b00-175998e7350a",
+            "merge_behavior": "AWAIT_ANY",
+        },
         "ports": [{"id": "832f81ec-427b-42a8-825c-e62c43c1f961", "name": "default", "type": "DEFAULT"}],
     }
     assert not DeepDiff(
@@ -602,6 +610,10 @@ def test_serialize_workflow__try_wrapped():
                 ],
             }
         ],
+        "trigger": {
+            "id": "e02a2701-22c0-4533-8b00-175998e7350a",
+            "merge_behavior": "AWAIT_ANY",
+        },
         "ports": [{"id": "832f81ec-427b-42a8-825c-e62c43c1f961", "name": "default", "type": "DEFAULT"}],
     }
 

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_conditional_node_serialization.py

@@ -424,6 +424,10 @@ def test_serialize_workflow():
                 "name": "CategoryConditionalNode",
                 "module": ["tests", "workflows", "basic_conditional_node", "workflow"],
             },
+            "trigger": {
+                "id": "dd89e228-a23e-422b-80b2-34362c1c050e",
+                "merge_behavior": "AWAIT_ANY",
+            },
             "ports": [
                 {
                     "id": "3a45b81f-95e4-4cbd-8997-bfdbe30251e8",
@@ -952,6 +956,10 @@ def test_conditional_node_serialize_all_operators_with_lhs_and_rhs(descriptor, o
                 "name": "SimpleConditionalNode",
                 "module": ["tests", "workflows", "basic_conditional_node", "workflow_with_only_one_conditional_node"],
             },
+            "trigger": {
+                "id": "c6e99e94-bc8e-47a4-b75c-cc96c6bedbb0",
+                "merge_behavior": "AWAIT_ANY",
+            },
             "ports": [
                 {
                     "id": "2ff87aa6-37cf-43dd-af9d-13b9198ab70a",
@@ -1063,6 +1071,10 @@ def test_conditional_node_serialize_all_operators_with_expression(descriptor, op
                 "name": "SimpleConditionalNode",
                 "module": ["tests", "workflows", "basic_conditional_node", "workflow_with_only_one_conditional_node"],
             },
+            "trigger": {
+                "id": "c6e99e94-bc8e-47a4-b75c-cc96c6bedbb0",
+                "merge_behavior": "AWAIT_ANY",
+            },
             "ports": [
                 {
                     "id": "2ff87aa6-37cf-43dd-af9d-13b9198ab70a",
@@ -1186,6 +1198,10 @@ def test_conditional_node_serialize_all_operators_with_value_and_start_and_end(d
                 "name": "SimpleConditionalNode",
                 "module": ["tests", "workflows", "basic_conditional_node", "workflow_with_only_one_conditional_node"],
             },
+            "trigger": {
+                "id": "c6e99e94-bc8e-47a4-b75c-cc96c6bedbb0",
+                "merge_behavior": "AWAIT_ANY",
+            },
             "ports": [
                 {
                     "id": "2ff87aa6-37cf-43dd-af9d-13b9198ab70a",

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_error_node_serialization.py

@@ -114,6 +114,11 @@ def test_serialize_workflow():
                 "name": "FailNode",
                 "module": ["tests", "workflows", "basic_error_node", "workflow"],
             },
+            "trigger": {
+                "id": "70c19f1c-309c-4a5d-ba65-664c0bb2fedf",
+                "merge_behavior": "AWAIT_ATTRIBUTES",
+            },
+            "ports": [],
         },
         error_node,
         ignore_order=True,

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_guardrail_node_serialization.py

@@ -117,6 +117,10 @@ def test_serialize_workflow():
             "module": ["tests", "workflows", "basic_guardrail_node", "workflow"],
             "name": "ExampleGuardrailNode",
         },
+        "trigger": {
+            "id": "ce5b85b1-eded-46dd-b4b7-020afcdc67ab",
+            "merge_behavior": "AWAIT_ANY",
+        },
         "ports": [{"id": "0ed87407-697e-4ae9-ab9b-6c5cc2e57cf7", "name": "default", "type": "DEFAULT"}],
     }
 

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_inline_subworkflow_serialization.py

@@ -324,6 +324,10 @@ def test_serialize_workflow():
                 "name": "ExampleInlineSubworkflowNode",
                 "module": ["tests", "workflows", "basic_inline_subworkflow", "workflow"],
             },
+            "trigger": {
+                "id": "859a75a6-1bd2-4350-9509-4af66245e8e4",
+                "merge_behavior": "AWAIT_ATTRIBUTES",
+            },
             "ports": [{"id": "cfd831bc-ee7f-44d0-8d76-0ba0cd0277dc", "name": "default", "type": "DEFAULT"}],
         },
         subworkflow_node,

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_map_node_serialization.py

@@ -286,6 +286,10 @@ def test_serialize_workflow():
                 "name": "MapFruitsNode",
                 "module": ["tests", "workflows", "basic_map_node", "workflow"],
             },
+            "trigger": {
+                "id": "b5e8182e-20c5-482b-b4c5-4dde48c01472",
+                "merge_behavior": "AWAIT_ATTRIBUTES",
+            },
             "ports": [{"id": "a2171a61-0657-43ad-b6d9-cf93ce3270d0", "name": "default", "type": "DEFAULT"}],
         },
         map_node,

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_merge_node_serialization.py

@@ -83,6 +83,10 @@ def test_serialize_workflow__await_all():
                 "module": ["tests", "workflows", "basic_merge_node", "await_all_workflow"],
                 "name": "AwaitAllMergeNode",
             },
+            "trigger": {
+                "id": "0efd256f-f5f6-45fe-9adb-651780f5e63d",
+                "merge_behavior": "AWAIT_ALL",
+            },
             "ports": [{"id": "3bbc469f-0fb0-4b3d-a28b-746fefec2818", "name": "default", "type": "DEFAULT"}],
         },
         merge_node,

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_prompt_deployment_serialization.py

@@ -150,6 +150,10 @@ def test_serialize_workflow(vellum_client):
             "name": "ExamplePromptDeploymentNode",
             "module": ["tests", "workflows", "basic_text_prompt_deployment", "workflow"],
         },
+        "trigger": {
+            "id": "b7605c48-0937-4ecc-914e-0d1058130e65",
+            "merge_behavior": "AWAIT_ANY",
+        },
         "ports": [{"id": "2f26c7e0-283d-4f04-b639-adebb56bc679", "name": "default", "type": "DEFAULT"}],
         "outputs": [
             {"id": "180355a8-e67c-4ce6-9ac3-e5dbb75a6629", "name": "json", "type": "JSON", "value": None},
@@ -384,6 +388,10 @@ def test_serialize_workflow_with_prompt_and_templating(vellum_client):
                 "workflow_with_prompt_deployment_json_reference",
             ],
         },
+        "trigger": {
+            "id": "b7605c48-0937-4ecc-914e-0d1058130e65",
+            "merge_behavior": "AWAIT_ANY",
+        },
         "ports": [{"id": "2f26c7e0-283d-4f04-b639-adebb56bc679", "name": "default", "type": "DEFAULT"}],
         "outputs": [
             {"id": "180355a8-e67c-4ce6-9ac3-e5dbb75a6629", "name": "json", "type": "JSON", "value": None},
@@ -479,6 +487,10 @@ def test_serialize_workflow_with_prompt_and_templating(vellum_client):
                 "workflow_with_prompt_deployment_json_reference",
             ],
         },
+        "trigger": {
+            "id": "58427684-3848-498a-8299-c6b0fc70265d",
+            "merge_behavior": "AWAIT_ATTRIBUTES",
+        },
         "ports": [{"id": "39317827-df43-4f5a-bfbc-20bffc839748", "name": "default", "type": "DEFAULT"}],
     }
 

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_search_node_serialization.py

@@ -238,6 +238,10 @@ def test_serialize_workflow():
             "name": "SimpleSearchNode",
             "module": ["tests", "workflows", "basic_search_node", "workflow"],
         },
+        "trigger": {
+            "id": "6d50305f-588b-469f-a042-b0767d3f99b1",
+            "merge_behavior": "AWAIT_ANY",
+        },
         "ports": [{"id": "00ae06b3-f8d9-4ae6-9fbf-e4ff4d520e9b", "name": "default", "type": "DEFAULT"}],
     }
 

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_subworkflow_deployment_serialization.py

@@ -153,6 +153,10 @@ def test_serialize_workflow(vellum_client):
             "module": ["tests", "workflows", "basic_subworkflow_deployment", "workflow"],
             "name": "ExampleSubworkflowDeploymentNode",
         },
+        "trigger": {
+            "id": "e4d80502-9281-42c8-91e3-10817bcd7d9e",
+            "merge_behavior": "AWAIT_ANY",
+        },
         "ports": [{"id": "ab0db8a9-7b53-4d88-8667-273b31303273", "name": "default", "type": "DEFAULT"}],
     }
 

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_templating_node_serialization.py

@@ -112,6 +112,10 @@ def test_serialize_workflow():
                 "name": "ExampleTemplatingNode",
                 "module": ["tests", "workflows", "basic_templating_node", "workflow_with_json_input"],
             },
+            "trigger": {
+                "id": "58427684-3848-498a-8299-c6b0fc70265d",
+                "merge_behavior": "AWAIT_ATTRIBUTES",
+            },
             "ports": [{"id": "39317827-df43-4f5a-bfbc-20bffc839748", "name": "default", "type": "DEFAULT"}],
         },
         templating_node,

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_terminal_node_serialization.py

@@ -103,4 +103,9 @@ def test_serialize_workflow():
                 "value": {"type": "WORKFLOW_INPUT", "input_variable_id": "e39a7b63-de15-490a-ae9b-8112c767aea0"},
             }
         ],
+        "trigger": {
+            "id": "0173d3c6-11d1-44b7-b070-ca9ff5119046",
+            "merge_behavior": "AWAIT_ANY",
+        },
+        "ports": [],
     }

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_composio_serialization.py

@@ -54,6 +54,7 @@ def test_serialize_workflow():
         "action": "GITHUB_CREATE_AN_ISSUE",
         "description": "Create a new issue in a GitHub repository",
         "user_id": None,
+        "name": "github_create_an_issue",
     }
 
     # AND the rest of the node structure should be correct

vellum_ee/workflows/display/tests/workflow_serialization/test_complex_terminal_node_serialization.py

@@ -81,6 +81,11 @@ def test_serialize_workflow__missing_final_output_node():
                         },
                     }
                 ],
+                "trigger": {
+                    "id": "a0c2eb7a-398e-4f28-b63d-f3bae9b563ee",
+                    "merge_behavior": "AWAIT_ANY",
+                },
+                "ports": [],
             },
             {
                 "id": "bb88768d-472e-4997-b7ea-de09163d1b4c",

vellum_ee/workflows/display/utils/events.py

@@ -0,0 +1,24 @@
+from vellum.workflows.events.workflow import WorkflowExecutionInitiatedEvent
+from vellum_ee.workflows.display.utils.registry import (
+    get_parent_display_context_from_event,
+    register_workflow_display_class,
+    register_workflow_display_context,
+)
+from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
+
+
+def event_enricher(event: WorkflowExecutionInitiatedEvent) -> WorkflowExecutionInitiatedEvent:
+    workflow_definition = event.body.workflow_definition
+    workflow_display = get_workflow_display(
+        workflow_class=workflow_definition,
+        parent_display_context=get_parent_display_context_from_event(event),
+        dry_run=True,
+    )
+    register_workflow_display_context(event.span_id, workflow_display.display_context)
+
+    if event.body.workflow_definition.is_dynamic:
+        register_workflow_display_class(workflow_definition, workflow_display.__class__)
+        workflow_version_exec_config = workflow_display.serialize()
+        setattr(event.body, "workflow_version_exec_config", workflow_version_exec_config)
+
+    return event

vellum_ee/workflows/display/utils/tests/test_events.py

@@ -0,0 +1,69 @@
+import pytest
+from uuid import uuid4
+from typing import Optional
+
+from vellum.workflows.events.workflow import WorkflowExecutionInitiatedBody, WorkflowExecutionInitiatedEvent
+from vellum.workflows.inputs.base import BaseInputs
+from vellum.workflows.workflows.base import BaseWorkflow
+from vellum_ee.workflows.display.utils.events import event_enricher
+
+
+@pytest.mark.parametrize(
+    "is_dynamic,expected_config",
+    [
+        (False, None),
+        (
+            True,
+            {
+                "workflow_raw_data": {
+                    "nodes": [
+                        {
+                            "id": "86607b18-7872-49f3-a592-fda1428f70aa",
+                            "type": "ENTRYPOINT",
+                            "inputs": [],
+                            "data": {
+                                "label": "Entrypoint Node",
+                                "source_handle_id": "d1fe8f4c-53d7-43a0-b210-73ebdc60bf57",
+                            },
+                            "display_data": {"position": {"x": 0.0, "y": -50.0}},
+                            "base": None,
+                            "definition": None,
+                        }
+                    ],
+                    "edges": [],
+                    "display_data": {"viewport": {"x": 0.0, "y": 0.0, "zoom": 1.0}},
+                    "definition": {
+                        "name": "TestWorkflow",
+                        "module": ["vellum_ee", "workflows", "display", "utils", "tests", "test_events"],
+                    },
+                    "output_values": [],
+                },
+                "input_variables": [],
+                "state_variables": [],
+                "output_variables": [],
+            },
+        ),
+    ],
+)
+def test_event_enricher_static_workflow(is_dynamic: bool, expected_config: Optional[dict]):
+    """Test event_enricher with a static workflow (is_dynamic=False)."""
+    # GIVEN a workflow class with the specified is_dynamic value
+    _is_dynamic = is_dynamic
+
+    class TestWorkflow(BaseWorkflow):
+        is_dynamic = _is_dynamic
+
+    event: WorkflowExecutionInitiatedEvent = WorkflowExecutionInitiatedEvent(
+        trace_id=uuid4(),
+        span_id=uuid4(),
+        body=WorkflowExecutionInitiatedBody(
+            workflow_definition=TestWorkflow,
+            inputs=BaseInputs(),
+        ),
+    )
+
+    # WHEN the event_enricher is called with mocked dependencies
+    event_enricher(event)
+
+    # AND workflow_version_exec_config is set to the expected config
+    assert event.body.workflow_version_exec_config == expected_config