vellum-ai 0.12.8__py3-none-any.whl → 0.12.11__py3-none-any.whl

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

@@ -0,0 +1,207 @@
+from deepdiff import DeepDiff
+
+from vellum_ee.workflows.display.workflows import VellumWorkflowDisplay
+from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
+
+from tests.workflows.basic_templating_node.workflow_with_json_input import BasicTemplatingNodeWorkflowWithJson
+
+
+def test_serialize_workflow():
+    # GIVEN a Workflow that uses a vellum templating node",
+
+    # WHEN we serialize it
+    workflow_display = get_workflow_display(
+        base_display_class=VellumWorkflowDisplay, workflow_class=BasicTemplatingNodeWorkflowWithJson
+    )
+
+    serialized_workflow: dict = workflow_display.serialize()
+
+    # THEN we should get a serialized representation of the Workflow
+    assert serialized_workflow.keys() == {
+        "workflow_raw_data",
+        "input_variables",
+        "output_variables",
+    }
+
+    # AND its input variables should be what we expect
+    input_variables = serialized_workflow["input_variables"]
+    assert len(input_variables) == 1
+    assert input_variables == [
+        {
+            "id": "f4435f88-703f-40e5-9197-d39b0e43ab72",
+            "key": "info",
+            "type": "JSON",
+            "default": None,
+            "required": True,
+            "extensions": {"color": None},
+        }
+    ]
+
+    # AND its output variables should be what we expect
+    output_variables = serialized_workflow["output_variables"]
+    assert len(output_variables) == 1
+    assert not DeepDiff(
+        [{"id": "62ec9b08-6437-4f8d-bc20-983d317bc348", "key": "result", "type": "JSON"}],
+        output_variables,
+        ignore_order=True,
+    )
+
+    # AND its raw data should be what we expect
+    workflow_raw_data = serialized_workflow["workflow_raw_data"]
+    assert workflow_raw_data.keys() == {"edges", "nodes", "display_data", "definition"}
+    assert len(workflow_raw_data["edges"]) == 2
+    assert len(workflow_raw_data["nodes"]) == 3
+
+    # AND each node should be serialized correctly
+    entrypoint_node = workflow_raw_data["nodes"][0]
+    assert entrypoint_node == {
+        "id": "09579c0a-acbf-4e04-8724-2230a8aa6534",
+        "type": "ENTRYPOINT",
+        "inputs": [],
+        "data": {"label": "Entrypoint Node", "source_handle_id": "34069190-0942-4e0c-8700-b33b9dea4ea0"},
+        "display_data": {"position": {"x": 0.0, "y": 0.0}},
+        "definition": {"name": "BaseNode", "module": ["vellum", "workflows", "nodes", "bases", "base"], "bases": []},
+    }
+
+    templating_node = workflow_raw_data["nodes"][1]
+    assert not DeepDiff(
+        {
+            "id": "51cbe21d-0232-4362-bc54-5bc283297aa6",
+            "type": "TEMPLATING",
+            "inputs": [
+                {
+                    "id": "7c775379-d589-4d79-b876-dcd224d72966",
+                    "key": "template",
+                    "value": {
+                        "rules": [
+                            {
+                                "type": "CONSTANT_VALUE",
+                                "data": {"type": "STRING", "value": "The meaning of {{ info }} is not known"},
+                            }
+                        ],
+                        "combinator": "OR",
+                    },
+                },
+                {
+                    "id": "bb712499-4265-4d71-bc14-0c3d8ca6a7de",
+                    "key": "info",
+                    "value": {
+                        "rules": [
+                            {
+                                "type": "INPUT_VARIABLE",
+                                "data": {"input_variable_id": "f4435f88-703f-40e5-9197-d39b0e43ab72"},
+                            }
+                        ],
+                        "combinator": "OR",
+                    },
+                },
+            ],
+            "data": {
+                "label": "Example Templating Node",
+                "output_id": "6834cae4-8173-4fa6-88f7-bc09d335bdd1",
+                "error_output_id": None,
+                "source_handle_id": "39317827-df43-4f5a-bfbc-20bffc839748",
+                "target_handle_id": "58427684-3848-498a-8299-c6b0fc70265d",
+                "template_node_input_id": "7c775379-d589-4d79-b876-dcd224d72966",
+                "output_type": "JSON",
+            },
+            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "definition": {
+                "name": "ExampleTemplatingNode",
+                "module": ["tests", "workflows", "basic_templating_node", "workflow_with_json_input"],
+                "bases": [
+                    {
+                        "name": "TemplatingNode",
+                        "module": ["vellum", "workflows", "nodes", "core", "templating_node", "node"],
+                    }
+                ],
+            },
+        },
+        templating_node,
+    )
+
+    final_output_node = workflow_raw_data["nodes"][-1]
+    assert not DeepDiff(
+        {
+            "id": "9f75228b-1d5b-4c30-a581-6087e6a1b738",
+            "type": "TERMINAL",
+            "data": {
+                "label": "Final Output",
+                "name": "result",
+                "target_handle_id": "16ba108e-61a8-4338-8a5b-4f1278d7fd7b",
+                "output_id": "62ec9b08-6437-4f8d-bc20-983d317bc348",
+                "output_type": "JSON",
+                "node_input_id": "310072db-6d07-4d29-8e5c-9c6909300925",
+            },
+            "inputs": [
+                {
+                    "id": "310072db-6d07-4d29-8e5c-9c6909300925",
+                    "key": "node_input",
+                    "value": {
+                        "rules": [
+                            {
+                                "type": "NODE_OUTPUT",
+                                "data": {
+                                    "node_id": "51cbe21d-0232-4362-bc54-5bc283297aa6",
+                                    "output_id": "6834cae4-8173-4fa6-88f7-bc09d335bdd1",
+                                },
+                            }
+                        ],
+                        "combinator": "OR",
+                    },
+                }
+            ],
+            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "definition": {
+                "name": "FinalOutputNode",
+                "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
+                "bases": [
+                    {"name": "BaseNode", "module": ["vellum", "workflows", "nodes", "bases", "base"], "bases": []}
+                ],
+            },
+        },
+        final_output_node,
+        ignore_order=True,
+    )
+
+    # AND each edge should be serialized correctly
+    serialized_edges = workflow_raw_data["edges"]
+    assert not DeepDiff(
+        [
+            {
+                "id": "9f753b40-68de-4186-838e-806a14853935",
+                "source_node_id": "09579c0a-acbf-4e04-8724-2230a8aa6534",
+                "source_handle_id": "34069190-0942-4e0c-8700-b33b9dea4ea0",
+                "target_node_id": "51cbe21d-0232-4362-bc54-5bc283297aa6",
+                "target_handle_id": "58427684-3848-498a-8299-c6b0fc70265d",
+                "type": "DEFAULT",
+            },
+            {
+                "id": "00cbff5c-5cce-4ef4-81b2-1a11d9b42597",
+                "source_node_id": "51cbe21d-0232-4362-bc54-5bc283297aa6",
+                "source_handle_id": "39317827-df43-4f5a-bfbc-20bffc839748",
+                "target_node_id": "9f75228b-1d5b-4c30-a581-6087e6a1b738",
+                "target_handle_id": "16ba108e-61a8-4338-8a5b-4f1278d7fd7b",
+                "type": "DEFAULT",
+            },
+        ],
+        serialized_edges,
+        ignore_order=True,
+    )
+
+    # AND the display data should be what we expect
+    display_data = workflow_raw_data["display_data"]
+    assert display_data == {
+        "viewport": {
+            "x": 0.0,
+            "y": 0.0,
+            "zoom": 1.0,
+        }
+    }
+
+    # AND the definition should be what we expect
+    definition = workflow_raw_data["definition"]
+    assert definition == {
+        "name": "BasicTemplatingNodeWorkflowWithJson",
+        "module": ["tests", "workflows", "basic_templating_node", "workflow_with_json_input"],
+    }

vellum/client/types/fulfilled_enum.py

@@ -1,5 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-
-FulfilledEnum = typing.Literal["FULFILLED"]

vellum/workflows/nodes/bases/base_subworkflow_node/__init__.py

@@ -1,5 +0,0 @@
-from .node import BaseSubworkflowNode
-
-__all__ = [
-    "BaseSubworkflowNode",
-]

vellum/workflows/nodes/bases/base_subworkflow_node/node.py

@@ -1,10 +0,0 @@
-from typing import ClassVar, Generic
-
-from vellum.workflows.nodes.bases import BaseNode
-from vellum.workflows.types.core import EntityInputsInterface
-from vellum.workflows.types.generics import StateType
-
-
-class BaseSubworkflowNode(BaseNode[StateType], Generic[StateType]):
-    # Inputs that are passed to the Subworkflow
-    subworkflow_inputs: ClassVar[EntityInputsInterface] = {}