vellum-ai 0.14.23__py3-none-any.whl → 0.14.25__py3-none-any.whl

vellum_ee/workflows/display/utils/vellum.py

@@ -1,6 +1,9 @@
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING, Any, Literal, Optional, Union
 
+from vellum.client.core.pydantic_utilities import UniversalBaseModel
+from vellum.client.types.array_vellum_value import ArrayVellumValue
 from vellum.client.types.logical_operator import LogicalOperator
+from vellum.client.types.vellum_value import VellumValue
 from vellum.client.types.vellum_variable_type import VellumVariableType
 from vellum.workflows.descriptors.base import BaseDescriptor
 from vellum.workflows.expressions.and_ import AndExpression
@@ -39,23 +42,66 @@ from vellum.workflows.references.vellum_secret import VellumSecretReference
 from vellum.workflows.utils.vellum_variables import primitive_type_to_vellum_variable_type
 from vellum.workflows.vellum_client import create_vellum_client
 from vellum_ee.workflows.display.utils.expressions import get_child_descriptor
-from vellum_ee.workflows.display.vellum import (
-    ConstantValuePointer,
-    ExecutionCounterData,
-    ExecutionCounterPointer,
-    InputVariableData,
-    InputVariablePointer,
-    NodeInputValuePointerRule,
-    NodeOutputData,
-    NodeOutputPointer,
-    WorkspaceSecretData,
-    WorkspaceSecretPointer,
-)
 
 if TYPE_CHECKING:
     from vellum_ee.workflows.display.types import WorkflowDisplayContext
 
 
+class ConstantValuePointer(UniversalBaseModel):
+    type: Literal["CONSTANT_VALUE"] = "CONSTANT_VALUE"
+    data: VellumValue
+
+
+ArrayVellumValue.model_rebuild()
+
+
+class NodeOutputData(UniversalBaseModel):
+    node_id: str
+    output_id: str
+
+
+class NodeOutputPointer(UniversalBaseModel):
+    type: Literal["NODE_OUTPUT"] = "NODE_OUTPUT"
+    data: NodeOutputData
+
+
+class InputVariableData(UniversalBaseModel):
+    input_variable_id: str
+
+
+class InputVariablePointer(UniversalBaseModel):
+    type: Literal["INPUT_VARIABLE"] = "INPUT_VARIABLE"
+    data: InputVariableData
+
+
+class WorkspaceSecretData(UniversalBaseModel):
+    type: VellumVariableType
+    workspace_secret_id: Optional[str] = None
+
+
+class WorkspaceSecretPointer(UniversalBaseModel):
+    type: Literal["WORKSPACE_SECRET"] = "WORKSPACE_SECRET"
+    data: WorkspaceSecretData
+
+
+class ExecutionCounterData(UniversalBaseModel):
+    node_id: str
+
+
+class ExecutionCounterPointer(UniversalBaseModel):
+    type: Literal["EXECUTION_COUNTER"] = "EXECUTION_COUNTER"
+    data: ExecutionCounterData
+
+
+NodeInputValuePointerRule = Union[
+    NodeOutputPointer,
+    InputVariablePointer,
+    ConstantValuePointer,
+    WorkspaceSecretPointer,
+    ExecutionCounterPointer,
+]
+
+
 def infer_vellum_variable_type(value: Any) -> VellumVariableType:
     inferred_type: VellumVariableType
 

vellum_ee/workflows/display/vellum.py

@@ -1,43 +1,24 @@
-from dataclasses import dataclass, field
+from dataclasses import dataclass
 from uuid import UUID
-from typing import List, Literal, Optional, Union
+from typing import List, Literal, Optional
 
-from pydantic import Field
-
-from vellum import VellumVariableType
-from vellum.client.types.array_vellum_value import ArrayVellumValue
-from vellum.client.types.vellum_value import VellumValue
 from vellum.core import UniversalBaseModel
 from vellum_ee.workflows.display.base import (
-    EdgeDisplay,
     EdgeDisplayOverrides,
-    EntrypointDisplay,
     EntrypointDisplayOverrides,
     StateValueDisplay,
     StateValueDisplayOverrides,
     WorkflowInputsDisplay,
     WorkflowInputsDisplayOverrides,
-    WorkflowMetaDisplay,
     WorkflowMetaDisplayOverrides,
     WorkflowOutputDisplayOverrides,
 )
