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_merge_node_serialization.py

@@ -36,7 +36,6 @@ def test_serialize_workflow__await_all():
 
     # 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"]) == 6
     assert len(workflow_raw_data["nodes"]) == 6
 
@@ -53,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},
         },
     }
 
@@ -63,19 +62,19 @@ def test_serialize_workflow__await_all():
     merge_node = next(node for node in workflow_raw_data["nodes"] if node["type"] == "MERGE")
     assert not DeepDiff(
         {
-            "id": "37c10e8a-771b-432b-a767-31f5007851f0",
+            "id": "f07c263c-65a3-4b58-83c1-f4a29123f167",
             "type": "MERGE",
             "inputs": [],
             "data": {
                 "label": "Await All Merge Node",
                 "merge_strategy": "AWAIT_ALL",
                 "target_handles": [
-                    {"id": "f40ff7fb-de1b-4aa4-ba3c-7630f7357cbf"},
-                    {"id": "42eeb66c-9792-4609-8c71-3a56f668f4dc"},
+                    {"id": "4441c835-7d16-4c43-8599-e948b57eaab1"},
+                    {"id": "6da3e50a-8c6d-4de1-8ee9-da26f7c9552f"},
                 ],
-                "source_handle_id": "3bbc469f-0fb0-4b3d-a28b-746fefec2818",
+                "source_handle_id": "da1bdfe9-8e99-4d06-842f-a76af95a713a",
             },
-            "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",
@@ -84,6 +83,11 @@ def test_serialize_workflow__await_all():
                 "module": ["tests", "workflows", "basic_merge_node", "await_all_workflow"],
                 "name": "AwaitAllMergeNode",
             },
+            "trigger": {
+                "id": "1c68b622-cc93-4678-a0c4-89f06b6cad1f",
+                "merge_behavior": "AWAIT_ALL",
+            },
+            "ports": [{"id": "da1bdfe9-8e99-4d06-842f-a76af95a713a", "name": "default", "type": "DEFAULT"}],
         },
         merge_node,
         ignore_order_func=lambda x: x.path() == "root['data']['target_handles']",
@@ -122,8 +126,8 @@ def test_serialize_workflow__await_all():
                         {
                             "type": "NODE_OUTPUT",
                             "data": {
-                                "node_id": "634f0202-9ea9-4c62-b152-1a58c595cffb",
-                                "output_id": "d4266640-9718-4a74-b24b-500448d87871",
+                                "node_id": "6c34a839-552b-436d-ba6a-501f663883c8",
+                                "output_id": "f6148c39-9ec1-4b10-9d24-d19bec9eeea8",
                             },
                         }
                     ],
