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

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_conditional_node_serialization.py

@@ -285,7 +285,7 @@ def test_serialize_workflow():
                     {
                         "id": "de7b0b4e-7803-4d36-a275-2e7e3f60342b",
                         "type": "IF",
-                        "source_handle_id": "561b4e3a-8db3-448a-8933-1115937082ff",
+                        "source_handle_id": "3a45b81f-95e4-4cbd-8997-bfdbe30251e8",
                         "data": {
                             "id": "2ccd0730-26d1-4fb4-baa9-1a2a182dd9a0",
                             "rules": [
@@ -309,7 +309,7 @@ def test_serialize_workflow():
                     {
                         "id": "5e783d17-6808-441a-ac6c-33a4e184f4e0",
                         "type": "ELIF",
-                        "source_handle_id": "0644f22e-2680-441a-9554-eedf1d3d22a9",
+                        "source_handle_id": "7202f702-1ebc-4067-ab1e-ec67e49158ee",
                         "data": {
                             "id": "cc3f0d92-b603-42cc-b2e9-83e3b23b3bcb",
                             "rules": [
@@ -333,7 +333,7 @@ def test_serialize_workflow():
                     {
                         "id": "6bd2f643-9cf5-4e7f-9113-f90e5c8057be",
                         "type": "ELIF",
-                        "source_handle_id": "2a48d274-ecfc-4f40-95ac-bc697663f10c",
+                        "source_handle_id": "cf45705d-1a47-43a6-9d24-a7fdf78baae0",
                         "data": {
                             "id": "a5a0f391-7052-452f-9fe1-a5781a491591",
                             "rules": [
@@ -357,7 +357,7 @@ def test_serialize_workflow():
                     {
                         "id": "0a058485-18a4-4e20-8a30-6da8196ac46f",
                         "type": "ELIF",
-                        "source_handle_id": "90fc9dc8-0a74-4a98-b6ac-55ffce4a2881",
+                        "source_handle_id": "f04610dd-61cf-41b0-b337-2235e101cdb0",
                         "data": {
                             "id": "efe7a851-2a67-4189-99ec-bc193242b270",
                             "rules": [
@@ -409,7 +409,7 @@ def test_serialize_workflow():
                     {
                         "id": "c2fa8a44-923b-462a-b0d2-fa800a152e52",
                         "type": "ELSE",
-                        "source_handle_id": "493024f4-8010-4e1a-abae-b6adbc6fb208",
+                        "source_handle_id": "f9dde637-ea90-465f-a871-caf8380ae377",
                         "data": None,
                     },
                 ],
@@ -913,7 +913,7 @@ def test_conditional_node_serialize_all_operators_with_lhs_and_rhs(descriptor, o
                     {
                         "id": "a4c32611-fd58-4b98-9d08-313cfd1c214e",
                         "type": "IF",
-                        "source_handle_id": "8124a6cf-4a34-4149-adc0-68696c11bd4e",
+                        "source_handle_id": "2ff87aa6-37cf-43dd-af9d-13b9198ab70a",
                         "data": {
                             "id": "650e7105-3e76-43ca-858f-b290970b438b",
                             "rules": [
@@ -937,7 +937,7 @@ def test_conditional_node_serialize_all_operators_with_lhs_and_rhs(descriptor, o
                     {
                         "id": "342e5497-ea2b-4e5c-99cf-e6492f133a3c",
                         "type": "ELSE",
-                        "source_handle_id": "4df924c0-7bed-4f4a-9db4-2bfe51841755",
+                        "source_handle_id": "9a7e8c2e-0228-4321-8f74-61cb5778f3df",
                         "data": None,
                     },
                 ],
@@ -1024,7 +1024,7 @@ def test_conditional_node_serialize_all_operators_with_expression(descriptor, op
                     {
                         "id": "a4c32611-fd58-4b98-9d08-313cfd1c214e",
                         "type": "IF",
-                        "source_handle_id": "8124a6cf-4a34-4149-adc0-68696c11bd4e",
+                        "source_handle_id": "2ff87aa6-37cf-43dd-af9d-13b9198ab70a",
                         "data": {
                             "id": "650e7105-3e76-43ca-858f-b290970b438b",
                             "rules": [
@@ -1048,7 +1048,7 @@ def test_conditional_node_serialize_all_operators_with_expression(descriptor, op
                     {
                         "id": "342e5497-ea2b-4e5c-99cf-e6492f133a3c",
                         "type": "ELSE",
-                        "source_handle_id": "4df924c0-7bed-4f4a-9db4-2bfe51841755",
+                        "source_handle_id": "9a7e8c2e-0228-4321-8f74-61cb5778f3df",
                         "data": None,
                     },
                 ],
@@ -1147,7 +1147,7 @@ def test_conditional_node_serialize_all_operators_with_value_and_start_and_end(d
                     {
                         "id": "a4c32611-fd58-4b98-9d08-313cfd1c214e",
                         "type": "IF",
-                        "source_handle_id": "8124a6cf-4a34-4149-adc0-68696c11bd4e",
+                        "source_handle_id": "2ff87aa6-37cf-43dd-af9d-13b9198ab70a",
                         "data": {
                             "id": "650e7105-3e76-43ca-858f-b290970b438b",
                             "rules": [
@@ -1171,7 +1171,7 @@ def test_conditional_node_serialize_all_operators_with_value_and_start_and_end(d
                     {
                         "id": "342e5497-ea2b-4e5c-99cf-e6492f133a3c",
                         "type": "ELSE",
-                        "source_handle_id": "4df924c0-7bed-4f4a-9db4-2bfe51841755",
+                        "source_handle_id": "9a7e8c2e-0228-4321-8f74-61cb5778f3df",
                         "data": None,
                     },
                 ],

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