vellum-ai 0.14.40__py3-none-any.whl → 0.14.42__py3-none-any.whl

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

@@ -154,6 +154,7 @@ def test_serialize_workflow(vellum_client):
             "module": ["tests", "workflows", "basic_subworkflow_deployment", "workflow"],
             "name": "ExampleSubworkflowDeploymentNode",
         },
+        "ports": [{"id": "ab0db8a9-7b53-4d88-8667-273b31303273", "name": "default", "type": "DEFAULT"}],
     }
 
     assert not DeepDiff(

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

@@ -113,6 +113,7 @@ def test_serialize_workflow():
                 "name": "ExampleTemplatingNode",
                 "module": ["tests", "workflows", "basic_templating_node", "workflow_with_json_input"],
             },
+            "ports": [{"id": "39317827-df43-4f5a-bfbc-20bffc839748", "name": "default", "type": "DEFAULT"}],
         },
         templating_node,
     )

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

@@ -68,7 +68,7 @@ def test_serialize_workflow():
             "label": "Basic Final Output Node",
             "name": "basic-final-output-node",
             "target_handle_id": "0173d3c6-11d1-44b7-b070-ca9ff5119046",
-            "output_id": "97349956-d228-4b51-a64b-1331f788373f",
+            "output_id": "aa63e3f6-fde3-4d19-84ef-29982d44d709",
             "output_type": "STRING",
             "node_input_id": "5322567a-f40c-400a-96b3-c3b054db543e",
         },
@@ -98,7 +98,7 @@ def test_serialize_workflow():
         },
         "outputs": [
             {
-                "id": "97349956-d228-4b51-a64b-1331f788373f",
+                "id": "aa63e3f6-fde3-4d19-84ef-29982d44d709",
                 "name": "value",
                 "type": "STRING",
                 "value": {"type": "WORKFLOW_INPUT", "input_variable_id": "e39a7b63-de15-490a-ae9b-8112c767aea0"},

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

@@ -92,7 +92,7 @@ def test_serialize_workflow__missing_final_output_node():
                     "label": "First Final Output Node",
                     "name": "first-final-output-node",
                     "target_handle_id": "a0c2eb7a-398e-4f28-b63d-f3bae9b563ee",
-                    "output_id": "5517e50d-7f40-4f7c-acb2-e329d79a25bf",
+                    "output_id": "0cd02933-c5b9-47c9-aede-e97c5870e8aa",
                     "output_type": "STRING",
                     "node_input_id": "16363762-c14a-4162-8fab-525079d3cffe",
                 },
@@ -122,7 +122,7 @@ def test_serialize_workflow__missing_final_output_node():
                 },
                 "outputs": [
                     {
-                        "id": "5517e50d-7f40-4f7c-acb2-e329d79a25bf",
+                        "id": "0cd02933-c5b9-47c9-aede-e97c5870e8aa",
                         "name": "value",
                         "type": "STRING",
                         "value": {

vellum_ee/workflows/display/workflows/base_workflow_display.py

@@ -31,7 +31,6 @@ from vellum_ee.workflows.display.base import (
 )
 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
 from vellum_ee.workflows.display.nodes.utils import raise_if_descriptor
@@ -241,14 +240,9 @@ class BaseWorkflowDisplay(Generic[WorkflowType]):
                 )
 
                 if source_node_display:
-                    if isinstance(source_node_display, BaseNodeVellumDisplay):
-                        source_handle_id = source_node_display.get_source_handle_id(
-                            port_displays=self.display_context.port_displays
-                        )
-                    else:
-                        source_handle_id = source_node_display.get_node_port_display(
-                            source_node_display._node.Ports.default
-                        ).id
+                    source_handle_id = source_node_display.get_source_handle_id(
+                        port_displays=self.display_context.port_displays
+                    )
 
                     synthetic_output_edges.append(
                         {
@@ -617,9 +611,7 @@ class BaseWorkflowDisplay(Generic[WorkflowType]):
         node_event_displays = {}
         for node_id in node_displays:
             node, current_node_display = node_displays[node_id]
-            input_display = {}
-            if isinstance(current_node_display, BaseNodeVellumDisplay):
-                input_display = current_node_display.node_input_ids_by_name
+            input_display = current_node_display.node_input_ids_by_name
             output_display = {
                 output.name: current_node_display.output_display[output].id
                 for output in current_node_display.output_display

vellum/client/types/node_event_display_context.py

@@ -1,30 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-from ..core.pydantic_utilities import update_forward_refs
-
-
-class NodeEventDisplayContext(UniversalBaseModel):
-    input_display: typing.Dict[str, str]
-    output_display: typing.Dict[str, str]
-    port_display: typing.Dict[str, str]
-    subworkflow_display: typing.Optional["WorkflowEventDisplayContext"] = None
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
-
-
-from .workflow_event_display_context import WorkflowEventDisplayContext  # noqa: E402
-
-update_forward_refs(WorkflowEventDisplayContext, NodeEventDisplayContext=NodeEventDisplayContext)
-update_forward_refs(NodeEventDisplayContext)

vellum/client/types/workflow_event_display_context.py

@@ -1,28 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-from ..core.pydantic_utilities import update_forward_refs
-
-
-class WorkflowEventDisplayContext(UniversalBaseModel):
-    node_displays: typing.Dict[str, "NodeEventDisplayContext"]
-    workflow_inputs: typing.Dict[str, str]
-    workflow_outputs: typing.Dict[str, str]
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
-
-
-from .node_event_display_context import NodeEventDisplayContext  # noqa: E402
-
-update_forward_refs(WorkflowEventDisplayContext)

vellum/types/workflow_event_display_context.py

@@ -1,3 +0,0 @@
-# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
-
-from vellum.client.types.workflow_event_display_context import *

vellum_ee/workflows/display/nodes/base_node_vellum_display.py

@@ -1,40 +0,0 @@
-from uuid import UUID
-from typing import ClassVar, Dict, Optional
-
-from vellum.workflows.nodes.utils import get_unadorned_node
-from vellum.workflows.ports import Port
-from vellum.workflows.types.generics import NodeType
-from vellum_ee.workflows.display.editor.types import NodeDisplayComment, NodeDisplayData
-from vellum_ee.workflows.display.nodes.base_node_display import BaseNodeDisplay
-from vellum_ee.workflows.display.nodes.types import PortDisplay
-
-
-class BaseNodeVellumDisplay(BaseNodeDisplay[NodeType]):
-    # Used to explicitly set display data for a node
-    display_data: ClassVar[Optional[NodeDisplayData]] = None
-
-    def get_display_data(self) -> NodeDisplayData:
-        explicit_value = self._get_explicit_node_display_attr("display_data", NodeDisplayData)
-        docstring = self._node.__doc__
-
-        if explicit_value and explicit_value.comment and docstring:
-            comment = (
-                NodeDisplayComment(value=docstring, expanded=explicit_value.comment.expanded)
-                if explicit_value.comment.expanded
-                else NodeDisplayComment(value=docstring)
-            )
-            return NodeDisplayData(
-                position=explicit_value.position,
-                width=explicit_value.width,
-                height=explicit_value.height,
-                comment=comment,
-            )
-
-        return explicit_value if explicit_value else NodeDisplayData()
-
-    def get_source_handle_id(self, port_displays: Dict[Port, PortDisplay]) -> UUID:
-        unadorned_node = get_unadorned_node(self._node)
-        default_port = unadorned_node.Ports.default
-
-        default_port_display = port_displays[default_port]
-        return default_port_display.id