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

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

@@ -104,17 +104,8 @@ def test_serialize_workflow():
     assert entrypoint_node == {
         "id": "089b3201-537a-4ed7-8d15-2524a00e8534",
         "type": "ENTRYPOINT",
-        "definition": {
-            "bases": [],
-            "module": [
-                "vellum",
-                "workflows",
-                "nodes",
-                "bases",
-                "base",
-            ],
-            "name": "BaseNode",
-        },
+        "base": None,
+        "definition": None,
         "inputs": [],
         "data": {
             "label": "Entrypoint Node",
@@ -426,22 +417,13 @@ def test_serialize_workflow():
                 "version": "2",
             },
             "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "base": {
+                "name": "ConditionalNode",
+                "module": ["vellum", "workflows", "nodes", "displayable", "conditional_node", "node"],
+            },
             "definition": {
                 "name": "CategoryConditionalNode",
                 "module": ["tests", "workflows", "basic_conditional_node", "workflow"],
-                "bases": [
-                    {
-                        "name": "ConditionalNode",
-                        "module": [
-                            "vellum",
-                            "workflows",
-                            "nodes",
-                            "displayable",
-                            "conditional_node",
-                            "node",
-                        ],
-                    }
-                ],
             },
         },
         conditional_node,
@@ -456,20 +438,7 @@ def test_serialize_workflow():
             {
                 "id": "9c22ee47-01da-4e4e-863d-b4a6874bed66",
                 "type": "TERMINAL",
-                "definition": {
-                    "bases": [
-                        {
-                            "bases": [],
-                            "module": [
-                                "vellum",
-                                "workflows",
-                                "nodes",
-                                "bases",
-                                "base",
-                            ],
-                            "name": "BaseNode",
-                        },
-                    ],
+                "base": {
                     "module": [
                         "vellum",
                         "workflows",
@@ -480,6 +449,7 @@ def test_serialize_workflow():
                     ],
                     "name": "FinalOutputNode",
                 },
+                "definition": None,
                 "data": {
                     "label": "Final Output",
                     "name": "statement",
@@ -511,20 +481,7 @@ def test_serialize_workflow():
             {
                 "id": "47f0931c-41f6-4b84-bf39-0c486941f599",
                 "type": "TERMINAL",
-                "definition": {
-                    "bases": [
-                        {
-                            "bases": [],
-                            "module": [
-                                "vellum",
-                                "workflows",
-                                "nodes",
-                                "bases",
-                                "base",
-                            ],
-                            "name": "BaseNode",
-                        },
-                    ],
+                "base": {
                     "module": [
                         "vellum",
                         "workflows",
@@ -535,6 +492,7 @@ def test_serialize_workflow():
                     ],
                     "name": "FinalOutputNode",
                 },
+                "definition": None,
                 "data": {
                     "label": "Final Output",
                     "name": "compliment",
@@ -566,20 +524,7 @@ def test_serialize_workflow():
             {
                 "id": "e3d29229-f746-4125-819e-f847acbed307",
                 "type": "TERMINAL",
-                "definition": {
-                    "bases": [
-                        {
-                            "bases": [],
-                            "module": [
-                                "vellum",
-                                "workflows",
-                                "nodes",
-                                "bases",
-                                "base",
-                            ],
-                            "name": "BaseNode",
-                        },
-                    ],
+                "base": {
                     "module": [
                         "vellum",
                         "workflows",
@@ -590,6 +535,7 @@ def test_serialize_workflow():
                     ],
                     "name": "FinalOutputNode",
                 },
+                "definition": None,
                 "data": {
                     "label": "Final Output",
                     "name": "complaint",
@@ -621,20 +567,7 @@ def test_serialize_workflow():
             {
                 "id": "6efa7b45-0580-406d-85aa-439117ba8021",
                 "type": "TERMINAL",
-                "definition": {
-                    "bases": [
-                        {
-                            "bases": [],
-                            "module": [
-                                "vellum",
-                                "workflows",
-                                "nodes",
-                                "bases",
-                                "base",
-                            ],
-                            "name": "BaseNode",
-                        },
-                    ],
+                "base": {
                     "module": [
                         "vellum",
                         "workflows",
@@ -645,6 +578,7 @@ def test_serialize_workflow():
                     ],
                     "name": "FinalOutputNode",
                 },
+                "definition": None,
                 "data": {
                     "label": "Final Output",
                     "name": "fallthrough",
@@ -676,20 +610,7 @@ def test_serialize_workflow():
             {
                 "id": "fa11b84b-1d76-4adc-ab28-cbbaa933c267",
                 "type": "TERMINAL",
-                "definition": {
-                    "bases": [
-                        {
-                            "bases": [],
-                            "module": [
-                                "vellum",
-                                "workflows",
-                                "nodes",
-                                "bases",
-                                "base",
-                            ],
-                            "name": "BaseNode",
-                        },
-                    ],
+                "base": {
                     "module": [
                         "vellum",
                         "workflows",
@@ -700,6 +621,7 @@ def test_serialize_workflow():
                     ],
                     "name": "FinalOutputNode",
                 },
+                "definition": None,
                 "data": {
                     "label": "Final Output",
                     "name": "question",
@@ -977,27 +899,13 @@ def test_conditional_node_serialize_all_operators_with_lhs_and_rhs(descriptor, o
                 "version": "2",
             },
             "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "base": {
+                "name": "ConditionalNode",
+                "module": ["vellum", "workflows", "nodes", "displayable", "conditional_node", "node"],
+            },
             "definition": {
                 "name": "SimpleConditionalNode",
-                "module": [
-                    "tests",
-                    "workflows",
-                    "basic_conditional_node",
-                    "workflow_with_only_one_conditional_node",
-                ],
-                "bases": [
-                    {
-                        "name": "ConditionalNode",
-                        "module": [
-                            "vellum",
-                            "workflows",
-                            "nodes",
-                            "displayable",
-                            "conditional_node",
-                            "node",
-                        ],
-                    }
-                ],
+                "module": ["tests", "workflows", "basic_conditional_node", "workflow_with_only_one_conditional_node"],
             },
         },
         conditional_node,
@@ -1083,27 +991,13 @@ def test_conditional_node_serialize_all_operators_with_expression(descriptor, op
                 "version": "2",
             },
             "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "base": {
+                "name": "ConditionalNode",
+                "module": ["vellum", "workflows", "nodes", "displayable", "conditional_node", "node"],
+            },
             "definition": {
                 "name": "SimpleConditionalNode",
-                "module": [
-                    "tests",
-                    "workflows",
-                    "basic_conditional_node",
-                    "workflow_with_only_one_conditional_node",
-                ],
-                "bases": [
-                    {
-                        "name": "ConditionalNode",
-                        "module": [
-                            "vellum",
-                            "workflows",
-                            "nodes",
-                            "displayable",
-                            "conditional_node",
-                            "node",
-                        ],
-                    }
-                ],
+                "module": ["tests", "workflows", "basic_conditional_node", "workflow_with_only_one_conditional_node"],
             },
         },
         conditional_node,
@@ -1202,27 +1096,13 @@ def test_conditional_node_serialize_all_operators_with_value_and_start_and_end(d
                 "version": "2",
             },
             "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "base": {
+                "name": "ConditionalNode",
+                "module": ["vellum", "workflows", "nodes", "displayable", "conditional_node", "node"],
+            },
             "definition": {
                 "name": "SimpleConditionalNode",
-                "module": [
-                    "tests",
-                    "workflows",
-                    "basic_conditional_node",
-                    "workflow_with_only_one_conditional_node",
-                ],
-                "bases": [
-                    {
-                        "name": "ConditionalNode",
-                        "module": [
-                            "vellum",
-                            "workflows",
-                            "nodes",
-                            "displayable",
-                            "conditional_node",
-                            "node",
-                        ],
-                    }
-                ],
+                "module": ["tests", "workflows", "basic_conditional_node", "workflow_with_only_one_conditional_node"],
             },
         },
         conditional_node,

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

@@ -72,11 +72,8 @@ def test_serialize_workflow():
             "source_handle_id": "7d86498b-84ed-4feb-8e62-2188058c2c4e",
         },
         "display_data": {"position": {"x": 0.0, "y": 0.0}},
-        "definition": {
-            "name": "BaseNode",
-            "module": ["vellum", "workflows", "nodes", "bases", "base"],
-            "bases": [],
-        },
+        "base": None,
+        "definition": None,
     }
 
     error_node, error_index = next(
@@ -100,22 +97,13 @@ def test_serialize_workflow():
                 "error_output_id": "None",
             },
             "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "base": {
+                "name": "ErrorNode",
+                "module": ["vellum", "workflows", "nodes", "core", "error_node", "node"],
+            },
             "definition": {
                 "name": "FailNode",
                 "module": ["tests", "workflows", "basic_error_node", "workflow"],
-                "bases": [
-                    {
-                        "name": "ErrorNode",
-                        "module": [
-                            "vellum",
-                            "workflows",
-                            "nodes",
-                            "core",
-                            "error_node",
-                            "node",
-                        ],
-                    }
-                ],
             },
         },
         error_node,
@@ -157,7 +145,7 @@ def test_serialize_workflow():
                 }
             ],
             "display_data": {"position": {"x": 0.0, "y": 0.0}},
-            "definition": {
+            "base": {
                 "name": "FinalOutputNode",
                 "module": [
                     "vellum",
@@ -167,14 +155,8 @@ def test_serialize_workflow():
                     "final_output_node",
                     "node",
                 ],
-                "bases": [
-                    {
-                        "name": "BaseNode",
-                        "module": ["vellum", "workflows", "nodes", "bases", "base"],
-                        "bases": [],
-                    }
-                ],
             },
+            "definition": None,
         },
         terminal_node,
     )

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

@@ -65,17 +65,8 @@ def test_serialize_workflow(vellum_client):
         "inputs": [],
         "data": {"label": "Entrypoint Node", "source_handle_id": "40201804-8beb-43ad-8873-a027759512f1"},
         "display_data": {"position": {"x": 0.0, "y": 0.0}},
-        "definition": {
-            "name": "BaseNode",
-            "module": [
-                "vellum",
-                "workflows",
-                "nodes",
-                "bases",
-                "base",
-            ],
-            "bases": [],
-        },
+        "base": None,
+        "definition": None,
     }
 
     api_node = workflow_raw_data["nodes"][1]
@@ -113,13 +104,11 @@ def test_serialize_workflow(vellum_client):
                 }
             ],
             "display_data": {"position": {"x": 0.0, "y": 0.0}},
