vellum-ai 0.14.89__py3-none-any.whl → 1.0.0__py3-none-any.whl

vellum_ee/workflows/display/vellum.py

@@ -1,135 +1,9 @@
-from dataclasses import dataclass
-from uuid import UUID
-from typing import List, Literal, Optional
+from typing import List, Literal
 
-from vellum.core import UniversalBaseModel
-from vellum_ee.workflows.display.base import (
-    EdgeDisplayOverrides,
-    EntrypointDisplayOverrides,
-    StateValueDisplayOverrides,
-    WorkflowInputsDisplay,
-    WorkflowMetaDisplayOverrides,
-    WorkflowOutputDisplayOverrides,
-)
-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.client.core import UniversalBaseModel
 from vellum_ee.workflows.display.utils.vellum import NodeInputValuePointerRule
 
 
-@dataclass
-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
-
-
-@dataclass
-class WorkflowInputsVellumDisplayOverrides(WorkflowInputsDisplay):
-    """
-    DEPRECATED: Use WorkflowInputsDisplay instead. Will be removed in 0.15.0
-    """
-
-    required: Optional[bool] = None
-
-
-@dataclass
-class WorkflowInputsVellumDisplay(WorkflowInputsVellumDisplayOverrides):
-    pass
-
-
-@dataclass
-class StateValueVellumDisplayOverrides(StateValueDisplayOverrides):
-    """
-    DEPRECATED: Use StateValueDisplay instead. Will be removed in 0.15.0
-    """
-
-    required: Optional[bool] = None
-
-
-@dataclass
-class StateValueVellumDisplay(StateValueVellumDisplayOverrides):
-    """
-    DEPRECATED: Use StateValueDisplay instead. Will be removed in 0.15.0
-    """
-
-    pass
-
-
-@dataclass
-class EdgeVellumDisplayOverrides(EdgeDisplayOverrides):
-    """
-    DEPRECATED: Use EdgeDisplay instead. Will be removed in 0.15.0
-    """
-
-    pass
-
-
-@dataclass
-class EdgeVellumDisplay(EdgeVellumDisplayOverrides):
-    """
-    DEPRECATED: Use EdgeDisplay instead. Will be removed in 0.15.0
-    """
-
-    source_node_id: UUID
-    source_handle_id: UUID
-    target_node_id: UUID
-    target_handle_id: UUID
-    type: Literal["DEFAULT"] = "DEFAULT"
-
-
-@dataclass
-class EntrypointVellumDisplayOverrides(EntrypointDisplayOverrides):
-    """
-    DEPRECATED: Use EntrypointDisplay instead. Will be removed in 0.15.0
-    """
-
-    pass
-
-
-@dataclass
-class EntrypointVellumDisplay(EntrypointVellumDisplayOverrides):
-    """
-    DEPRECATED: Use EntrypointDisplay instead. Will be removed in 0.15.0
-    """
-
-    pass
-
-
-@dataclass
-class WorkflowOutputVellumDisplayOverrides(WorkflowOutputDisplayOverrides):
-    """
-    DEPRECATED: Use WorkflowOutputDisplay instead. Will be removed in 0.15.0
-    """
-
-    label: Optional[str] = None
-    node_id: Optional[UUID] = None
-    display_data: Optional[NodeDisplayData] = None
-    target_handle_id: Optional[UUID] = None
-
-
-@dataclass
-class WorkflowOutputVellumDisplay(WorkflowOutputVellumDisplayOverrides):
-    """
-    DEPRECATED: Use WorkflowOutputDisplay instead. Will be removed in 0.15.0
-    """
-
-    pass
-
-
 class NodeInputValuePointer(UniversalBaseModel):
     rules: List[NodeInputValuePointerRule]
     combinator: Literal["OR"] = "OR"

vellum_ee/workflows/display/workflows/__init__.py

@@ -2,4 +2,3 @@
 
 # These must be imported to register the classes in the registry
 from .base_workflow_display import BaseWorkflowDisplay
-from .vellum_workflow_display import VellumWorkflowDisplay

vellum_ee/workflows/display/workflows/base_workflow_display.py

@@ -9,7 +9,7 @@ from uuid import UUID
 from typing import Any, Dict, ForwardRef, Generic, Iterator, List, Optional, Tuple, Type, TypeVar, Union, cast, get_args
 
 from vellum.client import Vellum as VellumClient
-from vellum.core.pydantic_utilities import UniversalBaseModel
+from vellum.client.core.pydantic_utilities import UniversalBaseModel
 from vellum.workflows import BaseWorkflow
 from vellum.workflows.constants import undefined
 from vellum.workflows.descriptors.base import BaseDescriptor

vellum_ee/workflows/tests/local_workflow/display/nodes/final_output.py

@@ -1,8 +1,8 @@
 from uuid import UUID
 
+from vellum_ee.workflows.display.editor import NodeDisplayData, NodeDisplayPosition
 from vellum_ee.workflows.display.nodes import BaseFinalOutputNodeDisplay
 from vellum_ee.workflows.display.nodes.types import NodeOutputDisplay
