vellum-ai 1.2.1__tar.gz → 1.2.2__tar.gz

{vellum_ai-1.2.1 → vellum_ai-1.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 1.2.1
+Version: 1.2.2
 Summary: 
 License: MIT
 Requires-Python: >=3.9,<4.0

{vellum_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/vellum_ee/workflows/display/nodes/vellum/error_node.py

@@ -2,7 +2,7 @@ from uuid import UUID
 from typing import Any, ClassVar, Generic, Optional, TypeVar
 
 from vellum.workflows.nodes import ErrorNode
-from vellum.workflows.types.core import JsonObject
+from vellum.workflows.types.core import JsonArray, JsonObject
 from vellum_ee.workflows.display.nodes.base_node_display import BaseNodeDisplay
 from vellum_ee.workflows.display.nodes.utils import raise_if_descriptor
 from vellum_ee.workflows.display.nodes.vellum.utils import create_node_input
@@ -20,6 +20,10 @@ class BaseErrorNodeDisplay(BaseNodeDisplay[_ErrorNodeType], Generic[_ErrorNodeTy
 
     __serializable_inputs__ = {ErrorNode.error}
 
+    def serialize_ports(self, display_context: "WorkflowDisplayContext") -> JsonArray:
+        """Error nodes have no ports."""
+        return []
+
     def serialize(self, display_context: WorkflowDisplayContext, **_kwargs) -> JsonObject:
         node_id = self.node_id
         error_source_input_id = self.node_input_ids_by_name.get(
@@ -47,9 +51,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_ai-1.2.1 → vellum_ai-1.2.2}/ee/vellum_ee/workflows/display/nodes/vellum/final_output_node.py

@@ -2,7 +2,7 @@ from uuid import UUID
 from typing import ClassVar, Generic, Optional, TypeVar
 
 from vellum.workflows.nodes.displayable.final_output_node import FinalOutputNode
-from vellum.workflows.types.core import JsonObject
+from vellum.workflows.types.core import JsonArray, JsonObject
 from vellum.workflows.utils.uuids import uuid4_from_hash
 from vellum_ee.workflows.display.nodes.base_node_display import BaseNodeDisplay
 from vellum_ee.workflows.display.nodes.utils import to_kebab_case
@@ -19,6 +19,10 @@ NODE_INPUT_KEY = "node_input"
 class BaseFinalOutputNodeDisplay(BaseNodeDisplay[_FinalOutputNodeType], Generic[_FinalOutputNodeType]):
     output_name: ClassVar[Optional[str]] = None
 
+    def serialize_ports(self, display_context: "WorkflowDisplayContext") -> JsonArray:
+        """Final output nodes have no ports."""
+        return []
+
     def serialize(self, display_context: WorkflowDisplayContext, **_kwargs) -> JsonObject:
         node = self._node
         node_id = self.node_id
@@ -45,9 +49,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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/ee/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_ai-1.2.1 → vellum_ai-1.2.2}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-ai"
 
 [tool.poetry]
 name = "vellum-ai"
-version = "1.2.1"
+version = "1.2.2"
 description = ""
 readme = "README.md"
 authors = []

{vellum_ai-1.2.1 → vellum_ai-1.2.2}/src/vellum/client/core/client_wrapper.py

@@ -27,10 +27,10 @@ class BaseClientWrapper:
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "vellum-ai/1.2.1",
+            "User-Agent": "vellum-ai/1.2.2",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "1.2.1",
+            "X-Fern-SDK-Version": "1.2.2",
             **(self.get_custom_headers() or {}),
         }
         if self._api_version is not None: