vellum-ai 0.14.72__py3-none-any.whl → 0.14.74__py3-none-any.whl

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

@@ -63,7 +63,7 @@ def test_serialize_workflow():
         "type": "ENTRYPOINT",
         "inputs": [],
         "data": {"label": "Entrypoint Node", "source_handle_id": "8294baa6-8bf4-4b54-a56b-407b64851b77"},
-        "display_data": {"position": {"x": 0.0, "y": 0.0}},
+        "display_data": {"position": {"x": 0.0, "y": -50.0}},
         "base": None,
         "definition": None,
     }
@@ -153,7 +153,7 @@ def test_serialize_workflow():
                 },
                 "ml_model_name": "gpt-4o",
             },
-            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "display_data": {"position": {"x": 200.0, "y": -50.0}},
             "base": {
                 "name": "InlinePromptNode",
                 "module": ["vellum", "workflows", "nodes", "displayable", "inline_prompt_node", "node"],
@@ -240,6 +240,27 @@ def test_serialize_workflow():
                         },
                     },
                 },
+                {
+                    "id": "2b98319f-f43d-42d9-a8b0-b148d5de0a2c",
+                    "name": "parameters",
+                    "value": {
+                        "type": "CONSTANT_VALUE",
+                        "value": {
+                            "type": "JSON",
+                            "value": {
+                                "stop": [],
+                                "temperature": 0.0,
+                                "max_tokens": 4096,
+                                "top_p": 1.0,
+                                "top_k": 0,
+                                "frequency_penalty": 0.0,
+                                "presence_penalty": 0.0,
+                                "logit_bias": None,
+                                "custom_parameters": None,
+                            },
+                        },
+                    },
+                },
             ],
         },
         prompt_node,
@@ -277,7 +298,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": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
@@ -437,3 +458,99 @@ def test_serialize_workflow_with_descriptor_blocks():
             },
         }
     ]
+
+
+def test_serialize_workflow_with_nested_descriptor_blocks():
+    """Test that serialization handles BaseDescriptor instances nested in ChatMessageBlock.blocks."""
+
+    class TestInputs(BaseInputs):
+        noun: str
+
+    class UpstreamNode(BaseNode):
+        class Outputs(BaseNode.Outputs):
+            results: list
+
+        def run(self) -> Outputs:
+            return self.Outputs(results=["test"])
+
+    chat_block = ChatMessagePromptBlock(chat_role="SYSTEM", blocks=[JinjaPromptBlock(template="Hello")])
+
+    class TestInlinePromptNodeWithNestedDescriptorBlocks(InlinePromptNode):
+        ml_model = "gpt-4o"
+        blocks = [chat_block]
+        prompt_inputs = {"noun": TestInputs.noun}
+
+    object.__setattr__(chat_block, "blocks", [UpstreamNode.Outputs.results[0]])
+
+    class TestWorkflow(BaseWorkflow[TestInputs, BaseState]):
+        graph = UpstreamNode >> TestInlinePromptNodeWithNestedDescriptorBlocks
+
+    workflow_display = get_workflow_display(workflow_class=TestWorkflow)
+    serialized: dict = workflow_display.serialize()
+
+    prompt_nodes = [node for node in serialized["workflow_raw_data"]["nodes"] if node["type"] == "PROMPT"]
+    prompt_node = prompt_nodes[0]
+
+    blocks = prompt_node["data"]["exec_config"]["prompt_template_block_data"]["blocks"]
+    descriptor_blocks = [block for block in blocks if not isinstance(block, dict) or not block.get("block_type")]
+    assert len(descriptor_blocks) == 0, "BaseDescriptor blocks should not appear in serialized blocks"
+
+    blocks_attr = next((attr for attr in prompt_node["attributes"] if attr["name"] == "blocks"), None)
+    assert blocks_attr is not None, "blocks attribute should be present when blocks contain nested BaseDescriptor"
+    assert blocks_attr["value"]["type"] == "ARRAY_REFERENCE", "blocks attribute should be serialized as ARRAY_REFERENCE"
+    assert blocks_attr["value"]["items"] == [
+        {
+            "entries": [
+                {
+                    "id": "24a203be-3cba-4b20-bc84-9993a476c120",
+                    "key": "block_type",
+                    "value": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": "CHAT_MESSAGE"}},
+                },
+                {
+                    "id": "c06269e6-f74c-4860-8fa5-22dcbdc89399",
+                    "key": "state",
+                    "value": {"type": "CONSTANT_VALUE", "value": {"type": "JSON", "value": None}},
+                },
+                {
+                    "id": "dd9c0d43-b931-4dc8-8b3a-a7507ddff0c1",
+                    "key": "cache_config",
+                    "value": {"type": "CONSTANT_VALUE", "value": {"type": "JSON", "value": None}},
+                },
+                {
+                    "id": "bef22f2b-0b6e-4910-88cc-6df736d2e20e",
+                    "key": "chat_role",
+                    "value": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": "SYSTEM"}},
+                },
+                {
+                    "id": "c0beec30-f85e-4a78-a3fb-baee54a692f8",
+                    "key": "chat_source",
+                    "value": {"type": "CONSTANT_VALUE", "value": {"type": "JSON", "value": None}},
+                },
+                {
+                    "id": "f601f4f2-62fe-4697-9fe0-99ca8aa64500",
+                    "key": "chat_message_unterminated",
+                    "value": {"type": "CONSTANT_VALUE", "value": {"type": "JSON", "value": None}},
+                },
+                {
+                    "id": "ad550008-64e3-44a3-a32a-84ec226db31c",
+                    "key": "blocks",
+                    "value": {
+                        "items": [
+                            {
+                                "lhs": {
+                                    "node_id": "9fe5d3a3-7d26-4692-aa2d-e67c673b0c2b",
+                                    "node_output_id": "92f9a1b7-d33b-4f00-b4c2-e6f58150e166",
+                                    "type": "NODE_OUTPUT",
+                                },
+                                "operator": "accessField",
+                                "rhs": {"type": "CONSTANT_VALUE", "value": {"type": "NUMBER", "value": 0.0}},
+                                "type": "BINARY_EXPRESSION",
+                            }
+                        ],
+                        "type": "ARRAY_REFERENCE",
+                    },
+                },
+            ],
+            "type": "DICTIONARY_REFERENCE",
+        }
+    ]

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

@@ -83,7 +83,7 @@ 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},
         },
     }
 
@@ -123,7 +123,7 @@ 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,
                         },
@@ -131,7 +131,7 @@ def test_serialize_workflow():
                             "id": "1381c078-efa2-4255-89a1-7b4cb742c7fc",
                             "label": "StartNode",
                             "type": "GENERIC",
-                            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+                            "display_data": {"position": {"x": 200.0, "y": -50.0}},
                             "base": {"name": "BaseNode", "module": ["vellum", "workflows", "nodes", "bases", "base"]},
                             "definition": {
                                 "name": "StartNode",
@@ -207,7 +207,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"],
@@ -243,7 +243,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"],
@@ -315,7 +315,7 @@ def test_serialize_workflow():
                     {"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"],
@@ -366,7 +366,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"],
@@ -410,7 +410,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"],

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

@@ -70,7 +70,7 @@ def test_serialize_workflow():
         "base": None,
         "definition": None,
         "display_data": {
-            "position": {"x": 0.0, "y": 0.0},
+            "position": {"x": 0.0, "y": -50.0},
         },
     }
 
@@ -110,7 +110,7 @@ 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,
                         },
@@ -118,7 +118,7 @@ def test_serialize_workflow():
                             "id": "baf6d316-dc75-41e8-96c0-015aede96309",
                             "label": "Iteration",
                             "type": "GENERIC",
-                            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+                            "display_data": {"position": {"x": 200.0, "y": -50.0}},
                             "base": {
                                 "name": "BaseNode",
                                 "module": ["vellum", "workflows", "nodes", "bases", "base"],
@@ -191,7 +191,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": [
@@ -277,7 +277,7 @@ def test_serialize_workflow():
                 "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"],
@@ -322,7 +322,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",

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

@@ -52,7 +52,7 @@ def test_serialize_workflow__await_all():
             "source_handle_id": "017d40f5-8326-4e42-a409-b08995defaa8",
         },
         "display_data": {
-            "position": {"x": 0.0, "y": 0.0},
+            "position": {"x": 0.0, "y": -50.0},
         },
     }
 
@@ -74,7 +74,7 @@ def test_serialize_workflow__await_all():
                 ],
                 "source_handle_id": "3bbc469f-0fb0-4b3d-a28b-746fefec2818",
             },
-            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "display_data": {"position": {"x": 400.0, "y": -50.0}},
             "base": {
                 "module": ["vellum", "workflows", "nodes", "displayable", "merge_node", "node"],
                 "name": "MergeNode",
@@ -131,7 +131,7 @@ def test_serialize_workflow__await_all():
                 },
             }
         ],
-        "display_data": {"position": {"x": 0.0, "y": 0.0}},
+        "display_data": {"position": {"x": 800.0, "y": -50.0}},
     }
 
     # AND each edge should be serialized correctly

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

@@ -93,7 +93,7 @@ def test_serialize_workflow(vellum_client):
             "source_handle_id": "7065a943-1cab-4afd-9690-e678c5b74a2f",
         },
         "display_data": {
-            "position": {"x": 0.0, "y": 0.0},
+            "position": {"x": 0.0, "y": -50.0},
         },
     }
 
@@ -141,7 +141,7 @@ def test_serialize_workflow(vellum_client):
             "release_tag": "LATEST",
             "ml_model_fallbacks": ["gpt-4o", "gemini-1.5-pro"],
         },
-        "display_data": {"position": {"x": 0.0, "y": 0.0}},
+        "display_data": {"position": {"x": 200.0, "y": -50.0}},
         "base": {
             "name": "PromptDeploymentNode",
             "module": ["vellum", "workflows", "nodes", "displayable", "prompt_deployment_node", "node"],
@@ -202,8 +202,8 @@ def test_serialize_workflow(vellum_client):
         },
         "display_data": {
             "position": {
-                "x": 0.0,
-                "y": 0.0,
+                "x": 400.0,
+                "y": -50.0,
             },
         },
     }
@@ -321,7 +321,7 @@ def test_serialize_workflow_with_prompt_and_templating(vellum_client):
         "type": "ENTRYPOINT",
         "inputs": [],
         "data": {"label": "Entrypoint Node", "source_handle_id": "cc0f4028-1039-4063-971d-7dacbb01b379"},
-        "display_data": {"position": {"x": 0.0, "y": 0.0}},
+        "display_data": {"position": {"x": 0.0, "y": -50.0}},
         "base": None,
         "definition": None,
     }
@@ -370,7 +370,7 @@ def test_serialize_workflow_with_prompt_and_templating(vellum_client):
             "release_tag": "LATEST",
             "ml_model_fallbacks": None,
         },
-        "display_data": {"position": {"x": 0.0, "y": 0.0}},
+        "display_data": {"position": {"x": 200.0, "y": -50.0}},
         "base": {
             "name": "PromptDeploymentNode",
             "module": ["vellum", "workflows", "nodes", "displayable", "prompt_deployment_node", "node"],
@@ -465,7 +465,7 @@ def test_serialize_workflow_with_prompt_and_templating(vellum_client):
             "template_node_input_id": "7c775379-d589-4d79-b876-dcd224d72966",
             "output_type": "STRING",
         },
-        "display_data": {"position": {"x": 0.0, "y": 0.0}},
+        "display_data": {"position": {"x": 400.0, "y": -50.0}},
         "base": {
             "name": "TemplatingNode",
             "module": ["vellum", "workflows", "nodes", "core", "templating_node", "node"],
@@ -512,7 +512,7 @@ def test_serialize_workflow_with_prompt_and_templating(vellum_client):
                 },
             }
         ],
-        "display_data": {"position": {"x": 0.0, "y": 0.0}},
+        "display_data": {"position": {"x": 600.0, "y": -50.0}},
         "base": {
             "name": "FinalOutputNode",
             "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],

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

@@ -50,7 +50,7 @@ def test_serialize_workflow():
         "type": "ENTRYPOINT",
         "inputs": [],
         "data": {"label": "Entrypoint Node", "source_handle_id": "df80b4aa-2ba1-49a2-8375-fb1f78eee31f"},
-        "display_data": {"position": {"x": 0.0, "y": 0.0}},
+        "display_data": {"position": {"x": 0.0, "y": -50.0}},
         "base": None,
         "definition": None,
     }
@@ -229,7 +229,7 @@ def test_serialize_workflow():
             "external_id_filters_node_input_id": "036dee8f-194a-4b92-9739-69c98a4aa1b9",
             "metadata_filters_node_input_id": "855d3f57-e633-467e-a348-a394360247df",
         },
-        "display_data": {"position": {"x": 0.0, "y": 0.0}},
+        "display_data": {"position": {"x": 200.0, "y": -50.0}},
         "base": {
             "name": "SearchNode",
             "module": ["vellum", "workflows", "nodes", "displayable", "search_node", "node"],
@@ -271,7 +271,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": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],

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

@@ -99,7 +99,7 @@ def test_serialize_workflow(vellum_client):
             "source_handle_id": "13d9eb34-aecb-496d-9e57-d5e786b0bc7c",
         },
         "display_data": {
-            "position": {"x": 0.0, "y": 0.0},
+            "position": {"x": 0.0, "y": -50.0},
         },
     }
 