-
-
-class NodeDisplayPosition(UniversalBaseModel):
-    x: float = 0.0
-    y: float = 0.0
-
-
-class NodeDisplayComment(UniversalBaseModel):
-    value: Optional[str] = None
-    expanded: Optional[bool] = None
-
-
-class NodeDisplayData(UniversalBaseModel):
-    position: NodeDisplayPosition = Field(default_factory=NodeDisplayPosition)
-    width: Optional[int] = None
-    height: Optional[int] = None
-    comment: Optional[NodeDisplayComment] = None
+from vellum_ee.workflows.display.base import WorkflowDisplayData  # noqa: F401 - Remove in 0.15.0
+from vellum_ee.workflows.display.base import WorkflowDisplayDataViewport  # noqa: F401 - Remove in 0.15.0
+from vellum_ee.workflows.display.editor.types import NodeDisplayComment  # noqa: F401 - Remove in 0.15.0
+from vellum_ee.workflows.display.editor.types import NodeDisplayData
+from vellum_ee.workflows.display.editor.types import NodeDisplayPosition  # noqa: F401 - Remove in 0.15.0
+from vellum_ee.workflows.display.utils.vellum import NodeInputValuePointerRule
 
 
 class CodeResourceDefinition(UniversalBaseModel):
@@ -45,26 +26,21 @@ class CodeResourceDefinition(UniversalBaseModel):
     module: List[str]
 
 
-class WorkflowDisplayDataViewport(UniversalBaseModel):
-    x: float = 0.0
-    y: float = 0.0
-    zoom: float = 1.0
-
-
-class WorkflowDisplayData(UniversalBaseModel):
-    viewport: WorkflowDisplayDataViewport = Field(default_factory=WorkflowDisplayDataViewport)
-
-
 @dataclass
-class WorkflowMetaVellumDisplayOverrides(WorkflowMetaDisplay, WorkflowMetaDisplayOverrides):
-    entrypoint_node_id: UUID
-    entrypoint_node_source_handle_id: UUID
-    entrypoint_node_display: NodeDisplayData
-    display_data: WorkflowDisplayData = field(default_factory=WorkflowDisplayData)
+class WorkflowMetaVellumDisplayOverrides(WorkflowMetaDisplayOverrides):
+    """
+    DEPRECATED: Use WorkflowMetaDisplay instead. Will be removed in 0.15.0
+    """
+
+    pass
 
 
 @dataclass
 class WorkflowMetaVellumDisplay(WorkflowMetaVellumDisplayOverrides):
+    """
+    DEPRECATED: Use WorkflowMetaDisplay instead. Will be removed in 0.15.0
+    """
+
     pass
 
 
@@ -115,13 +91,21 @@ class EdgeVellumDisplay(EdgeVellumDisplayOverrides):
 
 
 @dataclass
-class EntrypointVellumDisplayOverrides(EntrypointDisplay, EntrypointDisplayOverrides):
-    edge_display: EdgeDisplay
+class EntrypointVellumDisplayOverrides(EntrypointDisplayOverrides):
+    """
+    DEPRECATED: Use EntrypointDisplay instead. Will be removed in 0.15.0
+    """
+
+    pass
 
 
 @dataclass
 class EntrypointVellumDisplay(EntrypointVellumDisplayOverrides):
-    edge_display: EdgeDisplay
+    """
+    DEPRECATED: Use EntrypointDisplay instead. Will be removed in 0.15.0
+    """
+
+    pass
 
 
 @dataclass
@@ -145,61 +129,6 @@ class WorkflowOutputVellumDisplay(WorkflowOutputVellumDisplayOverrides):
     pass
 
 