-from vellum_ee.workflows.display.vellum import NodeDisplayData, NodeDisplayPosition
 
 from ...nodes.final_output import FinalOutput
 

vellum_ee/workflows/tests/local_workflow/display/nodes/templating_node.py

@@ -1,8 +1,8 @@
 from uuid import UUID
 
+from vellum_ee.workflows.display.editor import NodeDisplayData, NodeDisplayPosition
 from vellum_ee.workflows.display.nodes import BaseTemplatingNodeDisplay
 from vellum_ee.workflows.display.nodes.types import NodeOutputDisplay, PortDisplayOverrides
-from vellum_ee.workflows.display.vellum import NodeDisplayData, NodeDisplayPosition
 
 from ...nodes.templating_node import TemplatingNode
 

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

@@ -1,16 +1,15 @@
 from uuid import UUID
 
-from vellum_ee.workflows.display.base import EdgeDisplay
-from vellum_ee.workflows.display.vellum import (
-    EntrypointVellumDisplayOverrides,
-    NodeDisplayData,
-    NodeDisplayPosition,
+from vellum_ee.workflows.display.base import (
+    EdgeDisplay,
+    EntrypointDisplay,
     WorkflowDisplayData,
     WorkflowDisplayDataViewport,
-    WorkflowInputsVellumDisplayOverrides,
-    WorkflowMetaVellumDisplayOverrides,
-    WorkflowOutputVellumDisplayOverrides,
+    WorkflowInputsDisplay,
+    WorkflowMetaDisplay,
+    WorkflowOutputDisplay,
 )
+from vellum_ee.workflows.display.editor import NodeDisplayData, NodeDisplayPosition
 from vellum_ee.workflows.display.workflows import BaseWorkflowDisplay
 
 from ..inputs import Inputs
@@ -20,7 +19,7 @@ from ..workflow import Workflow
 
 
 class WorkflowDisplay(BaseWorkflowDisplay[Workflow]):
-    workflow_display = WorkflowMetaVellumDisplayOverrides(
+    workflow_display = WorkflowMetaDisplay(
         entrypoint_node_id=UUID("0bf86989-13f2-438c-ab9c-d172e5771d31"),
         entrypoint_node_source_handle_id=UUID("23448f09-81ad-4378-abbd-1cccff350627"),
         entrypoint_node_display=NodeDisplayData(position=NodeDisplayPosition(x=1545, y=330), width=124, height=48),
@@ -29,12 +28,10 @@ class WorkflowDisplay(BaseWorkflowDisplay[Workflow]):
         ),
     )
     inputs_display = {
-        Inputs.input_value: WorkflowInputsVellumDisplayOverrides(
-            id=UUID("2268a996-bd17-4832-b3ff-f5662d54b306"), name="input-value", required=True
-        )
+        Inputs.input_value: WorkflowInputsDisplay(id=UUID("2268a996-bd17-4832-b3ff-f5662d54b306"), name="input-value")
     }
     entrypoint_displays = {
-        TemplatingNode: EntrypointVellumDisplayOverrides(
+        TemplatingNode: EntrypointDisplay(
             id=UUID("0bf86989-13f2-438c-ab9c-d172e5771d31"),
             edge_display=EdgeDisplay(id=UUID("38532a0e-9432-4ed2-8a34-48a29fd6984d")),
         )
@@ -43,7 +40,7 @@ class WorkflowDisplay(BaseWorkflowDisplay[Workflow]):
         (TemplatingNode.Ports.default, FinalOutput): EdgeDisplay(id=UUID("417c56a4-cdc1-4f9d-a10c-b535163f51e8"))
     }
     output_displays = {
-        Workflow.Outputs.final_output: WorkflowOutputVellumDisplayOverrides(
+        Workflow.Outputs.final_output: WorkflowOutputDisplay(
             id=UUID("5469b810-6ea6-4362-9e79-e360d44a1405"),
             name="final-output",
         )

vellum/workflows/nodes/experimental/tool_calling_node/__init__.py

@@ -1,3 +0,0 @@
-from vellum.workflows.nodes.displayable.tool_calling_node import ToolCallingNode
-
-__all__ = ["ToolCallingNode"]

vellum/workflows/nodes/experimental/tool_calling_node/node.py

@@ -1,3 +0,0 @@
-from vellum.workflows.nodes.displayable.tool_calling_node import ToolCallingNode
-
-__all__ = ["ToolCallingNode"]

vellum_ee/workflows/display/workflows/vellum_workflow_display.py

@@ -1,9 +0,0 @@
-from typing import Generic
-
-from vellum.workflows.types.generics import WorkflowType
-from vellum_ee.workflows.display.workflows.base_workflow_display import BaseWorkflowDisplay
-
-
-# DEPRECATED: Use BaseWorkflowDisplay instead - This file will be removed in 0.15.0
-class VellumWorkflowDisplay(BaseWorkflowDisplay[WorkflowType], Generic[WorkflowType]):
-    pass