@@ -144,7 +144,7 @@ def test_serialize_workflow(vellum_client):
             "workflow_deployment_id": deployment.id,
             "release_tag": "LATEST",
         },
-        "display_data": {"position": {"x": 0.0, "y": 0.0}},
+        "display_data": {"position": {"x": 200.0, "y": -50.0}},
         "base": {
             "module": ["vellum", "workflows", "nodes", "displayable", "subworkflow_deployment_node", "node"],
             "name": "SubworkflowDeploymentNode",
@@ -199,7 +199,7 @@ def test_serialize_workflow(vellum_client):
                         },
                     }
                 ],
-                "display_data": {"position": {"x": 0.0, "y": 0.0}},
+                "display_data": {"position": {"x": 400.0, "y": -175.0}},
             },
             {
                 "id": "94afd0ac-1ec4-486b-a6fb-fa1ec7029d19",
@@ -242,7 +242,7 @@ def test_serialize_workflow(vellum_client):
                         },
                     }
                 ],
-                "display_data": {"position": {"x": 0.0, "y": 0.0}},
+                "display_data": {"position": {"x": 400.0, "y": 75.0}},
             },
         ],
         workflow_raw_data["nodes"][2:],

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

@@ -56,7 +56,7 @@ def test_serialize_workflow():
         "type": "ENTRYPOINT",
         "inputs": [],
         "data": {"label": "Entrypoint Node", "source_handle_id": "34069190-0942-4e0c-8700-b33b9dea4ea0"},
-        "display_data": {"position": {"x": 0.0, "y": 0.0}},
+        "display_data": {"position": {"x": 0.0, "y": -50.0}},
         "base": None,
         "definition": None,
     }
@@ -103,7 +103,7 @@ def test_serialize_workflow():
                 "template_node_input_id": "7c775379-d589-4d79-b876-dcd224d72966",
                 "output_type": "JSON",
             },
-            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+            "display_data": {"position": {"x": 200.0, "y": -50.0}},
             "base": {
                 "name": "TemplatingNode",
                 "module": ["vellum", "workflows", "nodes", "core", "templating_node", "node"],
@@ -148,7 +148,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": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],

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

@@ -54,7 +54,7 @@ def test_serialize_workflow():
         "type": "ENTRYPOINT",
         "inputs": [],
         "data": {"label": "Entrypoint Node", "source_handle_id": "8b8d52a2-844f-44fe-a6c4-142fa70d391b"},
-        "display_data": {"position": {"x": 0.0, "y": 0.0}},
+        "display_data": {"position": {"x": 0.0, "y": -50.0}},
         "base": None,
         "definition": None,
     }
@@ -86,7 +86,7 @@ def test_serialize_workflow():
                 },
             }
         ],
