vellum-ai 0.12.17__py3-none-any.whl → 0.13.1__py3-none-any.whl

vellum_ee/workflows/display/tests/test_vellum_workflow_display.py

@@ -30,11 +30,8 @@ def test_vellum_workflow_display__serialize_empty_workflow():
             "nodes": [
                 {
                     "data": {"label": "Entrypoint Node", "source_handle_id": "508b8b82-3517-4672-a155-18c9c7b9c545"},
-                    "definition": {
-                        "bases": [],
-                        "module": ["vellum", "workflows", "nodes", "bases", "base"],
-                        "name": "BaseNode",
-                    },
+                    "base": None,
+                    "definition": None,
                     "display_data": {"position": {"x": 0.0, "y": 0.0}},
                     "id": "9eef0c18-f322-4d56-aa89-f088d3e53f6a",
                     "inputs": [],

vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/conftest.py

@@ -1,16 +1,29 @@
 import pytest
 from uuid import uuid4
+from typing import Dict, Tuple, Type
 
+from vellum.workflows.nodes.bases.base import BaseNode
+from vellum.workflows.references.output import OutputReference
+from vellum.workflows.references.workflow_input import WorkflowInputReference
+from vellum.workflows.types.core import JsonObject
+from vellum.workflows.types.generics import NodeType
+from vellum_ee.workflows.display.base import WorkflowInputsDisplayType
 from vellum_ee.workflows.display.nodes.get_node_display_class import get_node_display_class
+from vellum_ee.workflows.display.nodes.types import NodeOutputDisplay
 from vellum_ee.workflows.display.nodes.vellum.base_node import BaseNodeDisplay
-from vellum_ee.workflows.display.types import WorkflowDisplayContext
+from vellum_ee.workflows.display.types import NodeDisplayType, WorkflowDisplayContext
 from vellum_ee.workflows.display.vellum import NodeDisplayData, WorkflowMetaVellumDisplay
 from vellum_ee.workflows.display.workflows.vellum_workflow_display import VellumWorkflowDisplay
 
 
 @pytest.fixture()
 def serialize_node():
-    def _serialize_node(node_class) -> dict:
+    def _serialize_node(
+        node_class: Type[NodeType],
+        global_workflow_input_displays: Dict[WorkflowInputReference, WorkflowInputsDisplayType] = {},
+        global_node_displays: Dict[Type[BaseNode], NodeDisplayType] = {},
+        global_node_output_displays: Dict[OutputReference, Tuple[Type[BaseNode], NodeOutputDisplay]] = {},
+    ) -> JsonObject:
         node_display_class = get_node_display_class(BaseNodeDisplay, node_class)
         node_display = node_display_class()
 
@@ -22,6 +35,9 @@ def serialize_node():
                 entrypoint_node_display=NodeDisplayData(),
             ),
             node_displays={node_class: node_display},
+            global_workflow_input_displays=global_workflow_input_displays,
+            global_node_displays=global_node_displays,
+            global_node_output_displays=global_node_output_displays,
         )
         return node_display.serialize(context)
 

vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_adornments_serialization.py

@@ -0,0 +1,67 @@
+import pytest
+
+from deepdiff import DeepDiff
+
+from vellum.workflows.inputs.base import BaseInputs
+from vellum.workflows.nodes.bases.base import BaseNode
+from vellum.workflows.nodes.core.map_node.node import MapNode
+from vellum.workflows.outputs.base import BaseOutputs
+from vellum.workflows.state.base import BaseState
+
+
+class Inputs(BaseInputs):
+    foo: str
+
+
+class State(BaseState):
+    bar: str
+
+
+@MapNode.wrap(items=[1, 2, 3])
+class MapGenericNode(BaseNode):
+    item = MapNode.SubworkflowInputs.item
+    foo = Inputs.foo
+    bar = State.bar
+
+    class Outputs(BaseOutputs):
+        value: str
+
+    def run(self) -> Outputs:
+        return self.Outputs(value=f"{self.foo} {self.bar} {self.item}")
+
+
+@pytest.mark.skip(reason="not implemented")
+def test_serialize_node__try(serialize_node):
+    serialized_node = serialize_node(MapGenericNode)
+    assert not DeepDiff(
+        {
+            "id": "c2ed23f7-f6cb-4a56-a91c-2e5f9d8fda7f",
+            "label": "BasicGenericNode",
+            "type": "GENERIC",
+            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "base": {"name": "BaseNode", "module": ["vellum", "workflows", "nodes", "bases", "base"]},
+            "definition": {
+                "name": "BasicGenericNode",
+                "module": [
+                    "vellum_ee",
+                    "workflows",
+                    "display",
+                    "tests",
+                    "workflow_serialization",
+                    "generic_nodes",
+                    "test_adornments_serialization",
+                ],
+            },
+            "trigger": {"id": "9d3a1b3d-4a38-4f2e-bbf1-dd8be152bce8", "merge_behavior": "AWAIT_ANY"},
+            "ports": [
+                {
+                    "id": "4fbf0fff-a42e-4410-852a-238b5059198e",
+                    "type": "DEFAULT",
+                }
+            ],
+            "adornments": None,
+            "attributes": [],
+        },
+        serialized_node,
+        ignore_order=True,
+    )

vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_attributes_serialization.py

@@ -0,0 +1,66 @@
+import pytest
+
+from deepdiff import DeepDiff
+
+from vellum.workflows.inputs.base import BaseInputs
+from vellum.workflows.nodes.bases.base import BaseNode
+
+
+class Inputs(BaseInputs):
+    input: str
+
+
+class BasicGenericNode(BaseNode):
+    attr: str = "hello"
+
+    class Outputs(BaseNode.Outputs):
+        output = Inputs.input
+
+
+@pytest.mark.skip(reason="not implemented")
+def test_serialize_node__try(serialize_node):
+    serialized_node = serialize_node(BasicGenericNode)
+    assert not DeepDiff(
+        {
+            "id": "c2ed23f7-f6cb-4a56-a91c-2e5f9d8fda7f",
+            "label": "BasicGenericNode",
+            "type": "GENERIC",
+            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "base": {"name": "BaseNode", "module": ["vellum", "workflows", "nodes", "bases", "base"]},
+            "definition": {
+                "name": "BasicGenericNode",
+                "module": [
+                    "vellum_ee",
+                    "workflows",
+                    "display",
+                    "tests",
+                    "workflow_serialization",
+                    "generic_nodes",
+                    "test_attributes_serialization",
+                ],
+            },
+            "trigger": {"id": "9d3a1b3d-4a38-4f2e-bbf1-dd8be152bce8", "merge_behavior": "AWAIT_ANY"},
+            "ports": [
+                {
+                    "id": "4fbf0fff-a42e-4410-852a-238b5059198e",
+                    "type": "DEFAULT",
+                }
+            ],
+            "adornments": None,
+            "attributes": [
+                {
+                    "id": "e3c8b6e4-4b7d-4e3c-8a5c-6e5f4e3c8b6e",
+                    "name": "attr",
+                    "value": {
+                        "type": "CONSTANT_VALUE",
+                        "value": {
+                            "type": "STRING",
+                            "value": "hello",
+                        },
+                    },
+                }
+            ],
+        },
+        serialized_node,
+        ignore_order=True,
+    )