-            "definition": {
+            "base": {
                 "name": "FinalOutputNode",
                 "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
-                "bases": [
-                    {"name": "BaseNode", "module": ["vellum", "workflows", "nodes", "bases", "base"], "bases": []}
-                ],
             },
+            "definition": None,
         },
         final_output_node,
         ignore_order=True,

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

@@ -60,17 +60,8 @@ def test_serialize_workflow():
     assert entrypoint_node == {
         "id": "54c5c7d0-ab86-4ae9-b0b8-ea9ca7b87c14",
         "type": "ENTRYPOINT",
-        "definition": {
-            "bases": [],
-            "module": [
-                "vellum",
-                "workflows",
-                "nodes",
-                "bases",
-                "base",
-            ],
-            "name": "BaseNode",
-        },
+        "base": None,
+        "definition": None,
         "inputs": [],
         "data": {"label": "Entrypoint Node", "source_handle_id": "41840690-8d85-486e-a864-b0661ccf0f2e"},
         "display_data": {"position": {"x": 0.0, "y": 0.0}},
@@ -117,26 +108,12 @@ def test_serialize_workflow():
             "release_tag": "LATEST",
         },
         "display_data": {"position": {"x": 0.0, "y": 0.0}},
+        "base": {
+            "module": ["vellum", "workflows", "nodes", "displayable", "guardrail_node", "node"],
+            "name": "GuardrailNode",
+        },
         "definition": {
-            "bases": [
-                {
-                    "module": [
-                        "vellum",
-                        "workflows",
-                        "nodes",
-                        "displayable",
-                        "guardrail_node",
-                        "node",
-                    ],
-                    "name": "GuardrailNode",
-                }
-            ],
-            "module": [
-                "tests",
-                "workflows",
-                "basic_guardrail_node",
-                "workflow",
-            ],
+            "module": ["tests", "workflows", "basic_guardrail_node", "workflow"],
             "name": "ExampleGuardrailNode",
         },
     }
@@ -145,20 +122,7 @@ def test_serialize_workflow():
     assert final_output_node == {
         "id": "cbc7197e-67c9-4af5-b781-879c8fd3e4c9",
         "type": "TERMINAL",
-        "definition": {
-            "bases": [
-                {
-                    "bases": [],
-                    "module": [
-                        "vellum",
-                        "workflows",
-                        "nodes",
-                        "bases",
-                        "base",
-                    ],
-                    "name": "BaseNode",
-                },
-            ],
+        "base": {
             "module": [
                 "vellum",
                 "workflows",
@@ -169,6 +133,7 @@ def test_serialize_workflow():
             ],
             "name": "FinalOutputNode",
         },
+        "definition": None,
         "data": {
             "label": "Final Output",
             "name": "score",