-class ConstantValuePointer(UniversalBaseModel):
-    type: Literal["CONSTANT_VALUE"] = "CONSTANT_VALUE"
-    data: VellumValue
-
-
-ArrayVellumValue.model_rebuild()
-
-
-class NodeOutputData(UniversalBaseModel):
-    node_id: str
-    output_id: str
-
-
-class NodeOutputPointer(UniversalBaseModel):
-    type: Literal["NODE_OUTPUT"] = "NODE_OUTPUT"
-    data: NodeOutputData
-
-
-class InputVariableData(UniversalBaseModel):
-    input_variable_id: str
-
-
-class InputVariablePointer(UniversalBaseModel):
-    type: Literal["INPUT_VARIABLE"] = "INPUT_VARIABLE"
-    data: InputVariableData
-
-
-class WorkspaceSecretData(UniversalBaseModel):
-    type: VellumVariableType
-    workspace_secret_id: Optional[str] = None
-
-
-class WorkspaceSecretPointer(UniversalBaseModel):
-    type: Literal["WORKSPACE_SECRET"] = "WORKSPACE_SECRET"
-    data: WorkspaceSecretData
-
-
-class ExecutionCounterData(UniversalBaseModel):
-    node_id: str
-
-
-class ExecutionCounterPointer(UniversalBaseModel):
-    type: Literal["EXECUTION_COUNTER"] = "EXECUTION_COUNTER"
-    data: ExecutionCounterData
-
-
-NodeInputValuePointerRule = Union[
-    NodeOutputPointer,
-    InputVariablePointer,
-    ConstantValuePointer,
-    WorkspaceSecretPointer,
-    ExecutionCounterPointer,
-]
-
-
 class NodeInputValuePointer(UniversalBaseModel):
     rules: List[NodeInputValuePointerRule]
     combinator: Literal["OR"] = "OR"

vellum_ee/workflows/display/workflows/base_workflow_display.py

@@ -19,22 +19,29 @@ from vellum.workflows.types.generics import WorkflowType
 from vellum.workflows.utils.uuids import uuid4_from_hash
 from vellum_ee.workflows.display.base import (
     EdgeDisplay,
-    EntrypointDisplayOverridesType,
-    EntrypointDisplayType,
+    EntrypointDisplay,
     StateValueDisplayOverridesType,
     StateValueDisplayType,
     WorkflowInputsDisplayOverridesType,
     WorkflowInputsDisplayType,
-    WorkflowMetaDisplayOverridesType,
-    WorkflowMetaDisplayType,
+    WorkflowMetaDisplay,
     WorkflowOutputDisplay,
 )
+from vellum_ee.workflows.display.editor.types import NodeDisplayData
 from vellum_ee.workflows.display.nodes.base_node_display import BaseNodeDisplay
 from vellum_ee.workflows.display.nodes.base_node_vellum_display import BaseNodeVellumDisplay
 from vellum_ee.workflows.display.nodes.get_node_display_class import get_node_display_class
-from vellum_ee.workflows.display.nodes.types import NodeOutputDisplay, PortDisplay, PortDisplayOverrides
+from vellum_ee.workflows.display.nodes.types import NodeOutputDisplay, PortDisplay
 from vellum_ee.workflows.display.nodes.utils import raise_if_descriptor
-from vellum_ee.workflows.display.types import WorkflowDisplayContext
+from vellum_ee.workflows.display.types import (
+    EdgeDisplays,
+    EntrypointDisplays,
+    NodeDisplays,
+    NodeOutputDisplays,
+    PortDisplays,
+    WorkflowDisplayContext,
+    WorkflowOutputDisplays,
+)
 from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
 
 logger = logging.getLogger(__name__)
@@ -43,18 +50,14 @@ logger = logging.getLogger(__name__)
 class BaseWorkflowDisplay(
     Generic[
         WorkflowType,
-        WorkflowMetaDisplayType,
-        WorkflowMetaDisplayOverridesType,
         WorkflowInputsDisplayType,
         WorkflowInputsDisplayOverridesType,
         StateValueDisplayType,
         StateValueDisplayOverridesType,
-        EntrypointDisplayType,
-        EntrypointDisplayOverridesType,
     ]
 ):
     # Used to specify the display data for a workflow.
