vellum-ai 0.14.41__py3-none-any.whl → 1.11.2__py3-none-any.whl

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

@@ -67,7 +67,6 @@ def test_serialize_workflow():
 
     # 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"]) == 3
     assert len(workflow_raw_data["nodes"]) == 4
 
@@ -84,18 +83,18 @@ def test_serialize_workflow():
             "source_handle_id": "c344fdee-282b-40c9-8c97-6dd08830948c",
         },
         "display_data": {
-            "position": {"x": 0.0, "y": 0.0},
+            "position": {"x": 0.0, "y": -50.0},
         },
     }
 
     subworkflow_node = workflow_raw_data["nodes"][1]
     assert not DeepDiff(
         {
-            "id": "080e4343-c7ce-4f82-b9dd-e94c8cc92239",
+            "id": "cc246984-82bb-4da3-b447-ea66aa1cffe4",
             "type": "SUBWORKFLOW",
             "inputs": [
                 {
-                    "id": "704c4640-bfda-44f0-8da3-e9cfc4f21cf2",
+                    "id": "9ae1bf55-2024-48e4-b7fa-29c189545dc8",
                     "key": "metro",
                     "value": {
                         "rules": [
@@ -111,8 +110,8 @@ def test_serialize_workflow():
             "data": {
                 "label": "Example Inline Subworkflow Node",
                 "error_output_id": None,
-                "source_handle_id": "cfd831bc-ee7f-44d0-8d76-0ba0cd0277dc",
-                "target_handle_id": "859a75a6-1bd2-4350-9509-4af66245e8e4",
+                "source_handle_id": "4e6146c9-05f1-4494-b48c-7ff02224cf5a",
+                "target_handle_id": "955e83a5-6a19-4dfa-ab19-9a6fd63d017d",
                 "variant": "INLINE",
                 "workflow_raw_data": {
                     "nodes": [
@@ -124,31 +123,32 @@ def test_serialize_workflow():
                                 "label": "Entrypoint Node",
                                 "source_handle_id": "9914a6a0-9a99-430d-8ddd-f7c13847fe1a",
                             },
-                            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+                            "display_data": {"position": {"x": 0.0, "y": -50.0}},
                             "base": None,
                             "definition": None,
                         },
                         {
-                            "id": "1381c078-efa2-4255-89a1-7b4cb742c7fc",
-                            "label": "StartNode",
+                            "id": "45ba1e29-611f-4a6f-9f51-7cf6fe47141e",
+                            "label": "Start Node",
                             "type": "GENERIC",
-                            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+                            "should_file_merge": True,
+                            "display_data": {"position": {"x": 200.0, "y": -50.0}},
                             "base": {"name": "BaseNode", "module": ["vellum", "workflows", "nodes", "bases", "base"]},
                             "definition": {
                                 "name": "StartNode",
                                 "module": ["tests", "workflows", "basic_inline_subworkflow", "workflow"],
                             },
                             "trigger": {
-                                "id": "6492efcf-4437-4af1-9ad7-269795ccb27a",
+                                "id": "ab273dfa-6719-427c-9054-dd2c03d90d5d",
                                 "merge_behavior": "AWAIT_ATTRIBUTES",
                             },
                             "ports": [
-                                {"id": "1e739e86-a285-4438-9725-a152c15a63e3", "name": "default", "type": "DEFAULT"}
+                                {"id": "37a41308-2f6c-443d-a832-5859cef231e8", "name": "default", "type": "DEFAULT"}
                             ],
                             "adornments": None,
                             "attributes": [
                                 {
-                                    "id": "b0ac6b50-22a8-42ba-a707-1aa09a653205",
+                                    "id": "47970f44-eafc-4526-bd81-b1419b0a787a",
                                     "name": "metro",
                                     "value": {
                                         "type": "WORKFLOW_INPUT",
@@ -156,7 +156,7 @@ def test_serialize_workflow():
                                     },
                                 },
                                 {
-                                    "id": "c5f2d66c-5bb6-4d2a-8e4d-5356318cd3ba",
+                                    "id": "6d933d5e-84a2-434a-baf8-25315c9469af",
                                     "name": "date",
                                     "value": {
                                         "type": "WORKFLOW_INPUT",
@@ -166,13 +166,13 @@ def test_serialize_workflow():
                             ],
                             "outputs": [
                                 {
-                                    "id": "3f4c753e-f057-47bb-9748-7968283cc8aa",
+                                    "id": "1162f98c-fc51-49c3-a634-120ccbf2ddc2",
                                     "name": "temperature",
                                     "type": "NUMBER",
                                     "value": None,
                                 },
                                 {
-                                    "id": "2a4a62b3-cd26-4d2c-b3f1-eaa5f9dd22dd",
+                                    "id": "4d13651c-3c74-4bf4-8595-2339de83c59a",
                                     "name": "reasoning",
                                     "type": "STRING",
                                     "value": None,
@@ -199,8 +199,8 @@ def test_serialize_workflow():
                                             {
                                                 "type": "NODE_OUTPUT",
                                                 "data": {
-                                                    "node_id": "1381c078-efa2-4255-89a1-7b4cb742c7fc",
-                                                    "output_id": "3f4c753e-f057-47bb-9748-7968283cc8aa",
+                                                    "node_id": "45ba1e29-611f-4a6f-9f51-7cf6fe47141e",
+                                                    "output_id": "1162f98c-fc51-49c3-a634-120ccbf2ddc2",
                                                 },
                                             }
                                         ],
@@ -208,7 +208,7 @@ def test_serialize_workflow():
                                     },
                                 }
                             ],
-                            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+                            "display_data": {"position": {"x": 400.0, "y": 75.0}},
                             "base": {
                                 "name": "FinalOutputNode",
                                 "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
@@ -235,8 +235,8 @@ def test_serialize_workflow():
                                             {
                                                 "type": "NODE_OUTPUT",
                                                 "data": {
-                                                    "node_id": "1381c078-efa2-4255-89a1-7b4cb742c7fc",
-                                                    "output_id": "2a4a62b3-cd26-4d2c-b3f1-eaa5f9dd22dd",
+                                                    "node_id": "45ba1e29-611f-4a6f-9f51-7cf6fe47141e",
+                                                    "output_id": "4d13651c-3c74-4bf4-8595-2339de83c59a",
                                                 },
                                             }
                                         ],
@@ -244,7 +244,7 @@ def test_serialize_workflow():
                                     },
                                 }
                             ],
-                            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+                            "display_data": {"position": {"x": 400.0, "y": -175.0}},
                             "base": {
                                 "name": "FinalOutputNode",
                                 "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
@@ -254,25 +254,25 @@ def test_serialize_workflow():
                     ],
                     "edges": [
                         {
-                            "id": "fb2f58f0-9d49-4658-af78-afa9b94091a6",
+                            "id": "85ac167b-dc92-4197-b364-64e630cab237",
                             "source_node_id": "afa49a0f-db35-4552-9217-5b8f237e84bc",
                             "source_handle_id": "9914a6a0-9a99-430d-8ddd-f7c13847fe1a",
-                            "target_node_id": "1381c078-efa2-4255-89a1-7b4cb742c7fc",
-                            "target_handle_id": "6492efcf-4437-4af1-9ad7-269795ccb27a",
+                            "target_node_id": "45ba1e29-611f-4a6f-9f51-7cf6fe47141e",
+                            "target_handle_id": "ab273dfa-6719-427c-9054-dd2c03d90d5d",
                             "type": "DEFAULT",
                         },
                         {
                             "id": "6f16dfb8-d794-4be8-8860-6ea34f0b9e7c",
-                            "source_node_id": "1381c078-efa2-4255-89a1-7b4cb742c7fc",
-                            "source_handle_id": "1e739e86-a285-4438-9725-a152c15a63e3",
+                            "source_node_id": "45ba1e29-611f-4a6f-9f51-7cf6fe47141e",
+                            "source_handle_id": "37a41308-2f6c-443d-a832-5859cef231e8",
                             "target_node_id": "a773c3a5-78cb-4250-8d29-7282e8a579d3",
                             "target_handle_id": "804bb543-9cf4-457f-acf1-fb4b8b7d9259",
                             "type": "DEFAULT",
                         },
                         {
                             "id": "63b77ff0-5282-46ce-8da9-37ced05ac61c",
-                            "source_node_id": "1381c078-efa2-4255-89a1-7b4cb742c7fc",
-                            "source_handle_id": "1e739e86-a285-4438-9725-a152c15a63e3",
+                            "source_node_id": "45ba1e29-611f-4a6f-9f51-7cf6fe47141e",
+                            "source_handle_id": "37a41308-2f6c-443d-a832-5859cef231e8",
                             "target_node_id": "570f4d12-69ff-49f1-ba98-ade6283dd7c2",
                             "target_handle_id": "6d4c4a14-c388-4c7a-b223-eb39baf5c080",
                             "type": "DEFAULT",
@@ -283,14 +283,40 @@ def test_serialize_workflow():
                         "name": "NestedWorkflow",
                         "module": ["tests", "workflows", "basic_inline_subworkflow", "workflow"],
                     },
+                    "output_values": [
+                        {
+                            "output_variable_id": "2fc57139-7420-49e5-96a6-dcbb3ff5d622",
+                            "value": {
+                                "type": "NODE_OUTPUT",
+                                "node_id": "45ba1e29-611f-4a6f-9f51-7cf6fe47141e",
+                                "node_output_id": "1162f98c-fc51-49c3-a634-120ccbf2ddc2",
+                            },
+                        },
+                        {
+                            "output_variable_id": "fad5dd9f-3328-4e70-ad55-65a5325a4a82",
+                            "value": {
+                                "type": "NODE_OUTPUT",
+                                "node_id": "45ba1e29-611f-4a6f-9f51-7cf6fe47141e",
+                                "node_output_id": "4d13651c-3c74-4bf4-8595-2339de83c59a",
+                            },
+                        },
+                    ],
                 },
-                "input_variables": [{"id": "704c4640-bfda-44f0-8da3-e9cfc4f21cf2", "key": "metro", "type": "STRING"}],
+                "input_variables": [
+                    {
+                        "id": "9ae1bf55-2024-48e4-b7fa-29c189545dc8",
+                        "key": "metro",
+                        "type": "STRING",
+                        "required": True,
+                        "default": None,
+                    }
+                ],
                 "output_variables": [
                     {"id": "2fc57139-7420-49e5-96a6-dcbb3ff5d622", "key": "temperature", "type": "NUMBER"},
                     {"id": "fad5dd9f-3328-4e70-ad55-65a5325a4a82", "key": "reasoning", "type": "STRING"},
                 ],
             },
-            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "display_data": {"position": {"x": 200.0, "y": -50.0}},
             "base": {
                 "name": "InlineSubworkflowNode",
                 "module": ["vellum", "workflows", "nodes", "core", "inline_subworkflow_node", "node"],
@@ -299,9 +325,13 @@ def test_serialize_workflow():
                 "name": "ExampleInlineSubworkflowNode",
                 "module": ["tests", "workflows", "basic_inline_subworkflow", "workflow"],
             },
+            "trigger": {
+                "id": "955e83a5-6a19-4dfa-ab19-9a6fd63d017d",
+                "merge_behavior": "AWAIT_ATTRIBUTES",
+            },
+            "ports": [{"id": "4e6146c9-05f1-4494-b48c-7ff02224cf5a", "name": "default", "type": "DEFAULT"}],
         },
         subworkflow_node,
-        ignore_order=True,
     )
 
     temperature_terminal_node = next(
@@ -332,7 +362,7 @@ def test_serialize_workflow():
                             {
                                 "type": "NODE_OUTPUT",
                                 "data": {
-                                    "node_id": "080e4343-c7ce-4f82-b9dd-e94c8cc92239",
+                                    "node_id": "cc246984-82bb-4da3-b447-ea66aa1cffe4",
                                     "output_id": "0a7192da-5576-4933-bba4-de8adf5d7996",
                                 },
                             }
@@ -341,7 +371,7 @@ def test_serialize_workflow():
                     },
                 }
             ],
-            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "display_data": {"position": {"x": 400.0, "y": 75.0}},
             "base": {
                 "name": "FinalOutputNode",
                 "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
@@ -376,7 +406,7 @@ def test_serialize_workflow():
                             {
                                 "type": "NODE_OUTPUT",
                                 "data": {
-                                    "node_id": "080e4343-c7ce-4f82-b9dd-e94c8cc92239",
+                                    "node_id": "cc246984-82bb-4da3-b447-ea66aa1cffe4",
                                     "output_id": "86dd0202-c141-48a3-8382-2da60372e77c",
                                 },
                             }
@@ -385,7 +415,7 @@ def test_serialize_workflow():
                     },
                 }
             ],
-            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "display_data": {"position": {"x": 400.0, "y": -175.0}},
             "base": {
                 "name": "FinalOutputNode",
                 "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
@@ -404,25 +434,25 @@ def test_serialize_workflow():
     assert not DeepDiff(
         [
             {
-                "id": "71dd3569-ccf8-4352-ad42-3594be3a6c16",
+                "id": "4bfa5508-f7bc-4b8c-b468-f6f830caa660",
                 "source_node_id": "6358dcfe-b162-4e19-99ca-401d1ada9bdc",
                 "source_handle_id": "c344fdee-282b-40c9-8c97-6dd08830948c",
-                "target_node_id": "080e4343-c7ce-4f82-b9dd-e94c8cc92239",
-                "target_handle_id": "859a75a6-1bd2-4350-9509-4af66245e8e4",
+                "target_node_id": "cc246984-82bb-4da3-b447-ea66aa1cffe4",
+                "target_handle_id": "955e83a5-6a19-4dfa-ab19-9a6fd63d017d",
                 "type": "DEFAULT",
             },
             {
                 "id": "3c5d8990-48f5-42e1-893e-bc8308d2110a",
-                "source_node_id": "080e4343-c7ce-4f82-b9dd-e94c8cc92239",
-                "source_handle_id": "cfd831bc-ee7f-44d0-8d76-0ba0cd0277dc",
+                "source_node_id": "cc246984-82bb-4da3-b447-ea66aa1cffe4",
+                "source_handle_id": "4e6146c9-05f1-4494-b48c-7ff02224cf5a",
                 "target_node_id": "0779b232-82ab-4dbe-a340-6a85e6ab3368",
                 "target_handle_id": "9e077063-c394-4c7b-b0c6-e6686df67984",
                 "type": "DEFAULT",
             },
             {
                 "id": "de0b8090-a26e-4e09-9173-9f7400a5be4c",
-                "source_node_id": "080e4343-c7ce-4f82-b9dd-e94c8cc92239",
-                "source_handle_id": "cfd831bc-ee7f-44d0-8d76-0ba0cd0277dc",
+                "source_node_id": "cc246984-82bb-4da3-b447-ea66aa1cffe4",
+                "source_handle_id": "4e6146c9-05f1-4494-b48c-7ff02224cf5a",
                 "target_node_id": "31b74695-3f1c-47cf-8be8-a4d86cc589e8",
                 "target_handle_id": "8b525943-6c27-414b-a329-e29c0b217f72",
                 "type": "DEFAULT",

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

@@ -54,7 +54,6 @@ def test_serialize_workflow():
 
     # 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
 
@@ -71,18 +70,18 @@ def test_serialize_workflow():
         "base": None,
         "definition": None,
         "display_data": {
-            "position": {"x": 0.0, "y": 0.0},
+            "position": {"x": 0.0, "y": -50.0},
         },
     }
 
     map_node = workflow_raw_data["nodes"][1]
     assert not DeepDiff(
         {
-            "id": "bf83099a-40df-4445-b90d-1f6f1067ebe3",
+            "id": "f2f94af1-fcbe-497c-80ce-80952c8903c8",
             "type": "MAP",
             "inputs": [
                 {
-                    "id": "4c0a109f-599e-4d04-a396-a51474fc2996",
+                    "id": "0acef1e7-caa1-4d9f-bb69-6981729af18d",
                     "key": "items",
                     "value": {
                         "rules": [
@@ -98,8 +97,8 @@ def test_serialize_workflow():
             "data": {
                 "label": "Map Fruits Node",
                 "error_output_id": None,
-                "source_handle_id": "a2171a61-0657-43ad-b6d9-cf93ce3270d0",
-                "target_handle_id": "b5e8182e-20c5-482b-b4c5-4dde48c01472",
+                "source_handle_id": "aff8a80e-7ce7-43d2-9c9e-9d137efd3b33",
+                "target_handle_id": "e16e9d55-5f26-4d89-8c7a-939f1f463d80",
                 "variant": "INLINE",
                 "workflow_raw_data": {
                     "nodes": [
@@ -111,15 +110,16 @@ def test_serialize_workflow():
                                 "label": "Entrypoint Node",
                                 "source_handle_id": "520d3616-8369-4e79-9da5-3febae299c2a",
                             },
-                            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+                            "display_data": {"position": {"x": 0.0, "y": -50.0}},
                             "base": None,
                             "definition": None,
                         },
                         {
-                            "id": "baf6d316-dc75-41e8-96c0-015aede96309",
+                            "id": "9dffa9e5-00e3-4d20-8ab8-ad0412f35bf2",
                             "label": "Iteration",
                             "type": "GENERIC",
-                            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+                            "should_file_merge": True,
+                            "display_data": {"position": {"x": 200.0, "y": -50.0}},
                             "base": {
                                 "name": "BaseNode",
                                 "module": ["vellum", "workflows", "nodes", "bases", "base"],
@@ -129,16 +129,16 @@ def test_serialize_workflow():
                                 "module": ["tests", "workflows", "basic_map_node", "workflow"],
                             },
                             "trigger": {
-                                "id": "551d5528-f4e1-42ea-bde0-9de4b4968253",
+                                "id": "40631b74-8a35-4cb1-897d-3c93f8d5b625",
                                 "merge_behavior": "AWAIT_ATTRIBUTES",
                             },
                             "ports": [
-                                {"id": "71ada606-d791-4a59-a252-0795c5faeeaf", "type": "DEFAULT", "name": "default"}
+                                {"id": "8d5d36aa-f7c2-49bf-b4d5-cda6f6cd18f9", "type": "DEFAULT", "name": "default"}
                             ],
                             "adornments": None,
                             "attributes": [
                                 {
-                                    "id": "4306b0ce-f31c-4c8f-8122-659d40e60ffe",
+                                    "id": "52a43441-9162-40b1-96c6-445250af1c34",
                                     "name": "item",
                                     "value": {
                                         "type": "WORKFLOW_INPUT",
@@ -146,7 +146,7 @@ def test_serialize_workflow():
                                     },
                                 },
                                 {
-                                    "id": "bca02056-0fd9-44c6-8aa9-743dd1180d48",
+                                    "id": "dadea752-95db-4e5f-a0d4-f3920376ee8f",
                                     "name": "index",
                                     "value": {
                                         "type": "WORKFLOW_INPUT",
@@ -156,7 +156,7 @@ def test_serialize_workflow():
                             ],
                             "outputs": [
                                 {
-                                    "id": "a7bcb362-a2b8-4476-b0de-a361efeec204",
+                                    "id": "3d97725d-a423-4072-9a73-7b6517edc76e",
                                     "name": "count",
                                     "type": "NUMBER",
                                     "value": None,
@@ -183,8 +183,8 @@ def test_serialize_workflow():
                                             {
                                                 "type": "NODE_OUTPUT",
                                                 "data": {
-                                                    "node_id": "baf6d316-dc75-41e8-96c0-015aede96309",
-                                                    "output_id": "a7bcb362-a2b8-4476-b0de-a361efeec204",
+                                                    "node_id": "9dffa9e5-00e3-4d20-8ab8-ad0412f35bf2",
+                                                    "output_id": "3d97725d-a423-4072-9a73-7b6517edc76e",
                                                 },
                                             }
                                         ],
@@ -192,7 +192,7 @@ def test_serialize_workflow():
                                     },
                                 }
                             ],
-                            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+                            "display_data": {"position": {"x": 400.0, "y": -50.0}},
                             "base": {
                                 "name": "FinalOutputNode",
                                 "module": [
@@ -209,17 +209,17 @@ def test_serialize_workflow():
                     ],
                     "edges": [
                         {
-                            "id": "b59c050b-0f6a-4153-ab58-fa051222fa05",
+                            "id": "8b52ffcb-1ccf-41eb-ad89-705c6abc2369",
                             "source_node_id": "ff9bfe6e-839d-4d40-b8fc-313b3bbd0ab0",
                             "source_handle_id": "520d3616-8369-4e79-9da5-3febae299c2a",
-                            "target_node_id": "baf6d316-dc75-41e8-96c0-015aede96309",
-                            "target_handle_id": "551d5528-f4e1-42ea-bde0-9de4b4968253",
+                            "target_node_id": "9dffa9e5-00e3-4d20-8ab8-ad0412f35bf2",
+                            "target_handle_id": "40631b74-8a35-4cb1-897d-3c93f8d5b625",
                             "type": "DEFAULT",
                         },
                         {
                             "id": "14152688-6996-4d64-9231-a6e66a4827eb",
-                            "source_node_id": "baf6d316-dc75-41e8-96c0-015aede96309",
-                            "source_handle_id": "71ada606-d791-4a59-a252-0795c5faeeaf",
+                            "source_node_id": "9dffa9e5-00e3-4d20-8ab8-ad0412f35bf2",
+                            "source_handle_id": "8d5d36aa-f7c2-49bf-b4d5-cda6f6cd18f9",
                             "target_node_id": "6f4883b2-70b1-4e1c-ae15-7d0f5aec810b",
                             "target_handle_id": "9d74571f-b7f5-4c1d-8b7c-b9c648738a4d",
                             "type": "DEFAULT",
@@ -235,6 +235,16 @@ def test_serialize_workflow():
                             "workflow",
                         ],
                     },
+                    "output_values": [
+                        {
+                            "output_variable_id": "2a957315-fae0-4366-8a35-f0b315c5eade",
+                            "value": {
+                                "type": "NODE_OUTPUT",
+                                "node_id": "9dffa9e5-00e3-4d20-8ab8-ad0412f35bf2",
+                                "node_output_id": "3d97725d-a423-4072-9a73-7b6517edc76e",
+                            },
+                        }
+                    ],
                 },
                 "input_variables": [
                     {
@@ -254,7 +264,7 @@ def test_serialize_workflow():
                         "extensions": {"color": None},
                     },
                     {
-                        "id": "d6fc6c7a-235f-4b98-86f3-e258d1198f93",
+                        "id": "0acef1e7-caa1-4d9f-bb69-6981729af18d",
                         "key": "items",
                         "type": "JSON",
                         "required": True,
@@ -264,11 +274,11 @@ def test_serialize_workflow():
                 ],
                 "output_variables": [{"id": "2a957315-fae0-4366-8a35-f0b315c5eade", "key": "count", "type": "NUMBER"}],
                 "concurrency": None,
-                "items_input_id": "d6fc6c7a-235f-4b98-86f3-e258d1198f93",
+                "items_input_id": "0acef1e7-caa1-4d9f-bb69-6981729af18d",
                 "item_input_id": "b29bb546-9bc8-4136-857d-8c7a464ba9d4",
                 "index_input_id": "17e7ca49-668f-450d-a792-e1f97d13db67",
             },
-            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "display_data": {"position": {"x": 200.0, "y": -50.0}},
             "base": {
                 "name": "MapNode",
                 "module": ["vellum", "workflows", "nodes", "core", "map_node", "node"],
@@ -277,6 +287,11 @@ def test_serialize_workflow():
                 "name": "MapFruitsNode",
                 "module": ["tests", "workflows", "basic_map_node", "workflow"],
             },
+            "trigger": {
+                "id": "e16e9d55-5f26-4d89-8c7a-939f1f463d80",
+                "merge_behavior": "AWAIT_ATTRIBUTES",
+            },
+            "ports": [{"id": "aff8a80e-7ce7-43d2-9c9e-9d137efd3b33", "name": "default", "type": "DEFAULT"}],
         },
         map_node,
         ignore_order=True,
@@ -303,7 +318,7 @@ def test_serialize_workflow():
                             {
                                 "type": "NODE_OUTPUT",
                                 "data": {
-                                    "node_id": "bf83099a-40df-4445-b90d-1f6f1067ebe3",
+                                    "node_id": "f2f94af1-fcbe-497c-80ce-80952c8903c8",
                                     "output_id": "2a957315-fae0-4366-8a35-f0b315c5eade",
                                 },
                             }
@@ -312,7 +327,7 @@ def test_serialize_workflow():
                     },
                 }
             ],
-            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "display_data": {"position": {"x": 400.0, "y": -50.0}},
             "base": {
                 "module": [
                     "vellum",
@@ -337,17 +352,17 @@ def test_serialize_workflow():
     assert not DeepDiff(
         [
             {
-                "id": "f39477f3-445a-412e-a8ab-371baa8ae990",
+                "id": "528eb20a-9db4-4c01-87c0-39b9f5f09753",
                 "source_node_id": "c0aa464d-1685-4f15-a051-31b426fec92e",
                 "source_handle_id": "844d992e-60ab-4af2-a8ff-52cd858386f7",
-                "target_node_id": "bf83099a-40df-4445-b90d-1f6f1067ebe3",
-                "target_handle_id": "b5e8182e-20c5-482b-b4c5-4dde48c01472",
+                "target_node_id": "f2f94af1-fcbe-497c-80ce-80952c8903c8",
+                "target_handle_id": "e16e9d55-5f26-4d89-8c7a-939f1f463d80",
                 "type": "DEFAULT",
             },
             {
                 "id": "47a34f6e-d139-4702-aa46-6212bb8a150f",
-                "source_node_id": "bf83099a-40df-4445-b90d-1f6f1067ebe3",
-                "source_handle_id": "a2171a61-0657-43ad-b6d9-cf93ce3270d0",
+                "source_node_id": "f2f94af1-fcbe-497c-80ce-80952c8903c8",
+                "source_handle_id": "aff8a80e-7ce7-43d2-9c9e-9d137efd3b33",
                 "target_node_id": "bacc5d55-07d4-4a0a-a69e-831524480de5",
                 "target_handle_id": "720dd872-2f3d-47b9-8245-89387f04f300",
                 "type": "DEFAULT",
@@ -378,3 +393,7 @@ def test_serialize_workflow():
             "workflow",
         ],
     }
+
+    # AND the map node's items input ID should match the subworkflow's items input ID
+    items_input_id = map_node["data"]["items_input_id"]
+    assert map_node["inputs"][0]["id"] == items_input_id