@@ -131,7 +135,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
@@ -139,49 +143,49 @@ def test_serialize_workflow__await_all():
     assert not DeepDiff(
         [
             {
-                "id": "9a65dd52-c3eb-496e-9d34-46b39534a261",
+                "id": "ea747507-68d5-4c72-8d00-a464ff7a55f1",
                 "source_node_id": "dc8aecd0-49ba-4464-a45f-29d3bfd686e4",
                 "source_handle_id": "017d40f5-8326-4e42-a409-b08995defaa8",
-                "target_node_id": "59243c65-053f-4ea6-9157-3f3edb1477bf",
-                "target_handle_id": "e622fe61-3bca-4aff-86e1-25dad7bdf9d4",
+                "target_node_id": "0871708d-8f05-4bc8-b3fb-a8624dae51de",
+                "target_handle_id": "01543a95-6bcb-46e2-a16c-570008972b88",
                 "type": "DEFAULT",
             },
             {
-                "id": "e5598f3c-fb00-4f25-a0a6-9fb6af9d69a8",
+                "id": "2c28f06b-4f4a-4531-8cf9-860089ce1cc2",
                 "source_node_id": "dc8aecd0-49ba-4464-a45f-29d3bfd686e4",
                 "source_handle_id": "017d40f5-8326-4e42-a409-b08995defaa8",
-                "target_node_id": "127ef456-91bc-43c6-bd8b-1772db5e3cb5",
-                "target_handle_id": "e5cc41cb-71db-43ec-b3f0-c78706af3351",
+                "target_node_id": "0306d2a2-8e2a-49d1-bc4d-4026fbd98c4c",
+                "target_handle_id": "5b10f22e-ae81-4ca5-a682-12d3c3ec73c1",
                 "type": "DEFAULT",
             },
             {
-                "id": "8ff20817-974e-4a3a-bb65-f0ad73557649",
-                "source_node_id": "59243c65-053f-4ea6-9157-3f3edb1477bf",
-                "source_handle_id": "b9c5f52b-b714-46e8-a09c-38b4e770dd36",
-                "target_node_id": "37c10e8a-771b-432b-a767-31f5007851f0",
-                "target_handle_id": "42eeb66c-9792-4609-8c71-3a56f668f4dc",
+                "id": "3870f290-8da7-43a8-b875-60510c060380",
+                "source_node_id": "0306d2a2-8e2a-49d1-bc4d-4026fbd98c4c",
+                "source_handle_id": "4e76416e-4edf-4a4a-9202-bf3ee35fd911",
+                "target_node_id": "f07c263c-65a3-4b58-83c1-f4a29123f167",
+                "target_handle_id": "4441c835-7d16-4c43-8599-e948b57eaab1",
                 "type": "DEFAULT",
             },
             {
-                "id": "0d8c801c-d76a-437a-831a-530885b75f96",
-                "source_node_id": "127ef456-91bc-43c6-bd8b-1772db5e3cb5",
-                "source_handle_id": "b0bd17f3-4ce6-4232-9666-ec8afa161bf2",
-                "target_node_id": "37c10e8a-771b-432b-a767-31f5007851f0",
-                "target_handle_id": "f40ff7fb-de1b-4aa4-ba3c-7630f7357cbf",
+                "id": "e89010d4-1f5a-4db9-8b67-1fa68960b142",
+                "source_node_id": "0871708d-8f05-4bc8-b3fb-a8624dae51de",
+                "source_handle_id": "c11fe2da-9ccd-4434-9330-5e3760abe849",
+                "target_node_id": "f07c263c-65a3-4b58-83c1-f4a29123f167",
+                "target_handle_id": "6da3e50a-8c6d-4de1-8ee9-da26f7c9552f",
                 "type": "DEFAULT",
             },
             {
-                "id": "70c1005d-339a-41bc-b6c2-10bc30a0281c",
-                "source_node_id": "37c10e8a-771b-432b-a767-31f5007851f0",
-                "source_handle_id": "3bbc469f-0fb0-4b3d-a28b-746fefec2818",
-                "target_node_id": "634f0202-9ea9-4c62-b152-1a58c595cffb",
-                "target_handle_id": "acd48f48-54fb-4b2b-ab37-96d336f6dfb3",
+                "id": "5344fbf3-6a71-4890-93a6-c6a2c5a4e50c",
+                "source_node_id": "f07c263c-65a3-4b58-83c1-f4a29123f167",
+                "source_handle_id": "da1bdfe9-8e99-4d06-842f-a76af95a713a",
+                "target_node_id": "6c34a839-552b-436d-ba6a-501f663883c8",
+                "target_handle_id": "413c3699-a37b-49cd-93fd-fe6d041a0119",
                 "type": "DEFAULT",
             },
             {
                 "id": "3d031c93-09b1-4937-9f98-c30a7ba6823d",
-                "source_node_id": "634f0202-9ea9-4c62-b152-1a58c595cffb",
-                "source_handle_id": "de32167b-cf53-4df5-a344-1b9be852e9ff",
+                "source_node_id": "6c34a839-552b-436d-ba6a-501f663883c8",
+                "source_handle_id": "48beca40-00f3-4273-b2ea-5e1ac6494839",
                 "target_node_id": "8187ce10-62b7-4a2c-8c0f-297387915467",
                 "target_handle_id": "ff55701c-16d3-4348-a633-6a298e71377d",
                 "type": "DEFAULT",

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

@@ -7,6 +7,9 @@ from vellum import DeploymentRead
 from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
 
 from tests.workflows.basic_text_prompt_deployment.workflow import BasicTextPromptDeployment
+from tests.workflows.basic_text_prompt_deployment.workflow_with_prompt_deployment_json_reference import (
+    WorkflowWithPromptDeploymentJsonReferenceWorkflow,
+)
 
 
 def test_serialize_workflow(vellum_client):
@@ -74,7 +77,6 @@ def test_serialize_workflow(vellum_client):
 
     # 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
 
@@ -91,72 +93,73 @@ 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},
         },
     }
 
     prompt_node = workflow_raw_data["nodes"][1]
     assert prompt_node == {
-        "id": "56c74024-19a3-4c0d-a5f5-23e1e9f11b21",
+        "id": "a114abf2-cb76-49b5-8001-6c8df56d39ff",
         "type": "PROMPT",
         "inputs": [
             {
-                "id": "509cf3d4-db72-483e-897a-0f7f20e70d03",
+                "id": "c58d2244-a52e-4edb-84c5-efc334dea6f9",
                 "key": "city",
                 "value": {
-                    "combinator": "OR",
                     "rules": [
                         {
                             "type": "INPUT_VARIABLE",
-                            "data": {
-                                "input_variable_id": "52995b50-84c9-465f-8a4b-a4ee2a92e388",
-                            },
-                        },
+                            "data": {"input_variable_id": "52995b50-84c9-465f-8a4b-a4ee2a92e388"},
+                        }
                     ],
+                    "combinator": "OR",
                 },
             },
             {
-                "id": "7d6ec5ac-c582-4153-bd8a-b8794c367420",
+                "id": "d00707fb-b584-40bc-aecd-db11490bea3e",
                 "key": "date",
                 "value": {
-                    "combinator": "OR",
                     "rules": [
                         {
                             "type": "INPUT_VARIABLE",
-                            "data": {
-                                "input_variable_id": "aa3ca842-250c-4a3f-853f-23928c28d0f8",
-                            },
-                        },
+                            "data": {"input_variable_id": "aa3ca842-250c-4a3f-853f-23928c28d0f8"},
+                        }
                     ],
+                    "combinator": "OR",
                 },
             },
         ],
         "data": {
             "label": "Example Prompt Deployment Node",
-            "output_id": "4d38b850-79e3-4b85-9158-a41d0c535410",
+            "output_id": "f07b3521-4ba1-4a3c-8629-3a269406f519",
             "error_output_id": None,
-            "array_output_id": "0cf47d33-6d5f-466f-b826-e814f1d0348b",
-            "source_handle_id": "2f26c7e0-283d-4f04-b639-adebb56bc679",
-            "target_handle_id": "b7605c48-0937-4ecc-914e-0d1058130e65",
+            "array_output_id": "14d71e78-94a5-4c46-bad8-bec827a4f1e4",
+            "source_handle_id": "6699f465-dc6c-4fa7-8038-7ff49419b953",
+            "target_handle_id": "1407e51f-cb29-4a86-beeb-cc1870dc5525",
             "variant": "DEPLOYMENT",
             "prompt_deployment_id": deployment.id,
             "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": {
-            "module": ["vellum", "workflows", "nodes", "displayable", "prompt_deployment_node", "node"],
             "name": "PromptDeploymentNode",
+            "module": ["vellum", "workflows", "nodes", "displayable", "prompt_deployment_node", "node"],
         },
         "definition": {
-            "module": ["tests", "workflows", "basic_text_prompt_deployment", "workflow"],
             "name": "ExamplePromptDeploymentNode",
+            "module": ["tests", "workflows", "basic_text_prompt_deployment", "workflow"],
         },
+        "trigger": {
+            "id": "1407e51f-cb29-4a86-beeb-cc1870dc5525",
+            "merge_behavior": "AWAIT_ANY",
+        },
+        "ports": [{"id": "6699f465-dc6c-4fa7-8038-7ff49419b953", "name": "default", "type": "DEFAULT"}],
+        "outputs": [
+            {"id": "66687de8-3e00-4290-9177-54be727fef44", "name": "json", "type": "JSON", "value": None},
+            {"id": "f07b3521-4ba1-4a3c-8629-3a269406f519", "name": "text", "type": "STRING", "value": None},
+            {"id": "14d71e78-94a5-4c46-bad8-bec827a4f1e4", "name": "results", "type": "ARRAY", "value": None},
+        ],
     }
 
     final_output_node = workflow_raw_data["nodes"][2]