-        "display_data": {"position": {"x": 0.0, "y": 0.0}},
+        "display_data": {"position": {"x": 200.0, "y": -50.0}},
         "base": {
             "name": "FinalOutputNode",
             "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],

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

@@ -43,7 +43,7 @@ def test_serialize_workflow():
         "label": "GetCurrentWeatherNode",
         "type": "GENERIC",
         "display_data": {
-            "position": {"x": 0.0, "y": 0.0},
+            "position": {"x": 200.0, "y": -50.0},
             "comment": {"value": "\n    A tool calling node that calls the get_current_weather function.\n    "},
         },
         "base": {
@@ -131,6 +131,8 @@ def test_serialize_workflow():
                         "value": [
                             {
                                 "type": "INLINE_WORKFLOW",
+                                "name": "BasicInlineSubworkflowWorkflow",
+                                "description": "\n    A workflow that gets the weather for a given city and date.\n    ",  # noqa: E501
                                 "exec_config": {
                                     "workflow_raw_data": {
                                         "nodes": [
@@ -142,7 +144,7 @@ def test_serialize_workflow():
                                                     "label": "Entrypoint Node",
                                                     "source_handle_id": "c344fdee-282b-40c9-8c97-6dd08830948c",
                                                 },
-                                                "display_data": {"position": {"x": 0.0, "y": 0.0}},
+                                                "display_data": {"position": {"x": 0.0, "y": -50.0}},
                                                 "base": None,
                                                 "definition": None,
                                             },
@@ -150,7 +152,7 @@ def test_serialize_workflow():
                                                 "id": "1381c078-efa2-4255-89a1-7b4cb742c7fc",
                                                 "label": "StartNode",
                                                 "type": "GENERIC",
-                                                "display_data": {"position": {"x": 0.0, "y": 0.0}},
+                                                "display_data": {"position": {"x": 200.0, "y": -50.0}},
                                                 "base": {
                                                     "name": "BaseNode",
                                                     "module": ["vellum", "workflows", "nodes", "bases", "base"],
@@ -238,7 +240,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": [
@@ -281,7 +283,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": [
@@ -410,6 +412,11 @@ def test_serialize_workflow():
                 "name": "function_configs",
                 "value": {"type": "CONSTANT_VALUE", "value": {"type": "JSON", "value": None}},
             },
+            {
+                "id": "1668419e-a193-43a5-8a97-3394e89bf278",
+                "name": "max_prompt_iterations",
+                "value": {"type": "CONSTANT_VALUE", "value": {"type": "NUMBER", "value": 5.0}},
+            },
         ],
         "outputs": [
             {"id": "e62bc785-a914-4066-b79e-8c89a5d0ec6c", "name": "text", "type": "STRING", "value": None},

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

@@ -43,7 +43,7 @@ def test_serialize_workflow():
         "label": "GetCurrentWeatherNode",
         "type": "GENERIC",
         "display_data": {
-            "position": {"x": 0.0, "y": 0.0},
+            "position": {"x": 200.0, "y": -50.0},
             "comment": {"value": "\n    A tool calling node that calls the get_current_weather function.\n    "},
         },
         "base": {
@@ -131,6 +131,8 @@ def test_serialize_workflow():
                         "value": [
                             {
                                 "type": "CODE_EXECUTION",
+                                "name": "get_current_weather",
+                                "description": "\n    Get the current weather in a given location.\n    ",
                                 "definition": {
                                     "state": None,
                                     "cache_config": None,
@@ -183,6 +185,11 @@ def test_serialize_workflow():
                     ],
                 },
             },
+            {
+                "id": "1668419e-a193-43a5-8a97-3394e89bf278",
+                "name": "max_prompt_iterations",
+                "value": {"type": "CONSTANT_VALUE", "value": {"type": "NUMBER", "value": 5.0}},
+            },
         ],
         "outputs": [
             {"id": "e62bc785-a914-4066-b79e-8c89a5d0ec6c", "name": "text", "type": "STRING", "value": None},

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

@@ -0,0 +1,62 @@
+from deepdiff import DeepDiff
+
+from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
+
+from tests.workflows.basic_tool_calling_node_workflow_deployment.workflow import (
+    BasicToolCallingNodeWorkflowDeploymentWorkflow,
+)
+
+
+def test_serialize_workflow():
+    # GIVEN a Workflow that uses a generic node
+    # WHEN we serialize it
+    workflow_display = get_workflow_display(workflow_class=BasicToolCallingNodeWorkflowDeploymentWorkflow)
+
+    serialized_workflow: dict = workflow_display.serialize()
+    # THEN we should get a serialized representation of the Workflow
+    assert serialized_workflow.keys() == {
+        "workflow_raw_data",
+        "input_variables",
+        "state_variables",
+        "output_variables",
+    }
+
+    # AND its input variables should be what we expect
+    input_variables = serialized_workflow["input_variables"]
+    assert len(input_variables) == 1
+
+    # AND its output variables should be what we expect
+    output_variables = serialized_workflow["output_variables"]
+    assert len(output_variables) == 2
+    assert not DeepDiff(
+        [
+            {"id": "3626cfa6-76f8-465e-b156-c809e3a2cee9", "key": "text", "type": "STRING"},
+            {"id": "cb82117a-d649-4c3e-9342-c552028fa2ad", "key": "chat_history", "type": "CHAT_HISTORY"},
+        ],
+        output_variables,
+        ignore_order=True,
+    )
+
+    # AND its raw data should be what we expect
+    workflow_raw_data = serialized_workflow["workflow_raw_data"]
+    tool_calling_node = workflow_raw_data["nodes"][1]
+    function_attributes = next(attr for attr in tool_calling_node["attributes"] if attr["name"] == "functions")
+    assert function_attributes == {
+        "id": "73a94e3c-1935-4308-a68a-ecd5441804b7",
+        "name": "functions",
+        "value": {
+            "type": "CONSTANT_VALUE",
+            "value": {
+                "type": "JSON",
+                "value": [
+                    {
+                        "type": "WORKFLOW_DEPLOYMENT",
+                        "name": "deployment_1",
+                        "description": "Workflow deployment for deployment_1",
+                        "deployment": "deployment_1",
+                        "release_tag": "LATEST",
+                    }
+                ],
+            },
+        },
+    }