vellum-ai 0.14.2__py3-none-any.whl → 0.14.4__py3-none-any.whl

vellum_ee/workflows/display/workflows/vellum_workflow_display.py

@@ -31,8 +31,6 @@ from vellum_ee.workflows.display.vellum import (
     WorkflowInputsVellumDisplayOverrides,
     WorkflowMetaVellumDisplay,
     WorkflowMetaVellumDisplayOverrides,
-    WorkflowOutputVellumDisplay,
-    WorkflowOutputVellumDisplayOverrides,
 )
 from vellum_ee.workflows.display.workflows.base_workflow_display import BaseWorkflowDisplay
 
@@ -53,8 +51,6 @@ class VellumWorkflowDisplay(
         EntrypointVellumDisplayOverrides,
         EdgeVellumDisplay,
         EdgeVellumDisplayOverrides,
-        WorkflowOutputVellumDisplay,
-        WorkflowOutputVellumDisplayOverrides,
     ]
 ):
     node_display_base_class = BaseNodeDisplay
@@ -144,7 +140,7 @@ class VellumWorkflowDisplay(
 
         # Add a synthetic Terminal Node and track the Workflow's output variables for each Workflow output
         for workflow_output, workflow_output_display in self.display_context.workflow_output_displays.items():
-            final_output_node_id = workflow_output_display.node_id
+            final_output_node_id = uuid4_from_hash(f"{self.workflow_id}|node_id|{workflow_output.name}")
             inferred_type = infer_vellum_variable_type(workflow_output)
 
             # Remove the terminal node output from the unreferenced set
@@ -176,22 +172,17 @@ class VellumWorkflowDisplay(
                     except IndexError:
                         source_node_display = None
 
-                synthetic_target_handle_id = (
-                    str(workflow_output_display.target_handle_id)
-                    if workflow_output_display.target_handle_id
-                    else str(uuid4_from_hash(f"{self.workflow_id}|target_handle_id|{workflow_output_display.name}"))
-                )
-                synthetic_display_data = (
-                    workflow_output_display.display_data.dict()
-                    if workflow_output_display.display_data
-                    else NodeDisplayData().dict()
+                synthetic_target_handle_id = str(
+                    uuid4_from_hash(f"{self.workflow_id}|target_handle_id|{workflow_output_display.name}")
                 )
+                synthetic_display_data = NodeDisplayData().dict()
+                synthetic_node_label = "Final Output"
                 nodes.append(
                     {
                         "id": str(final_output_node_id),
                         "type": "TERMINAL",
                         "data": {
-                            "label": workflow_output_display.label,
+                            "label": synthetic_node_label,
                             "name": workflow_output_display.name,
                             "target_handle_id": synthetic_target_handle_id,
                             "output_id": str(workflow_output_display.id),
@@ -220,7 +211,7 @@ class VellumWorkflowDisplay(
                             "id": str(uuid4_from_hash(f"{self.workflow_id}|edge_id|{workflow_output_display.name}")),
                             "source_node_id": str(source_node_display.node_id),
                             "source_handle_id": str(source_handle_id),
-                            "target_node_id": str(workflow_output_display.node_id),
+                            "target_node_id": str(final_output_node_id),
                             "target_handle_id": synthetic_target_handle_id,
                             "type": "DEFAULT",
                         }
@@ -365,31 +356,6 @@ class VellumWorkflowDisplay(
 
         return EntrypointVellumDisplay(id=entrypoint_id, edge_display=edge_display)
 
-    def _generate_workflow_output_display(
-        self,
-        output: BaseDescriptor,
-        overrides: Optional[WorkflowOutputVellumDisplayOverrides] = None,
-    ) -> WorkflowOutputVellumDisplay:
-        if overrides:
-            return WorkflowOutputVellumDisplay(
-                id=overrides.id,
-                name=overrides.name,
-                label=overrides.label,
-                node_id=overrides.node_id,
-                target_handle_id=overrides.target_handle_id,
-                display_data=overrides.display_data,
-            )
-
-        output_id = uuid4_from_hash(f"{self.workflow_id}|id|{output.name}")
-        node_id = uuid4_from_hash(f"{self.workflow_id}|node_id|{output.name}")
-
-        return WorkflowOutputVellumDisplay(
-            id=output_id,
-            node_id=node_id,
-            name=output.name,
-            label="Final Output",
-        )
-
     def _generate_edge_display(
         self,
         edge: Edge,

vellum_ee/workflows/tests/local_workflow/display/workflow.py

@@ -47,8 +47,6 @@ class WorkflowDisplay(VellumWorkflowDisplay[Workflow]):
     output_displays = {
         Workflow.Outputs.final_output: WorkflowOutputVellumDisplayOverrides(
             id=UUID("5469b810-6ea6-4362-9e79-e360d44a1405"),
-            node_id=UUID("f3ef4b2b-fec9-4026-9cc6-e5eac295307f"),
             name="final-output",
-            label="Final Output",
         )
     }