vellum-ai 0.13.19__tar.gz → 0.13.21__tar.gz

{vellum_ai-0.13.19 → vellum_ai-0.13.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.13.19
+Version: 0.13.21
 Summary: 
 License: MIT
 Requires-Python: >=3.9,<4.0

{vellum_ai-0.13.19 → vellum_ai-0.13.21}/ee/vellum_ee/workflows/display/nodes/vellum/subworkflow_deployment_node.py

@@ -1,5 +1,5 @@
 from uuid import UUID
-from typing import ClassVar, Dict, Generic, Optional, TypeVar
+from typing import Generic, Optional, TypeVar
 
 from vellum.workflows.nodes import SubworkflowDeploymentNode
 from vellum.workflows.types.core import JsonObject
@@ -15,7 +15,6 @@ _SubworkflowDeploymentNodeType = TypeVar("_SubworkflowDeploymentNodeType", bound
 class BaseSubworkflowDeploymentNodeDisplay(
     BaseNodeVellumDisplay[_SubworkflowDeploymentNodeType], Generic[_SubworkflowDeploymentNodeType]
 ):
-    subworkflow_input_ids_by_name: ClassVar[Dict[str, UUID]] = {}
 
     def serialize(
         self, display_context: WorkflowDisplayContext, error_output_id: Optional[UUID] = None, **kwargs
@@ -30,7 +29,7 @@ class BaseSubworkflowDeploymentNodeDisplay(
                 input_name=variable_name,
                 value=variable_value,
                 display_context=display_context,
-                input_id=self.subworkflow_input_ids_by_name.get(variable_name),
+                input_id=self.node_input_ids_by_name.get(variable_name),
             )
             for variable_name, variable_value in subworkflow_inputs.items()
         ]

{vellum_ai-0.13.19 → vellum_ai-0.13.21}/ee/vellum_ee/workflows/display/nodes/vellum/utils.py

@@ -27,7 +27,7 @@ def create_node_input(
     input_name: str,
     value: Any,
     display_context: WorkflowDisplayContext,
-    input_id: Union[Optional[UUID], Optional[str]],
+    input_id: Union[Optional[UUID], Optional[str]] = None,
     pointer_type: Optional[Type[NodeInputValuePointerRule]] = ConstantValuePointer,
 ) -> NodeInput:
     input_id = str(input_id) if input_id else str(uuid4_from_hash(f"{node_id}|{input_name}"))

{vellum_ai-0.13.19 → vellum_ai-0.13.21}/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_api_node_serialization.py

@@ -210,11 +210,11 @@ def test_serialize_workflow(vellum_client):
                     "target_handle_id": "06853542-e1a1-4a00-bd1e-4ac40f347b32",
                     "output_id": "9a37bf7d-484e-4725-903e-f3254df38a0a",
                     "output_type": "JSON",
-                    "node_input_id": "b544fc9a-2747-47e9-b28b-3e88d87b0f95",
+                    "node_input_id": "b49f0f85-37fc-4686-81a7-287c06634661",
                 },
                 "inputs": [
                     {
-                        "id": "b544fc9a-2747-47e9-b28b-3e88d87b0f95",
+                        "id": "b49f0f85-37fc-4686-81a7-287c06634661",
                         "key": "node_input",
                         "value": {
                             "rules": [
@@ -233,14 +233,7 @@ def test_serialize_workflow(vellum_client):
                 "display_data": {"position": {"x": 0.0, "y": 0.0}},
                 "base": {
                     "name": "FinalOutputNode",
-                    "module": [
-                        "vellum",
-                        "workflows",
-                        "nodes",
-                        "displayable",
-                        "final_output_node",
-                        "node",
-                    ],
+                    "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
                 },
                 "definition": None,
             },
@@ -253,11 +246,11 @@ def test_serialize_workflow(vellum_client):
                     "target_handle_id": "80d0894f-642e-4d2e-b43a-f236e7bedb3c",
                     "output_id": "5090e96d-5787-4a08-bf58-129101cf2548",
                     "output_type": "JSON",
-                    "node_input_id": "6828b56e-80b7-4699-b9dd-fd1f0820732e",
+                    "node_input_id": "5e892e5b-0004-4a04-bd2e-1ea9e0e5d3f9",
                 },
                 "inputs": [
                     {
-                        "id": "6828b56e-80b7-4699-b9dd-fd1f0820732e",
+                        "id": "5e892e5b-0004-4a04-bd2e-1ea9e0e5d3f9",
                         "key": "node_input",
                         "value": {
                             "rules": [
@@ -276,14 +269,7 @@ def test_serialize_workflow(vellum_client):
                 "display_data": {"position": {"x": 0.0, "y": 0.0}},
                 "base": {
                     "name": "FinalOutputNode",
-                    "module": [
-                        "vellum",
-                        "workflows",
-                        "nodes",
-                        "displayable",
-                        "final_output_node",
-                        "node",
-                    ],
+                    "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
                 },
                 "definition": None,
             },
@@ -296,11 +282,11 @@ def test_serialize_workflow(vellum_client):
                     "target_handle_id": "0c98c306-b519-40d7-8b05-321b1dfd7f11",
                     "output_id": "44ea8d75-e2a8-4627-85b1-8504b65d25c9",
                     "output_type": "NUMBER",
-                    "node_input_id": "500ff745-344f-4094-9425-48c4b40b7a5d",
+                    "node_input_id": "14345321-7e6b-4e2a-918a-7a5b0064f047",
                 },
                 "inputs": [
                     {
-                        "id": "500ff745-344f-4094-9425-48c4b40b7a5d",
+                        "id": "14345321-7e6b-4e2a-918a-7a5b0064f047",
                         "key": "node_input",
                         "value": {
                             "rules": [
@@ -319,14 +305,7 @@ def test_serialize_workflow(vellum_client):
                 "display_data": {"position": {"x": 0.0, "y": 0.0}},
                 "base": {
                     "name": "FinalOutputNode",
-                    "module": [
-                        "vellum",
-                        "workflows",
-                        "nodes",
-                        "displayable",
-                        "final_output_node",
-                        "node",
-                    ],
+                    "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
                 },
                 "definition": None,
             },

{vellum_ai-0.13.19 → vellum_ai-0.13.21}/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_code_execution_node_serialization.py

@@ -125,11 +125,11 @@ def test_serialize_workflow_with_filepath():
                     "target_handle_id": "30fb0f4a-61c3-49de-a0aa-7dfdcee6ea07",
                     "output_id": "f6a3e3e0-f83f-4491-8b7a-b20fddd7160c",
                     "output_type": "NUMBER",
-                    "node_input_id": "ae302487-ff2a-457a-81ed-9e0348e91833",
+                    "node_input_id": "529bdd20-0985-4c99-87dc-590907937c1d",
                 },
                 "inputs": [
                     {
-                        "id": "ae302487-ff2a-457a-81ed-9e0348e91833",
+                        "id": "529bdd20-0985-4c99-87dc-590907937c1d",
                         "key": "node_input",
                         "value": {
                             "rules": [
@@ -161,11 +161,11 @@ def test_serialize_workflow_with_filepath():
                     "target_handle_id": "1e126004-9de7-42c0-b1e1-87f9eb0642e2",
                     "output_id": "1cee930f-342f-421c-89fc-ff212b3764bb",
                     "output_type": "STRING",
-                    "node_input_id": "c6593516-ffc5-49a8-8a65-1038cccec3f8",
+                    "node_input_id": "09501b65-d9b3-4920-81d4-96f93c840667",
                 },
                 "inputs": [
                     {
-                        "id": "c6593516-ffc5-49a8-8a65-1038cccec3f8",
+                        "id": "09501b65-d9b3-4920-81d4-96f93c840667",
                         "key": "node_input",
                         "value": {
                             "rules": [
@@ -358,11 +358,11 @@ def test_serialize_workflow_with_code():
                     "target_handle_id": "de8f2cc2-8c32-4782-87d5-4eb5afcd42e3",
                     "output_id": "283d6849-f3ed-4beb-b261-cf70f90e8d10",
                     "output_type": "NUMBER",
-                    "node_input_id": "b38ba7a8-0b2a-4146-8d58-9fa0bcba8cd5",
+                    "node_input_id": "101d5222-8578-413a-be80-b1acf7559a0d",
                 },
                 "inputs": [
                     {
-                        "id": "b38ba7a8-0b2a-4146-8d58-9fa0bcba8cd5",
+                        "id": "101d5222-8578-413a-be80-b1acf7559a0d",
                         "key": "node_input",
                         "value": {
                             "rules": [
@@ -394,11 +394,11 @@ def test_serialize_workflow_with_code():
                     "target_handle_id": "6b7d7f2c-5cc8-4005-9e66-cdb2c97b1998",
                     "output_id": "4c136180-050b-4422-a7a4-2a1c6729042c",
                     "output_type": "STRING",
-                    "node_input_id": "76d49710-1ed0-4105-a1d7-9190c0408558",
+                    "node_input_id": "de7ea5c4-da2f-4cfd-9192-0257d0f9ba53",
                 },
                 "inputs": [
                     {
-                        "id": "76d49710-1ed0-4105-a1d7-9190c0408558",
+                        "id": "de7ea5c4-da2f-4cfd-9192-0257d0f9ba53",
                         "key": "node_input",
                         "value": {
                             "rules": [
@@ -601,29 +601,17 @@ def test_serialize_workflow__try_wrapped():
             {
                 "id": "af4fc1ef-7701-43df-b5e7-4f354f707db2",
                 "type": "TERMINAL",
-                "base": {
-                    "module": [
-                        "vellum",
-                        "workflows",
-                        "nodes",
-                        "displayable",
-                        "final_output_node",
-                        "node",
-                    ],
-                    "name": "FinalOutputNode",
-                },
-                "definition": None,
                 "data": {
                     "label": "Final Output",
                     "name": "log",
                     "target_handle_id": "d243df8d-46f6-4928-ac31-7c775c5d73a9",
                     "output_id": "5fbd27a0-9831-49c7-93c8-9c2a28c78696",
                     "output_type": "STRING",
-                    "node_input_id": "569e1c05-53fa-413b-abf0-0353eaa44208",
+                    "node_input_id": "a6ffaacf-1284-4e70-907e-042bb281e6fa",
                 },
                 "inputs": [
                     {
-                        "id": "569e1c05-53fa-413b-abf0-0353eaa44208",
+                        "id": "a6ffaacf-1284-4e70-907e-042bb281e6fa",
                         "key": "node_input",
                         "value": {
                             "rules": [
@@ -640,33 +628,26 @@ def test_serialize_workflow__try_wrapped():
                     }
                 ],
                 "display_data": {"position": {"x": 0.0, "y": 0.0}},
-            },
-            {
-                "id": "4cbfa5f7-fc12-4ab2-81cb-168c5caef4f0",
-                "type": "TERMINAL",
                 "base": {
-                    "module": [
-                        "vellum",
-                        "workflows",
-                        "nodes",
-                        "displayable",
-                        "final_output_node",
-                        "node",
-                    ],
                     "name": "FinalOutputNode",
+                    "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
                 },
                 "definition": None,
+            },
+            {
+                "id": "4cbfa5f7-fc12-4ab2-81cb-168c5caef4f0",
+                "type": "TERMINAL",
                 "data": {
                     "label": "Final Output",
                     "name": "result",
                     "target_handle_id": "9c43709e-25cb-4548-b840-3fcf6a1c9f3e",
                     "output_id": "400f9ffe-e700-4204-a810-e06123565947",
                     "output_type": "NUMBER",
-                    "node_input_id": "3d6cb7ef-985f-48f1-ad23-de49be60666a",
+                    "node_input_id": "2cbb7351-a3e8-4a6b-b33e-204b04da7ad6",
                 },
                 "inputs": [
                     {
-                        "id": "3d6cb7ef-985f-48f1-ad23-de49be60666a",
+                        "id": "2cbb7351-a3e8-4a6b-b33e-204b04da7ad6",
                         "key": "node_input",
                         "value": {
                             "rules": [
@@ -683,6 +664,11 @@ def test_serialize_workflow__try_wrapped():
                     }
                 ],
                 "display_data": {"position": {"x": 0.0, "y": 0.0}},
+                "base": {
+                    "name": "FinalOutputNode",
+                    "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
+                },
+                "definition": None,
             },
         ],
         sorted(final_output_nodes, key=lambda x: x["id"], reverse=True),

{vellum_ai-0.13.19 → vellum_ai-0.13.21}/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_conditional_node_serialization.py

@@ -438,29 +438,17 @@ def test_serialize_workflow():
             {
                 "id": "9c22ee47-01da-4e4e-863d-b4a6874bed66",
                 "type": "TERMINAL",
-                "base": {
-                    "module": [
-                        "vellum",
-                        "workflows",
-                        "nodes",
-                        "displayable",
-                        "final_output_node",
-                        "node",
-                    ],
-                    "name": "FinalOutputNode",
-                },
-                "definition": None,
                 "data": {
                     "label": "Final Output",
                     "name": "statement",
                     "target_handle_id": "f02a8971-e9a4-4716-bfb4-d08f5614b5d8",
                     "output_id": "cdbe2adf-9951-409a-b9a8-b8b349037f4f",
                     "output_type": "STRING",
-                    "node_input_id": "bed69f8a-3a83-4e52-beba-8eb14f4b0ca9",
+                    "node_input_id": "2e742a40-cbee-4e19-9269-c62dc4a9204e",
                 },
                 "inputs": [
                     {
-                        "id": "bed69f8a-3a83-4e52-beba-8eb14f4b0ca9",
+                        "id": "2e742a40-cbee-4e19-9269-c62dc4a9204e",
                         "key": "node_input",
                         "value": {
                             "rules": [
@@ -477,33 +465,26 @@ def test_serialize_workflow():
                     }
                 ],
                 "display_data": {"position": {"x": 0.0, "y": 0.0}},
-            },
-            {
-                "id": "47f0931c-41f6-4b84-bf39-0c486941f599",
-                "type": "TERMINAL",
                 "base": {
-                    "module": [
-                        "vellum",
-                        "workflows",
-                        "nodes",
-                        "displayable",
-                        "final_output_node",
-                        "node",
-                    ],
                     "name": "FinalOutputNode",
+                    "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
                 },
                 "definition": None,
+            },
+            {
+                "id": "47f0931c-41f6-4b84-bf39-0c486941f599",
+                "type": "TERMINAL",
                 "data": {
                     "label": "Final Output",
                     "name": "compliment",
                     "target_handle_id": "a4d57adc-58c1-40c6-810b-ee5fd923bfc5",
                     "output_id": "93f2cb75-6fa2-4e46-9488-c0bcd29153c0",
                     "output_type": "STRING",
-                    "node_input_id": "e3f2d793-31a7-4670-8155-bd034d9f25e2",
+                    "node_input_id": "9ba792e6-55e3-4a14-8768-e8ef6955c934",
                 },
                 "inputs": [
                     {
-                        "id": "e3f2d793-31a7-4670-8155-bd034d9f25e2",
+                        "id": "9ba792e6-55e3-4a14-8768-e8ef6955c934",
                         "key": "node_input",
                         "value": {
                             "rules": [
@@ -520,33 +501,26 @@ def test_serialize_workflow():
                     }
                 ],
                 "display_data": {"position": {"x": 0.0, "y": 0.0}},
-            },
-            {
-                "id": "e3d29229-f746-4125-819e-f847acbed307",
-                "type": "TERMINAL",
                 "base": {
-                    "module": [
-                        "vellum",
-                        "workflows",
-                        "nodes",
-                        "displayable",
-                        "final_output_node",
-                        "node",
-                    ],
                     "name": "FinalOutputNode",
+                    "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
                 },
                 "definition": None,
+            },
+            {
+                "id": "e3d29229-f746-4125-819e-f847acbed307",
+                "type": "TERMINAL",
                 "data": {
                     "label": "Final Output",
                     "name": "complaint",
                     "target_handle_id": "c5dd9bf5-9e18-4dbc-8c20-2c0baf969ebe",
                     "output_id": "f936ae31-ba15-4864-8961-86231022a4d7",
                     "output_type": "STRING",
-                    "node_input_id": "9b35cc61-5887-478a-814d-9693ead5932f",
+                    "node_input_id": "47f426a4-8770-4f30-a285-5d21849063a5",
                 },
                 "inputs": [
                     {
-                        "id": "9b35cc61-5887-478a-814d-9693ead5932f",
+                        "id": "47f426a4-8770-4f30-a285-5d21849063a5",
                         "key": "node_input",
                         "value": {
                             "rules": [
@@ -563,33 +537,26 @@ def test_serialize_workflow():
                     }
                 ],
                 "display_data": {"position": {"x": 0.0, "y": 0.0}},
-            },
-            {
-                "id": "6efa7b45-0580-406d-85aa-439117ba8021",
-                "type": "TERMINAL",
                 "base": {
-                    "module": [
-                        "vellum",
-                        "workflows",
-                        "nodes",
-                        "displayable",
-                        "final_output_node",
-                        "node",
-                    ],
                     "name": "FinalOutputNode",
+                    "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
                 },
                 "definition": None,
+            },
+            {
+                "id": "6efa7b45-0580-406d-85aa-439117ba8021",
+                "type": "TERMINAL",
                 "data": {
                     "label": "Final Output",
                     "name": "fallthrough",
                     "target_handle_id": "2283cd2c-b077-4b5d-a96f-aa2cd6023eda",
                     "output_id": "62ad462f-f819-4940-99ab-b3f145507f57",
                     "output_type": "STRING",
-                    "node_input_id": "cee5378a-9011-4d03-bd3c-a32421bd093f",
+                    "node_input_id": "d2d3e7cc-f6d6-4ac1-ad19-e6f52a75b38f",
                 },
                 "inputs": [
                     {
-                        "id": "cee5378a-9011-4d03-bd3c-a32421bd093f",
+                        "id": "d2d3e7cc-f6d6-4ac1-ad19-e6f52a75b38f",
                         "key": "node_input",
                         "value": {
                             "rules": [
@@ -606,33 +573,26 @@ def test_serialize_workflow():
                     }
                 ],
                 "display_data": {"position": {"x": 0.0, "y": 0.0}},
-            },
-            {
-                "id": "fa11b84b-1d76-4adc-ab28-cbbaa933c267",
-                "type": "TERMINAL",
                 "base": {
-                    "module": [
-                        "vellum",
-                        "workflows",
-                        "nodes",
-                        "displayable",
-                        "final_output_node",
-                        "node",
-                    ],
                     "name": "FinalOutputNode",
+                    "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
                 },
                 "definition": None,
+            },
+            {
+                "id": "fa11b84b-1d76-4adc-ab28-cbbaa933c267",
+                "type": "TERMINAL",
                 "data": {
                     "label": "Final Output",
                     "name": "question",
                     "target_handle_id": "e1a6da28-02c5-40d7-8ac5-9fb07e2e3e1d",
                     "output_id": "c05f7d96-59a0-4d58-93d7-d451afd3f630",
                     "output_type": "STRING",
-                    "node_input_id": "1c1d34d9-0cda-471f-ac9a-40ac351c9aca",
+                    "node_input_id": "a13377c4-ae1c-4be4-b695-f9da590486ef",
                 },
                 "inputs": [
                     {
-                        "id": "1c1d34d9-0cda-471f-ac9a-40ac351c9aca",
+                        "id": "a13377c4-ae1c-4be4-b695-f9da590486ef",
                         "key": "node_input",
                         "value": {
                             "rules": [
@@ -649,6 +609,11 @@ def test_serialize_workflow():
                     }
                 ],
                 "display_data": {"position": {"x": 0.0, "y": 0.0}},
+                "base": {
+                    "name": "FinalOutputNode",
+                    "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
+                },
+                "definition": None,
             },
         ],
         workflow_raw_data["nodes"][7:12],

{vellum_ai-0.13.19 → vellum_ai-0.13.21}/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_error_node_serialization.py

@@ -138,11 +138,11 @@ def test_serialize_workflow():
                 "target_handle_id": "b070e9bc-e9b7-46d3-8f5b-0b646bd25cf0",
                 "output_id": "04c5c6be-f5e1-41b8-b668-39e179790d9e",
                 "output_type": "NUMBER",
-                "node_input_id": "39ff42c9-eae8-432e-ad41-e208fba77027",
+                "node_input_id": "c191a5c1-8912-49ad-bf4b-c6d4d58ed482",
             },
             "inputs": [
                 {
-                    "id": "39ff42c9-eae8-432e-ad41-e208fba77027",
+                    "id": "c191a5c1-8912-49ad-bf4b-c6d4d58ed482",
                     "key": "node_input",
                     "value": {
                         "rules": [

{vellum_ai-0.13.19 → vellum_ai-0.13.21}/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_generic_node_serialization.py

@@ -83,11 +83,11 @@ def test_serialize_workflow(vellum_client):
                 "target_handle_id": "8bd9f4f3-9f66-4d95-8e84-529b0002c531",
                 "output_id": "2b6389d0-266a-4be4-843e-4e543dd3d727",
                 "output_type": "STRING",
-                "node_input_id": "7a9f2d3a-0b23-4bd4-b567-e9493135b727",
+                "node_input_id": "545d6001-cfb5-4ccc-bcdf-3b03ccd67d90",
             },
             "inputs": [
                 {
-                    "id": "7a9f2d3a-0b23-4bd4-b567-e9493135b727",
+                    "id": "545d6001-cfb5-4ccc-bcdf-3b03ccd67d90",
                     "key": "node_input",
                     "value": {
                         "rules": [

{vellum_ai-0.13.19 → vellum_ai-0.13.21}/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_guardrail_node_serialization.py

@@ -140,11 +140,11 @@ def test_serialize_workflow():
             "target_handle_id": "001b97f6-2bc8-4d1e-9572-028dcf17df4e",
             "output_id": "2abd2b3b-c301-4834-a43f-5db3604f8422",
             "output_type": "NUMBER",
-            "node_input_id": "6580eb90-2e99-4010-bebe-4b9fc4cf0dfd",
+            "node_input_id": "6321442a-0d0d-4e25-965d-c24ff24712c5",
         },
         "inputs": [
             {
-                "id": "6580eb90-2e99-4010-bebe-4b9fc4cf0dfd",
+                "id": "6321442a-0d0d-4e25-965d-c24ff24712c5",
                 "key": "node_input",
                 "value": {
                     "rules": [