@@ -184,8 +187,8 @@ def test_serialize_workflow(vellum_client):
                     "rules": [
                         {
                             "data": {
-                                "node_id": "56c74024-19a3-4c0d-a5f5-23e1e9f11b21",
-                                "output_id": "4d38b850-79e3-4b85-9158-a41d0c535410",
+                                "node_id": "a114abf2-cb76-49b5-8001-6c8df56d39ff",
+                                "output_id": "f07b3521-4ba1-4a3c-8629-3a269406f519",
                             },
                             "type": "NODE_OUTPUT",
                         }
@@ -203,8 +206,8 @@ def test_serialize_workflow(vellum_client):
         },
         "display_data": {
             "position": {
-                "x": 0.0,
-                "y": 0.0,
+                "x": 400.0,
+                "y": -50.0,
             },
         },
     }
@@ -213,17 +216,17 @@ def test_serialize_workflow(vellum_client):
     serialized_edges = workflow_raw_data["edges"]
     assert serialized_edges == [
         {
-            "id": "8961d02b-074e-45ab-9f77-4e94606a4344",
+            "id": "89a6042d-2bfd-4512-9f75-bf11db3c493c",
             "source_handle_id": "7065a943-1cab-4afd-9690-e678c5b74a2f",
             "source_node_id": "d680afbd-de64-4cf6-aa50-912686c48c64",
-            "target_handle_id": "b7605c48-0937-4ecc-914e-0d1058130e65",
-            "target_node_id": "56c74024-19a3-4c0d-a5f5-23e1e9f11b21",
+            "target_handle_id": "1407e51f-cb29-4a86-beeb-cc1870dc5525",
+            "target_node_id": "a114abf2-cb76-49b5-8001-6c8df56d39ff",
             "type": "DEFAULT",
         },
         {
             "id": "c2cbf6ef-8582-45c8-a643-fc6ae8fe482f",
-            "source_handle_id": "2f26c7e0-283d-4f04-b639-adebb56bc679",
-            "source_node_id": "56c74024-19a3-4c0d-a5f5-23e1e9f11b21",
+            "source_handle_id": "6699f465-dc6c-4fa7-8038-7ff49419b953",
+            "source_node_id": "a114abf2-cb76-49b5-8001-6c8df56d39ff",
             "target_handle_id": "dced939a-9122-4290-8482-7daa9525dad6",
             "target_node_id": "64ff72c7-8ffc-4e1f-b7a7-e7cd0697f576",
             "type": "DEFAULT",
@@ -251,3 +254,331 @@ def test_serialize_workflow(vellum_client):
             "workflow",
         ],
     }
