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

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

@@ -16,8 +16,7 @@ class Inputs(BaseInputs):
 
 
 class BasicGenericNode(BaseNode):
-    class Outputs(BaseNode.Outputs):
-        output = Inputs.input
+    pass
 
 
 def test_serialize_node__basic(serialize_node):
@@ -45,11 +44,13 @@ def test_serialize_node__basic(serialize_node):
             "ports": [
                 {
                     "id": "4fbf0fff-a42e-4410-852a-238b5059198e",
+                    "name": "default",
                     "type": "DEFAULT",
                 }
             ],
             "adornments": None,
             "attributes": [],
+            "outputs": [],
         },
         serialized_node,
         ignore_order=True,
@@ -57,9 +58,6 @@ def test_serialize_node__basic(serialize_node):
 
 
 class IfGenericNode(BaseNode):
-    class Outputs(BaseNode.Outputs):
-        output = Inputs.input
-
     class Ports(BaseNode.Ports):
         if_branch = Port.on_if(Inputs.input.equals("hello"))
 
@@ -94,6 +92,7 @@ def test_serialize_node__if(serialize_node):
                 {
                     "id": "7605b4c0-a432-4517-b759-5858045a5146",
                     "type": "IF",
+                    "name": "if_branch",
                     "expression": {
                         "type": "BINARY_EXPRESSION",
                         "lhs": {
@@ -113,6 +112,7 @@ def test_serialize_node__if(serialize_node):
             ],
             "adornments": None,
             "attributes": [],
+            "outputs": [],
         },
         serialized_node,
         ignore_order=True,
@@ -120,9 +120,6 @@ def test_serialize_node__if(serialize_node):
 
 
 class IfElseGenericNode(BaseNode):
-    class Outputs(BaseNode.Outputs):
-        output = Inputs.input
-
     class Ports(BaseNode.Ports):
         if_branch = Port.on_if(Inputs.input.equals("hello"))
         else_branch = Port.on_else()
@@ -158,6 +155,7 @@ def test_serialize_node__if_else(serialize_node):
                 {
                     "id": "3eeb7f03-7d65-45aa-b0e5-c7a453f5cbdf",
                     "type": "IF",
+                    "name": "if_branch",
                     "expression": {
                         "type": "BINARY_EXPRESSION",
                         "lhs": {
@@ -177,11 +175,13 @@ def test_serialize_node__if_else(serialize_node):
                 {
                     "id": "b8472c77-74d5-4432-bf8b-6cd65d3dde06",
                     "type": "ELSE",
+                    "name": "else_branch",
                     "expression": None,
                 },
             ],
             "adornments": None,
             "attributes": [],
+            "outputs": [],
         },
         serialized_node,
         ignore_order=True,
@@ -189,9 +189,6 @@ def test_serialize_node__if_else(serialize_node):
 
 
 class IfElifElseGenericNode(BaseNode):
-    class Outputs(BaseNode.Outputs):
-        output = Inputs.input
-
     class Ports(BaseNode.Ports):
         if_branch = Port.on_if(Inputs.input.equals("hello"))
         elif_branch = Port.on_elif(Inputs.input.equals("world"))
@@ -228,6 +225,7 @@ def test_serialize_node__if_elif_else(serialize_node):
                 {
                     "id": "f6e0a2c0-192d-452f-bde4-32fb938e91bc",
                     "type": "IF",
+                    "name": "if_branch",
                     "expression": {
                         "type": "BINARY_EXPRESSION",
                         "lhs": {
@@ -247,6 +245,7 @@ def test_serialize_node__if_elif_else(serialize_node):
                 {
                     "id": "7e44de04-e816-4da8-9251-cf389442a5d6",
                     "type": "ELIF",
+                    "name": "elif_branch",
                     "expression": {
                         "type": "BINARY_EXPRESSION",
                         "lhs": {
@@ -267,10 +266,12 @@ def test_serialize_node__if_elif_else(serialize_node):
                     "id": "00db3698-ddf5-413b-8408-fff664c212d7",
                     "type": "ELSE",
                     "expression": None,
+                    "name": "else_branch",
                 },
             ],
             "adornments": None,
             "attributes": [],
+            "outputs": [],
         },
         serialized_node,
         ignore_order=True,
@@ -287,9 +288,6 @@ class NodeWithOutputDisplay(BaseNodeDisplay[NodeWithOutput]):
 
 
 class GenericNodeReferencingOutput(BaseNode):
-    class Outputs(BaseNode.Outputs):
-        output = NodeWithOutput.Outputs.output
-
     class Ports(BaseNode.Ports):
         if_branch = Port.on_if(NodeWithOutput.Outputs.output.equals("hello"))
 
@@ -330,6 +328,7 @@ def test_serialize_node__node_output_reference(serialize_node):
                 {
                     "id": "ec9a79b8-65c3-4de8-bd29-42c914d72d4f",
                     "type": "IF",
+                    "name": "if_branch",
                     "expression": {
                         "type": "BINARY_EXPRESSION",
                         "lhs": {
@@ -350,6 +349,7 @@ def test_serialize_node__node_output_reference(serialize_node):
             ],
             "adornments": None,
             "attributes": [],
+            "outputs": [],
         },
         serialized_node,
         ignore_order=True,
@@ -357,9 +357,6 @@ def test_serialize_node__node_output_reference(serialize_node):
 
 
 class GenericNodeReferencingSecret(BaseNode):
-    class Outputs(BaseNode.Outputs):
-        output = Inputs.input
-
     class Ports(BaseNode.Ports):
         if_branch = Port.on_if(VellumSecretReference(name="hello").equals("hello"))
 
@@ -395,6 +392,7 @@ def test_serialize_node__vellum_secret_reference(serialize_node):
                 {
                     "id": "a353d3f6-2a1f-457c-b8d1-13db5b45be8f",
                     "type": "IF",
+                    "name": "if_branch",
                     "expression": {
                         "type": "BINARY_EXPRESSION",
                         "lhs": {"type": "VELLUM_SECRET", "vellum_secret_name": "hello"},
@@ -411,6 +409,7 @@ def test_serialize_node__vellum_secret_reference(serialize_node):
             ],
             "adornments": None,
             "attributes": [],
+            "outputs": [],
         },
         serialized_node,
         ignore_order=True,
@@ -426,9 +425,6 @@ class NodeWithExecutionsDisplay(BaseNodeDisplay[NodeWithExecutions]):
 
 
 class GenericNodeReferencingExecutions(BaseNode):
-    class Outputs(BaseNode.Outputs):
-        output = NodeWithExecutions.Execution.count
-
     class Ports(BaseNode.Ports):
         if_branch = Port.on_if(NodeWithExecutions.Execution.count.equals(5))
 
@@ -465,6 +461,7 @@ def test_serialize_node__execution_count_reference(serialize_node):
                 {
                     "id": "1794c2eb-5cab-49fe-9354-dfc29f11b374",
                     "type": "IF",
+                    "name": "if_branch",
                     "expression": {
                         "type": "BINARY_EXPRESSION",
                         "lhs": {
@@ -484,6 +481,7 @@ def test_serialize_node__execution_count_reference(serialize_node):
             ],
             "adornments": None,
             "attributes": [],
+            "outputs": [],
         },
         serialized_node,
         ignore_order=True,
@@ -491,9 +489,6 @@ def test_serialize_node__execution_count_reference(serialize_node):
 
 
 class NullGenericNode(BaseNode):
-    class Outputs(BaseNode.Outputs):
-        output = Inputs.input
-
     class Ports(BaseNode.Ports):
         if_branch = Port.on_if(Inputs.input.is_null())
 
@@ -528,6 +523,7 @@ def test_serialize_node__null(serialize_node):
                 {
                     "id": "51932d23-492e-4b3b-8b03-6ad7303a80c9",
                     "type": "IF",
+                    "name": "if_branch",
                     "expression": {
                         "type": "UNARY_EXPRESSION",
                         "lhs": {
@@ -540,6 +536,7 @@ def test_serialize_node__null(serialize_node):
             ],
             "adornments": None,
             "attributes": [],
+            "outputs": [],
         },
         serialized_node,
         ignore_order=True,
@@ -551,9 +548,6 @@ class IntegerInputs(BaseInputs):
 
 
 class BetweenGenericNode(BaseNode):
-    class Outputs(BaseNode.Outputs):
-        output = IntegerInputs.input
-
     class Ports(BaseNode.Ports):
         if_branch = Port.on_if(IntegerInputs.input.between(1, 10))
 
@@ -589,6 +583,7 @@ def test_serialize_node__between(serialize_node):
                 {
                     "id": "a86bd19f-a9f7-45c3-80ff-73330b1b75af",
                     "type": "IF",
+                    "name": "if_branch",
                     "expression": {
                         "type": "TERNARY_EXPRESSION",
                         "base": {
@@ -615,6 +610,7 @@ def test_serialize_node__between(serialize_node):
             ],
             "adornments": None,
             "attributes": [],
+            "outputs": [],
         },
         serialized_node,
         ignore_order=True,
@@ -622,9 +618,6 @@ def test_serialize_node__between(serialize_node):
 
 
 class OrGenericNode(BaseNode):
-    class Outputs(BaseNode.Outputs):
-        output = Inputs.input
-
     class Ports(BaseNode.Ports):
         if_branch = Port.on_if(Inputs.input.equals("hello") | Inputs.input.equals("world"))
 
@@ -659,6 +652,7 @@ def test_serialize_node__or(serialize_node):
                 {
                     "id": "652a42f9-f4e7-4791-8167-8903ff839520",
                     "type": "IF",
+                    "name": "if_branch",
                     "expression": {
                         "type": "BINARY_EXPRESSION",
                         "lhs": {
@@ -697,6 +691,7 @@ def test_serialize_node__or(serialize_node):
             ],
             "adornments": None,
             "attributes": [],
+            "outputs": [],
         },
         serialized_node,
         ignore_order=True,
@@ -704,9 +699,6 @@ def test_serialize_node__or(serialize_node):
 
 
 class AndThenOrGenericNode(BaseNode):
-    class Outputs(BaseNode.Outputs):
-        output = Inputs.input
-
     class Ports(BaseNode.Ports):
         if_branch = Port.on_if(
             Inputs.input.equals("hello") & Inputs.input.equals("then") | Inputs.input.equals("world")
@@ -744,6 +736,7 @@ def test_serialize_node__and_then_or(serialize_node):
                 {
                     "id": "42c89e95-6bbf-4e85-8f26-d4b6fc55d99c",
                     "type": "IF",
+                    "name": "if_branch",
                     "expression": {
                         "type": "BINARY_EXPRESSION",
                         "lhs": {
@@ -801,6 +794,7 @@ def test_serialize_node__and_then_or(serialize_node):
             ],
             "adornments": None,
             "attributes": [],
+            "outputs": [],
         },
         serialized_node,
         ignore_order=True,
@@ -808,9 +802,6 @@ def test_serialize_node__and_then_or(serialize_node):
 
 
 class ParenthesizedAndThenOrGenericNode(BaseNode):
-    class Outputs(BaseNode.Outputs):
-        output = Inputs.input
-
     class Ports(BaseNode.Ports):
         if_branch = Port.on_if(
             Inputs.input.equals("hello") & (Inputs.input.equals("then") | Inputs.input.equals("world"))
@@ -848,6 +839,7 @@ def test_serialize_node__parenthesized_and_then_or(serialize_node):
                 {
                     "id": "cc07394b-f20b-4370-8a5b-af90e847a73f",
                     "type": "IF",
+                    "name": "if_branch",
                     "expression": {
                         "type": "BINARY_EXPRESSION",
                         "lhs": {
@@ -905,6 +897,7 @@ def test_serialize_node__parenthesized_and_then_or(serialize_node):
             ],
             "adornments": None,
             "attributes": [],
+            "outputs": [],
         },
         serialized_node,
         ignore_order=True,
@@ -912,9 +905,6 @@ def test_serialize_node__parenthesized_and_then_or(serialize_node):
 
 
 class OrThenAndGenericNode(BaseNode):
-    class Outputs(BaseNode.Outputs):
-        output = Inputs.input
-
     class Ports(BaseNode.Ports):
         if_branch = Port.on_if(
             Inputs.input.equals("hello") | Inputs.input.equals("then") & Inputs.input.equals("world")
@@ -952,6 +942,7 @@ def test_serialize_node__or_then_and(serialize_node):
                 {
                     "id": "daaff604-da1e-45e6-b3df-5bc8de8d55fe",
                     "type": "IF",
+                    "name": "if_branch",
                     "expression": {
                         "type": "BINARY_EXPRESSION",
                         "lhs": {
@@ -1009,6 +1000,7 @@ def test_serialize_node__or_then_and(serialize_node):
             ],
             "adornments": None,
             "attributes": [],
+            "outputs": [],
         },
         serialized_node,
         ignore_order=True,

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

@@ -10,8 +10,7 @@ class Inputs(BaseInputs):
 
 
 class BasicGenericNode(BaseNode):
-    class Outputs(BaseNode.Outputs):
-        output = Inputs.input
+    pass
 
 
 def test_serialize_node__basic(serialize_node):
@@ -39,11 +38,13 @@ def test_serialize_node__basic(serialize_node):
             "ports": [
                 {
                     "id": "4fbf0fff-a42e-4410-852a-238b5059198e",
+                    "name": "default",
                     "type": "DEFAULT",
                 }
             ],
             "adornments": None,
             "attributes": [],
+            "outputs": [],
         },
         serialized_node,
         ignore_order=True,
@@ -51,9 +52,6 @@ def test_serialize_node__basic(serialize_node):
 
 
 class AwaitAnyGenericNode(BaseNode):
-    class Outputs(BaseNode.Outputs):
-        output = Inputs.input
-
     class Trigger(BaseNode.Trigger):
         merge_behavior = MergeBehavior.AWAIT_ANY
 
@@ -83,11 +81,13 @@ def test_serialize_node__await_any(serialize_node):
             "ports": [
                 {
                     "id": "3e219c0a-e5f8-443a-ac78-1a458b189009",
+                    "name": "default",
                     "type": "DEFAULT",
                 }
             ],
             "adornments": None,
             "attributes": [],
+            "outputs": [],
         },
         serialized_node,
         ignore_order=True,
@@ -95,9 +95,6 @@ def test_serialize_node__await_any(serialize_node):
 
 
 class AwaitAllGenericNode(BaseNode):
-    class Outputs(BaseNode.Outputs):
-        output = Inputs.input
-
     class Trigger(BaseNode.Trigger):
         merge_behavior = MergeBehavior.AWAIT_ALL
 
@@ -127,11 +124,13 @@ def test_serialize_node__await_all(serialize_node):
             "ports": [
                 {
                     "id": "9797e93f-9fe6-48b2-a48e-476abbd20e32",
+                    "name": "default",
                     "type": "DEFAULT",
                 }
             ],
             "adornments": None,
             "attributes": [],
+            "outputs": [],
         },
         serialized_node,
         ignore_order=True,

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

@@ -141,9 +141,42 @@ def test_serialize_workflow():
                                 "module": ["tests", "workflows", "basic_inline_subworkflow", "workflow"],
                             },
                             "trigger": {"id": "a95a34f2-e894-4fb6-a2c9-15d12c1e3135", "merge_behavior": "AWAIT_ANY"},
-                            "ports": [{"id": "4a62dea3-6a4a-4390-88d8-4af5d2c81474", "type": "DEFAULT"}],
+                            "ports": [
+                                {"id": "4a62dea3-6a4a-4390-88d8-4af5d2c81474", "type": "DEFAULT", "name": "default"}
+                            ],
                             "adornments": None,
-                            "attributes": [],
+                            "attributes": [
+                                {
+                                    "id": "b0ac6b50-22a8-42ba-a707-1aa09a653205",
+                                    "name": "metro",
+                                    "value": {
+                                        "type": "WORKFLOW_INPUT",
+                                        "input_variable_id": "f2f5da15-026d-4905-bfe7-7d16bda20eed",
+                                    },
+                                },
+                                {
+                                    "id": "c5f2d66c-5bb6-4d2a-8e4d-5356318cd3ba",
+                                    "name": "date",
+                                    "value": {
+                                        "type": "WORKFLOW_INPUT",
+                                        "input_variable_id": "aba1e6e0-dfa7-4c15-a4e6-aec6feebfaca",
+                                    },
+                                },
+                            ],
+                            "outputs": [
+                                {
+                                    "id": "3f4c753e-f057-47bb-9748-7968283cc8aa",
+                                    "name": "temperature",
+                                    "type": "NUMBER",
+                                    "value": None,
+                                },
+                                {
+                                    "id": "2a4a62b3-cd26-4d2c-b3f1-eaa5f9dd22dd",
+                                    "name": "reasoning",
+                                    "type": "STRING",
+                                    "value": None,
+                                },
+                            ],
                         },
                         {
                             "id": "a773c3a5-78cb-4250-8d29-7282e8a579d3",

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

@@ -130,9 +130,36 @@ def test_serialize_workflow():
                                 "module": ["tests", "workflows", "basic_map_node", "workflow"],
                             },
                             "trigger": {"id": "01324747-9bc0-4ecd-a8ab-40dca5a94e2e", "merge_behavior": "AWAIT_ANY"},
-                            "ports": [{"id": "36791877-95b0-4390-a794-10accd8f548e", "type": "DEFAULT"}],
+                            "ports": [
+                                {"id": "36791877-95b0-4390-a794-10accd8f548e", "type": "DEFAULT", "name": "default"}
+                            ],
                             "adornments": None,
-                            "attributes": [],
+                            "attributes": [
+                                {
+                                    "id": "4306b0ce-f31c-4c8f-8122-659d40e60ffe",
+                                    "name": "item",
+                                    "value": {
+                                        "type": "WORKFLOW_INPUT",
+                                        "input_variable_id": "b29bb546-9bc8-4136-857d-8c7a464ba9d4",
+                                    },
+                                },
+                                {
+                                    "id": "bca02056-0fd9-44c6-8aa9-743dd1180d48",
+                                    "name": "index",
+                                    "value": {
+                                        "type": "WORKFLOW_INPUT",
+                                        "input_variable_id": "17e7ca49-668f-450d-a792-e1f97d13db67",
+                                    },
+                                },
+                            ],
+                            "outputs": [
+                                {
+                                    "id": "a7bcb362-a2b8-4476-b0de-a361efeec204",
+                                    "name": "count",
+                                    "type": "NUMBER",
+                                    "value": None,
+                                }
+                            ],
                         },
                         {
                             "id": "6f4883b2-70b1-4e1c-ae15-7d0f5aec810b",

vellum_ee/workflows/display/workflows/vellum_workflow_display.py

@@ -265,16 +265,18 @@ class VellumWorkflowDisplay(
         self, workflow_input: WorkflowInputReference, overrides: Optional[WorkflowInputsVellumDisplayOverrides] = None
     ) -> WorkflowInputsVellumDisplay:
         workflow_input_id: UUID
+        name = None
         required = None
         color = None
         if overrides:
             workflow_input_id = overrides.id
+            name = overrides.name
             required = overrides.required
             color = overrides.color
         else:
             workflow_input_id = uuid4_from_hash(f"{self.workflow_id}|inputs|id|{workflow_input.name}")
 
-        return WorkflowInputsVellumDisplay(id=workflow_input_id, required=required, color=color)
+        return WorkflowInputsVellumDisplay(id=workflow_input_id, name=name, required=required, color=color)
 
     def _generate_entrypoint_display(
         self,