vellum-ai 0.14.49__py3-none-any.whl → 0.14.51__py3-none-any.whl

vellum_ee/workflows/display/workflows/base_workflow_display.py

@@ -153,20 +153,8 @@ class BaseWorkflowDisplay(Generic[WorkflowType]):
             "definition": None,
         }
 
-        # Add all the nodes in the workflow
-        for node in self._workflow.get_nodes():
-            node_display = self.display_context.node_displays[node]
-
-            try:
-                serialized_node = node_display.serialize(self.display_context)
-            except NotImplementedError as e:
-                self.add_error(e)
-                continue
-
-            serialized_nodes[node_display.node_id] = serialized_node
-
-        # Add all unused nodes in the workflow
-        for node in self._workflow.get_unused_nodes():
+        # Add all the nodes in the workflows
+        for node in self._workflow.get_all_nodes():
             node_display = self.display_context.node_displays[node]
 
             try:
@@ -417,16 +405,7 @@ class BaseWorkflowDisplay(Generic[WorkflowType]):
 
         port_displays: PortDisplays = {}
 
-        for node in self._workflow.get_nodes():
-            self._enrich_node_displays(
-                node=node,
-                node_displays=node_displays,
-                global_node_displays=global_node_displays,
-                global_node_output_displays=global_node_output_displays,
-                port_displays=port_displays,
-            )
-
-        for node in self._workflow.get_unused_nodes():
+        for node in self._workflow.get_all_nodes():
             self._enrich_node_displays(
                 node=node,
                 node_displays=node_displays,

vellum_ee/workflows/display/workflows/tests/test_workflow_display.py

@@ -466,7 +466,7 @@ def test_serialize_workflow__array_values():
     assert "value" in array_output
     assert array_output["value"] == {
         "type": "CONSTANT_VALUE",
-        "value": {"type": "JSON", "items": ["item1", "item2", "item3"]},
+        "value": {"type": "JSON", "value": ["item1", "item2", "item3"]},
     }
 
     nested_array_outputs = [val for val in outputs if isinstance(val, dict) and val["name"] == "nested_array_value"]
@@ -477,7 +477,7 @@ def test_serialize_workflow__array_values():
     assert "value" in nested_array_output
     assert nested_array_output["value"] == {
         "type": "CONSTANT_VALUE",
-        "value": {"type": "JSON", "items": [["item1", "item2", "item3"], ["item4", "item5", "item6"]]},
+        "value": {"type": "JSON", "value": [["item1", "item2", "item3"], ["item4", "item5", "item6"]]},
     }
 
     mixed_array_outputs = [val for val in outputs if isinstance(val, dict) and val["name"] == "mixed_array_value"]
@@ -488,7 +488,7 @@ def test_serialize_workflow__array_values():
     assert "value" in mixed_array_output
     assert mixed_array_output["value"] == {
         "type": "CONSTANT_VALUE",
-        "value": {"type": "JSON", "items": [["item1"], "item2", "item3"]},
+        "value": {"type": "JSON", "value": [["item1"], "item2", "item3"]},
     }