+
+
+def test_serialize_workflow_with_prompt_and_templating(vellum_client):
+    # GIVEN a Workflow with stubbed out API calls
+    deployment = DeploymentRead(
+        id=str(uuid4()),
+        created=datetime.now(),
+        label="JSON Prompt Deployment",
+        name="json_prompt_deployment",
+        last_deployed_on=datetime.now(),
+        input_variables=[],
+        active_model_version_ids=[],
+        last_deployed_history_item_id=str(uuid4()),
+    )
+    vellum_client.deployments.retrieve.return_value = deployment
+
+    # WHEN we serialize it
+    workflow_display = get_workflow_display(workflow_class=WorkflowWithPromptDeploymentJsonReferenceWorkflow)
+    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) == 2
+    assert not DeepDiff(
+        [
+            {
+                "id": "ad577aca-0797-4deb-a484-574ef1a1f0c7",
+                "key": "city",
+                "type": "STRING",
+                "default": None,
+                "required": True,
+                "extensions": {"color": None},
+            },
+            {
+                "id": "066124c4-42bd-4764-aa75-6f230dbbed4a",
+                "key": "date",
+                "type": "STRING",
+                "default": None,
+                "required": True,
+                "extensions": {"color": None},
+            },
+        ],
+        input_variables,
+        ignore_order=True,
+    )
+
+    # AND its output variables should be what we expect
+    output_variables = serialized_workflow["output_variables"]
+    assert len(output_variables) == 1
+    assert output_variables == [{"id": "a7e4b449-5879-4d0c-8f00-d5d4985eb65c", "key": "text", "type": "STRING"}]
+
+    # AND its raw data should be what we expect
+    workflow_raw_data = serialized_workflow["workflow_raw_data"]
+    assert len(workflow_raw_data["edges"]) == 3
+    assert len(workflow_raw_data["nodes"]) == 4
+
+    # AND each node should be serialized correctly
+    entrypoint_node = workflow_raw_data["nodes"][0]
+    assert entrypoint_node == {
+        "id": "32c7f398-277c-456b-9279-aa1f867fb637",
+        "type": "ENTRYPOINT",
+        "inputs": [],
+        "data": {"label": "Entrypoint Node", "source_handle_id": "cc0f4028-1039-4063-971d-7dacbb01b379"},
+        "display_data": {"position": {"x": 0.0, "y": -50.0}},
+        "base": None,
+        "definition": None,
+    }
+
+    prompt_node = workflow_raw_data["nodes"][1]
+    assert prompt_node == {
+        "id": "0d1460e4-f207-4a69-bcea-7a3c7b325c02",
+        "type": "PROMPT",
+        "inputs": [
+            {
+                "id": "b99dec61-8c38-4b6b-96fa-cb83d5c9a9ef",
+                "key": "city",
+                "value": {
+                    "rules": [
+                        {
+                            "type": "INPUT_VARIABLE",
+                            "data": {"input_variable_id": "ad577aca-0797-4deb-a484-574ef1a1f0c7"},
+                        }
+                    ],
+                    "combinator": "OR",
+                },
+            },
+            {
+                "id": "6379398c-3c29-4327-9346-4d386a467f16",
+                "key": "date",
+                "value": {
+                    "rules": [
+                        {
+                            "type": "INPUT_VARIABLE",
+                            "data": {"input_variable_id": "066124c4-42bd-4764-aa75-6f230dbbed4a"},
+                        }
+                    ],
+                    "combinator": "OR",
+                },
+            },
+        ],
+        "data": {
+            "label": "Example Prompt Deployment Node",
+            "output_id": "b9d579e2-b3ee-4524-917f-7329fc09af59",
+            "error_output_id": None,
+            "array_output_id": "547b4a00-eb16-4df5-92f4-cdd2fe7a0848",
+            "source_handle_id": "7e29137d-af96-402c-8108-9a00e087d18e",
+            "target_handle_id": "f8017ad7-14f2-4e6f-8456-a081db5ed7cd",
+            "variant": "DEPLOYMENT",
+            "prompt_deployment_id": deployment.id,
+            "release_tag": "LATEST",
+            "ml_model_fallbacks": None,
+        },
+        "display_data": {"position": {"x": 200.0, "y": -50.0}},
+        "base": {
+            "name": "PromptDeploymentNode",
+            "module": ["vellum", "workflows", "nodes", "displayable", "prompt_deployment_node", "node"],
+        },
+        "definition": {
+            "name": "ExamplePromptDeploymentNode",
+            "module": [
+                "tests",
+                "workflows",
+                "basic_text_prompt_deployment",
+                "workflow_with_prompt_deployment_json_reference",
+            ],
+        },
+        "trigger": {
+            "id": "f8017ad7-14f2-4e6f-8456-a081db5ed7cd",
+            "merge_behavior": "AWAIT_ANY",
+        },
+        "ports": [{"id": "7e29137d-af96-402c-8108-9a00e087d18e", "name": "default", "type": "DEFAULT"}],
+        "outputs": [
+            {"id": "62bbe13a-9571-4165-9463-22092a04e450", "name": "json", "type": "JSON", "value": None},
+            {"id": "b9d579e2-b3ee-4524-917f-7329fc09af59", "name": "text", "type": "STRING", "value": None},
+            {"id": "547b4a00-eb16-4df5-92f4-cdd2fe7a0848", "name": "results", "type": "ARRAY", "value": None},
+        ],
+    }
+
+    templating_node = workflow_raw_data["nodes"][2]
+    assert templating_node == {
+        "id": "ac2de275-729f-4fba-9701-97beba80df34",
+        "type": "TEMPLATING",
+        "inputs": [
+            {
+                "id": "198f6350-237a-4103-b571-311738b7743f",
+                "key": "template",
+                "value": {
+                    "rules": [
+                        {
+                            "type": "CONSTANT_VALUE",
+                            "data": {
+                                "type": "STRING",
+                                "value": "The weather in {{ city }} on {{ date }} is {{ weather }}.",
+                            },
+                        }
+                    ],
+                    "combinator": "OR",
+                },
+            },
+            {
+                "id": "2439a747-3b8d-44e9-85c1-99b903d69997",
+                "key": "city",
+                "value": {
+                    "rules": [
+                        {
+                            "type": "INPUT_VARIABLE",
+                            "data": {"input_variable_id": "ad577aca-0797-4deb-a484-574ef1a1f0c7"},
+                        }
+                    ],
+                    "combinator": "OR",
+                },
+            },
+            {
+                "id": "d85dfd04-f304-4bbf-9d24-52d9ef8f45cc",
+                "key": "date",
+                "value": {
+                    "rules": [
+                        {
+                            "type": "INPUT_VARIABLE",
+                            "data": {"input_variable_id": "066124c4-42bd-4764-aa75-6f230dbbed4a"},
+                        }
+                    ],
+                    "combinator": "OR",
+                },
+            },
+            {
+                "id": "a555cbb1-cea9-4454-add0-61609b38a376",
+                "key": "weather",
+                "value": {
+                    "rules": [
+                        {
+                            "type": "NODE_OUTPUT",
+                            "data": {
+                                "node_id": "0d1460e4-f207-4a69-bcea-7a3c7b325c02",
+                                "output_id": "62bbe13a-9571-4165-9463-22092a04e450",
+                            },
+                        }
+                    ],
+                    "combinator": "OR",
+                },
+            },
+        ],
+        "data": {
+            "label": "Example Templating Node",
+            "output_id": "c8ad13db-d601-4c24-bef7-0196baa8079c",
+            "error_output_id": None,
+            "source_handle_id": "bfe059fb-987f-47dd-bfbb-c71fdf1e4971",
+            "target_handle_id": "fbd5aa14-b615-42c9-a85a-23eb1a6b5436",
+            "template_node_input_id": "198f6350-237a-4103-b571-311738b7743f",
+            "output_type": "STRING",
+        },
+        "display_data": {"position": {"x": 400.0, "y": -50.0}},
+        "base": {
+            "name": "TemplatingNode",
+            "module": ["vellum", "workflows", "nodes", "core", "templating_node", "node"],
+        },
+        "definition": {
+            "name": "ExampleTemplatingNode",
+            "module": [
+                "tests",
+                "workflows",
+                "basic_text_prompt_deployment",
+                "workflow_with_prompt_deployment_json_reference",
+            ],
+        },
+        "trigger": {
+            "id": "fbd5aa14-b615-42c9-a85a-23eb1a6b5436",
+            "merge_behavior": "AWAIT_ATTRIBUTES",
+        },
+        "ports": [{"id": "bfe059fb-987f-47dd-bfbb-c71fdf1e4971", "name": "default", "type": "DEFAULT"}],
+    }
+
+    final_output_node = workflow_raw_data["nodes"][3]
+    assert final_output_node == {
+        "id": "53de824d-a41d-4294-b511-c969932b05af",
+        "type": "TERMINAL",
+        "data": {
+            "label": "Final Output",
+            "name": "text",
+            "target_handle_id": "fee3d395-38c3-485f-ab61-1a0fdf71c4ce",
+            "output_id": "a7e4b449-5879-4d0c-8f00-d5d4985eb65c",
+            "output_type": "STRING",
+            "node_input_id": "cf380f81-c5ee-4bc9-8e26-ecf1307733a9",
+        },
+        "inputs": [
+            {
+                "id": "cf380f81-c5ee-4bc9-8e26-ecf1307733a9",
+                "key": "node_input",
+                "value": {
+                    "rules": [
+                        {
+                            "type": "NODE_OUTPUT",
+                            "data": {
+                                "node_id": "ac2de275-729f-4fba-9701-97beba80df34",
+                                "output_id": "c8ad13db-d601-4c24-bef7-0196baa8079c",
+                            },
+                        }
+                    ],
+                    "combinator": "OR",
+                },
+            }
+        ],
+        "display_data": {"position": {"x": 600.0, "y": -50.0}},
+        "base": {
+            "name": "FinalOutputNode",
+            "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
+        },
+        "definition": None,
+    }
+
+    # AND each edge should be serialized correctly
+    serialized_edges = workflow_raw_data["edges"]
+    assert serialized_edges == [
+        {
+            "id": "33e0220a-969a-4d38-86ac-0fe596ab2d88",
+            "source_node_id": "32c7f398-277c-456b-9279-aa1f867fb637",
+            "source_handle_id": "cc0f4028-1039-4063-971d-7dacbb01b379",
+            "target_node_id": "0d1460e4-f207-4a69-bcea-7a3c7b325c02",
+            "target_handle_id": "f8017ad7-14f2-4e6f-8456-a081db5ed7cd",
+            "type": "DEFAULT",
+        },
+        {
+            "id": "8eb009df-4daf-4ff6-8dc2-ff71a2829e2b",
+            "source_node_id": "0d1460e4-f207-4a69-bcea-7a3c7b325c02",
+            "source_handle_id": "7e29137d-af96-402c-8108-9a00e087d18e",
+            "target_node_id": "ac2de275-729f-4fba-9701-97beba80df34",
+            "target_handle_id": "fbd5aa14-b615-42c9-a85a-23eb1a6b5436",
+            "type": "DEFAULT",
+        },
+        {
+            "id": "1f720900-e5e1-49b7-9910-6ede79f6afd2",
+            "source_node_id": "ac2de275-729f-4fba-9701-97beba80df34",
+            "source_handle_id": "bfe059fb-987f-47dd-bfbb-c71fdf1e4971",
+            "target_node_id": "53de824d-a41d-4294-b511-c969932b05af",
+            "target_handle_id": "fee3d395-38c3-485f-ab61-1a0fdf71c4ce",
+            "type": "DEFAULT",
+        },
+    ]
+
+    # AND the display data should be what we expect
+    display_data = workflow_raw_data["display_data"]
+    assert display_data == {
+        "viewport": {
+            "x": 0.0,
+            "y": 0.0,
+            "zoom": 1.0,
+        }
+    }
+
+    # AND the definition should be what we expect
+    definition = workflow_raw_data["definition"]
+    assert definition == {
+        "name": "WorkflowWithPromptDeploymentJsonReferenceWorkflow",
+        "module": [
+            "tests",
+            "workflows",
+            "basic_text_prompt_deployment",
+            "workflow_with_prompt_deployment_json_reference",
+        ],
+    }