-    workflow_display: Optional[WorkflowMetaDisplayOverridesType] = None
+    workflow_display: Optional[WorkflowMetaDisplay] = None
 
     # Used to explicitly specify display data for a workflow's inputs.
     inputs_display: Dict[WorkflowInputReference, WorkflowInputsDisplayOverridesType] = {}
@@ -63,16 +66,16 @@ class BaseWorkflowDisplay(
     state_value_displays: Dict[StateValueReference, StateValueDisplayOverridesType] = {}
 
     # Used to explicitly specify display data for a workflow's entrypoints.
-    entrypoint_displays: Dict[Type[BaseNode], EntrypointDisplayOverridesType] = {}
+    entrypoint_displays: EntrypointDisplays = {}
 
     # Used to explicitly specify display data for a workflow's outputs.
-    output_displays: Dict[BaseDescriptor, WorkflowOutputDisplay] = {}
+    output_displays: WorkflowOutputDisplays = {}
 
     # Used to explicitly specify display data for a workflow's edges.
-    edge_displays: Dict[Tuple[Port, Type[BaseNode]], EdgeDisplay] = {}
+    edge_displays: EdgeDisplays = {}
 
     # Used to explicitly specify display data for a workflow's ports.
-    port_displays: Dict[Port, PortDisplayOverrides] = {}
+    port_displays: PortDisplays = {}
 
     # Used to store the mapping between workflows and their display classes
     _workflow_display_registry: Dict[Type[WorkflowType], Type["BaseWorkflowDisplay"]] = {}
@@ -87,10 +90,8 @@ class BaseWorkflowDisplay(
         *,
         parent_display_context: Optional[
             WorkflowDisplayContext[
-                WorkflowMetaDisplayType,
                 WorkflowInputsDisplayType,
                 StateValueDisplayType,
-                EntrypointDisplayType,
             ]
         ] = None,
         dry_run: bool = False,
@@ -183,50 +184,40 @@ class BaseWorkflowDisplay(
     def display_context(
         self,
     ) -> WorkflowDisplayContext[
-        WorkflowMetaDisplayType,
         WorkflowInputsDisplayType,
         StateValueDisplayType,
-        EntrypointDisplayType,
     ]:
-        workflow_display = self._generate_workflow_meta_display()
+        workflow_meta_display = self._generate_workflow_meta_display()
 
-        global_node_output_displays: Dict[OutputReference, Tuple[Type[BaseNode], NodeOutputDisplay]] = (
+        global_node_output_displays: NodeOutputDisplays = (
             copy(self._parent_display_context.global_node_output_displays) if self._parent_display_context else {}
         )
 
-        node_displays: Dict[Type[BaseNode], BaseNodeDisplay] = {}
+        node_displays: NodeDisplays = {}
 
-        global_node_displays: Dict[Type[BaseNode], BaseNodeDisplay] = (
+        global_node_displays: NodeDisplays = (
             copy(self._parent_display_context.global_node_displays) if self._parent_display_context else {}
         )
 
-        port_displays: Dict[Port, PortDisplay] = {}
+        port_displays: PortDisplays = {}
 
         for node in self._workflow.get_nodes():
-            extracted_node_displays = self._extract_node_displays(node)
-
-            for extracted_node, extracted_node_display in extracted_node_displays.items():
-                if extracted_node not in node_displays:
-                    node_displays[extracted_node] = extracted_node_display
-
-                if extracted_node not in global_node_displays:
-                    global_node_displays[extracted_node] = extracted_node_display
-
-            self._enrich_global_node_output_displays(node, extracted_node_displays[node], global_node_output_displays)
-            self._enrich_node_port_displays(node, extracted_node_displays[node], port_displays)
+            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():
-            extracted_node_displays = self._extract_node_displays(node)
-
-            for extracted_node, extracted_node_display in extracted_node_displays.items():
-                if extracted_node not in node_displays:
-                    node_displays[extracted_node] = extracted_node_display
-
-                if extracted_node not in global_node_displays:
-                    global_node_displays[extracted_node] = extracted_node_display
-
-            self._enrich_global_node_output_displays(node, extracted_node_displays[node], global_node_output_displays)
-            self._enrich_node_port_displays(node, extracted_node_displays[node], port_displays)
+            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,
+            )
 
         workflow_input_displays: Dict[WorkflowInputReference, WorkflowInputsDisplayType] = {}
         # If we're dealing with a nested workflow, then it should have access to the inputs of its parents.
@@ -253,14 +244,14 @@ class BaseWorkflowDisplay(
             state_value_displays[state_value] = state_value_display
             global_state_value_displays[state_value] = state_value_display
 
-        entrypoint_displays: Dict[Type[BaseNode], EntrypointDisplayType] = {}
+        entrypoint_displays: EntrypointDisplays = {}
         for entrypoint in self._workflow.get_entrypoints():
             if entrypoint in entrypoint_displays:
                 continue
 
             entrypoint_display_overrides = self.entrypoint_displays.get(entrypoint)
             entrypoint_displays[entrypoint] = self._generate_entrypoint_display(
-                entrypoint, workflow_display, node_displays, overrides=entrypoint_display_overrides
+                entrypoint, workflow_meta_display, node_displays, overrides=entrypoint_display_overrides
             )
 
         edge_displays: Dict[Tuple[Port, Type[BaseNode]], EdgeDisplay] = {}
@@ -296,7 +287,7 @@ class BaseWorkflowDisplay(
             )
 
         return WorkflowDisplayContext(
-            workflow_display=workflow_display,
+            workflow_display=workflow_meta_display,
             workflow_input_displays=workflow_input_displays,
             global_workflow_input_displays=global_workflow_input_displays,
             state_value_displays=state_value_displays,
@@ -311,9 +302,24 @@ class BaseWorkflowDisplay(
             workflow_display_class=self.__class__,
         )
 
-    @abstractmethod
-    def _generate_workflow_meta_display(self) -> WorkflowMetaDisplayType:
-        pass
+    def _generate_workflow_meta_display(self) -> WorkflowMetaDisplay:
+        overrides = self.workflow_display
+        if overrides:
+            return WorkflowMetaDisplay(
+                entrypoint_node_id=overrides.entrypoint_node_id,
+                entrypoint_node_source_handle_id=overrides.entrypoint_node_source_handle_id,
+                entrypoint_node_display=overrides.entrypoint_node_display,
+                display_data=overrides.display_data,
+            )
+
+        entrypoint_node_id = uuid4_from_hash(f"{self.workflow_id}|entrypoint_node_id")
+        entrypoint_node_source_handle_id = uuid4_from_hash(f"{self.workflow_id}|entrypoint_node_source_handle_id")
+
+        return WorkflowMetaDisplay(
+            entrypoint_node_id=entrypoint_node_id,
+            entrypoint_node_source_handle_id=entrypoint_node_source_handle_id,
+            entrypoint_node_display=NodeDisplayData(),
+        )
 
     @abstractmethod
     def _generate_workflow_input_display(
@@ -327,15 +333,31 @@ class BaseWorkflowDisplay(
     ) -> StateValueDisplayType:
         pass
 
-    @abstractmethod
     def _generate_entrypoint_display(
         self,
         entrypoint: Type[BaseNode],
-        workflow_display: WorkflowMetaDisplayType,
+        workflow_display: WorkflowMetaDisplay,
         node_displays: Dict[Type[BaseNode], BaseNodeDisplay],
-        overrides: Optional[EntrypointDisplayOverridesType] = None,
-    ) -> EntrypointDisplayType:
-        pass
+        overrides: Optional[EntrypointDisplay] = None,
+    ) -> EntrypointDisplay:
+        entrypoint_node_id = workflow_display.entrypoint_node_id
+
+        edge_display_overrides = overrides.edge_display if overrides else None
+        entrypoint_id = (
+            edge_display_overrides.id
+            if edge_display_overrides
+            else uuid4_from_hash(f"{self.workflow_id}|id|{entrypoint_node_id}")
+        )
+
+        entrypoint_target = get_unadorned_node(entrypoint)
+        target_node_display = node_displays[entrypoint_target]
+        target_node_id = target_node_display.node_id
+
+        edge_display = edge_display_overrides or self._generate_edge_display_from_source(
+            entrypoint_node_id, target_node_id
+        )
+
+        return EntrypointDisplay(id=entrypoint_id, edge_display=edge_display)
 
     def _generate_workflow_output_display(self, output: BaseDescriptor) -> WorkflowOutputDisplay:
         output_id = uuid4_from_hash(f"{self.workflow_id}|id|{output.name}")
@@ -418,6 +440,26 @@ class BaseWorkflowDisplay(
         )
         return display_meta
 
+    def _enrich_node_displays(
+        self,
+        node: Type[BaseNode],
+        node_displays: NodeDisplays,
+        global_node_displays: NodeDisplays,
+        global_node_output_displays: NodeOutputDisplays,
+        port_displays: PortDisplays,
+    ) -> None:
+        extracted_node_displays = self._extract_node_displays(node)
+
+        for extracted_node, extracted_node_display in extracted_node_displays.items():
+            if extracted_node not in node_displays:
+                node_displays[extracted_node] = extracted_node_display
+
+            if extracted_node not in global_node_displays:
+                global_node_displays[extracted_node] = extracted_node_display
+
+        self._enrich_global_node_output_displays(node, extracted_node_displays[node], global_node_output_displays)
+        self._enrich_node_port_displays(node, extracted_node_displays[node], port_displays)
+
     def _extract_node_displays(self, node: Type[BaseNode]) -> Dict[Type[BaseNode], BaseNodeDisplay]:
         node_display = self._get_node_display(node)
         additional_node_displays: Dict[Type[BaseNode], BaseNodeDisplay] = {

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

@@ -7,10 +7,10 @@ from vellum.workflows.nodes.core.retry_node.node import RetryNode
 from vellum.workflows.nodes.core.templating_node.node import TemplatingNode
 from vellum.workflows.nodes.core.try_node.node import TryNode
 from vellum.workflows.workflows.base import BaseWorkflow
+from vellum_ee.workflows.display.editor.types import NodeDisplayData, NodeDisplayPosition
 from vellum_ee.workflows.display.nodes import BaseNodeDisplay
 from vellum_ee.workflows.display.nodes.vellum.retry_node import BaseRetryNodeDisplay
 from vellum_ee.workflows.display.nodes.vellum.try_node import BaseTryNodeDisplay
-from vellum_ee.workflows.display.vellum import NodeDisplayData, NodeDisplayPosition
 from vellum_ee.workflows.display.workflows import VellumWorkflowDisplay
 from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
 

vellum_ee/workflows/display/workflows/vellum_workflow_display.py

@@ -1,9 +1,8 @@
 import logging
 from uuid import UUID
-from typing import Dict, Optional, Type, cast
+from typing import Optional, cast
 
 from vellum.workflows.descriptors.base import BaseDescriptor
-from vellum.workflows.nodes.bases import BaseNode
 from vellum.workflows.nodes.displayable.bases.utils import primitive_to_vellum_value
 from vellum.workflows.nodes.displayable.final_output_node import FinalOutputNode
 from vellum.workflows.nodes.utils import get_unadorned_node, get_unadorned_port
@@ -12,20 +11,16 @@ from vellum.workflows.references.output import OutputReference
 from vellum.workflows.types.core import JsonArray, JsonObject
 from vellum.workflows.types.generics import WorkflowType
 from vellum.workflows.utils.uuids import uuid4_from_hash
+from vellum_ee.workflows.display.editor.types import NodeDisplayData
 from vellum_ee.workflows.display.nodes.base_node_display import BaseNodeDisplay
 from vellum_ee.workflows.display.nodes.base_node_vellum_display import BaseNodeVellumDisplay
 from vellum_ee.workflows.display.nodes.vellum.utils import create_node_input
 from vellum_ee.workflows.display.utils.vellum import infer_vellum_variable_type
 from vellum_ee.workflows.display.vellum import (
-    EntrypointVellumDisplay,
-    EntrypointVellumDisplayOverrides,
-    NodeDisplayData,
     StateValueVellumDisplay,
     StateValueVellumDisplayOverrides,
     WorkflowInputsVellumDisplay,
     WorkflowInputsVellumDisplayOverrides,
-    WorkflowMetaVellumDisplay,
-    WorkflowMetaVellumDisplayOverrides,
 )
 from vellum_ee.workflows.display.workflows.base_workflow_display import BaseWorkflowDisplay
 
@@ -35,14 +30,10 @@ logger = logging.getLogger(__name__)
 class VellumWorkflowDisplay(
     BaseWorkflowDisplay[
         WorkflowType,
-        WorkflowMetaVellumDisplay,
-        WorkflowMetaVellumDisplayOverrides,
         WorkflowInputsVellumDisplay,
         WorkflowInputsVellumDisplayOverrides,
         StateValueVellumDisplay,
         StateValueVellumDisplayOverrides,
-        EntrypointVellumDisplay,
-        EntrypointVellumDisplayOverrides,
     ]
 ):
     node_display_base_class = BaseNodeDisplay
@@ -290,25 +281,6 @@ class VellumWorkflowDisplay(
             "output_variables": output_variables,
         }
 
-    def _generate_workflow_meta_display(self) -> WorkflowMetaVellumDisplay:
-        overrides = self.workflow_display
-        if overrides:
-            return WorkflowMetaVellumDisplay(
-                entrypoint_node_id=overrides.entrypoint_node_id,
-                entrypoint_node_source_handle_id=overrides.entrypoint_node_source_handle_id,
-                entrypoint_node_display=overrides.entrypoint_node_display,
-                display_data=overrides.display_data,
-            )
-
-        entrypoint_node_id = uuid4_from_hash(f"{self.workflow_id}|entrypoint_node_id")
-        entrypoint_node_source_handle_id = uuid4_from_hash(f"{self.workflow_id}|entrypoint_node_source_handle_id")
-
-        return WorkflowMetaVellumDisplay(
-            entrypoint_node_id=entrypoint_node_id,
-            entrypoint_node_source_handle_id=entrypoint_node_source_handle_id,
-            entrypoint_node_display=NodeDisplayData(),
-        )
-
     def _generate_workflow_input_display(
         self, workflow_input: WorkflowInputReference, overrides: Optional[WorkflowInputsVellumDisplayOverrides] = None
     ) -> WorkflowInputsVellumDisplay:
@@ -342,29 +314,3 @@ class VellumWorkflowDisplay(
             state_value_id = uuid4_from_hash(f"{self.workflow_id}|state_values|id|{state_value.name}")
 
         return StateValueVellumDisplay(id=state_value_id, name=name, required=required, color=color)
-
-    def _generate_entrypoint_display(
-        self,
-        entrypoint: Type[BaseNode],
-        workflow_display: WorkflowMetaVellumDisplay,
-        node_displays: Dict[Type[BaseNode], BaseNodeDisplay],
-        overrides: Optional[EntrypointVellumDisplayOverrides] = None,
-    ) -> EntrypointVellumDisplay:
-        entrypoint_node_id = workflow_display.entrypoint_node_id
-
-        edge_display_overrides = overrides.edge_display if overrides else None
-        entrypoint_id = (
-            edge_display_overrides.id
-            if edge_display_overrides
-            else uuid4_from_hash(f"{self.workflow_id}|id|{entrypoint_node_id}")
-        )
-
-        entrypoint_target = get_unadorned_node(entrypoint)
-        target_node_display = node_displays[entrypoint_target]
-        target_node_id = target_node_display.node_id
-
-        edge_display = edge_display_overrides or self._generate_edge_display_from_source(
-            entrypoint_node_id, target_node_id
-        )
-
-        return EntrypointVellumDisplay(id=entrypoint_id, edge_display=edge_display)