vellum-ai 0.14.61__py3-none-any.whl → 0.14.62__py3-none-any.whl

vellum/client/core/client_wrapper.py

@@ -18,7 +18,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.14.61",
+            "X-Fern-SDK-Version": "0.14.62",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

{vellum_ai-0.14.61.dist-info → vellum_ai-0.14.62.dist-info}/METADATA

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

{vellum_ai-0.14.61.dist-info → vellum_ai-0.14.62.dist-info}/RECORD

@@ -45,7 +45,7 @@ vellum_ee/workflows/display/nodes/vellum/inline_subworkflow_node.py,sha256=fQV5o
 vellum_ee/workflows/display/nodes/vellum/map_node.py,sha256=CiklGf5_tDbqE1XQm2mnbtoL01_2JYjcnB4FDTpMImQ,3824
 vellum_ee/workflows/display/nodes/vellum/merge_node.py,sha256=yBWeN4T_lOsDVnNOKWRiT7JYKu0IR5Fx2z99iq6QKSA,3273
 vellum_ee/workflows/display/nodes/vellum/note_node.py,sha256=3E0UqmgVYdtbj4nyq8gKju8EpMsRHMCQ0KLrJhug3XU,1084
-vellum_ee/workflows/display/nodes/vellum/prompt_deployment_node.py,sha256=dVb3ONkuovK8xE5MdDLVSBCoZoKPrCnhdcWLYcl14mw,3307
+vellum_ee/workflows/display/nodes/vellum/prompt_deployment_node.py,sha256=1NxFddxWCFtMe_je1cutP7qnoASoG94LJqKhRkoQwvw,3535
 vellum_ee/workflows/display/nodes/vellum/retry_node.py,sha256=X3xnlAU5JisL0jRvaG_V9RvTF7ZlGufTO8tXLLVhGIg,3280
 vellum_ee/workflows/display/nodes/vellum/search_node.py,sha256=3n1qa-zWIk0p_H94u0hjfDtGkanldC6EXVhg0xgLmE4,9544
 vellum_ee/workflows/display/nodes/vellum/subworkflow_deployment_node.py,sha256=MWLZBXHsmj5vKgqOgI2HHcHAJzLS0sqybn6idhwug8Y,2669
@@ -85,7 +85,7 @@ vellum_ee/workflows/display/tests/workflow_serialization/test_basic_inline_promp
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_inline_subworkflow_serialization.py,sha256=u2nquKoO3o2xIkU_uFPOb_s5YoLmULiq09vb6Ee0Cqw,21415
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_map_node_serialization.py,sha256=3gZuNM8sT6ovVaeoAvd2JoyKwuxokvowlhH8kwDUoZ8,16559
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_merge_node_serialization.py,sha256=IIJt7YZBzkhNtbmaMwCX4ENs58QtSIIoBHlMR6OwGU8,8342
-vellum_ee/workflows/display/tests/workflow_serialization/test_basic_prompt_deployment_serialization.py,sha256=mLr_1Ow5NZYSLzzEZPup14LckuT8ivOvq9Lz85pGj0c,8602
+vellum_ee/workflows/display/tests/workflow_serialization/test_basic_prompt_deployment_serialization.py,sha256=QXiRjwtiTPeMUl40Pvh_geeU99C3mv1aVS85oeIUwY4,21052
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_search_node_serialization.py,sha256=K0mECBUyTNX4I468goeaRf93N7Qfizoc1ctCH5_nNys,12936
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_subworkflow_deployment_serialization.py,sha256=KkYZc_bZuq1lmDcvUz3QxIqJLpQPCZioD1FHUNsMJY8,11211
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_templating_node_serialization.py,sha256=aZaqRDrkO3ytcmdM2eKJqHSt60MF070NMj6M2vgzOKc,7711
@@ -134,7 +134,7 @@ vellum/client/README.md,sha256=qmaVIP42MnxAu8jV7u-CsgVFfs3-pHQODrXdZdFxtaw,4749
 vellum/client/__init__.py,sha256=AYopGv2ZRVn3zsU8_km6KOvEHDbXiTPCVuYVI7bWvdA,120166
 vellum/client/core/__init__.py,sha256=SQ85PF84B9MuKnBwHNHWemSGuy-g_515gFYNFhvEE0I,1438
 vellum/client/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-vellum/client/core/client_wrapper.py,sha256=Y50hXu4pkEmrPDb1Vq6PeuiVG0NavAoEk1Dexp7lgSE,1869
+vellum/client/core/client_wrapper.py,sha256=6Ws3Z7WyEV9P2j-NicrbyppAaPyLzK4wAJ5mxUvUdJ4,1869
 vellum/client/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/client/core/file.py,sha256=d4NNbX8XvXP32z8KpK2Xovv33nFfruIrpz0QWxlgpZk,2663
 vellum/client/core/http_client.py,sha256=Z77OIxIbL4OAB2IDqjRq_sYa5yNYAWfmdhdCSSvh6Y4,19552
@@ -1712,8 +1712,8 @@ vellum/workflows/workflows/event_filters.py,sha256=GSxIgwrX26a1Smfd-6yss2abGCnad
 vellum/workflows/workflows/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/workflows/tests/test_base_workflow.py,sha256=fROqff6AZpCIzaSwOKSdtYy4XR0UZQ6ejxL3RJOSJVs,20447
 vellum/workflows/workflows/tests/test_context.py,sha256=VJBUcyWVtMa_lE5KxdhgMu0WYNYnUQUDvTF7qm89hJ0,2333
-vellum_ai-0.14.61.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
-vellum_ai-0.14.61.dist-info/METADATA,sha256=6rjJiOUgei1EVdl_iiBzai45u8cWbO3pjeKrt4RBB2E,5484
-vellum_ai-0.14.61.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_ai-0.14.61.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
-vellum_ai-0.14.61.dist-info/RECORD,,
+vellum_ai-0.14.62.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
+vellum_ai-0.14.62.dist-info/METADATA,sha256=Bzub_INWkihzZyUxbedAO9QOhQmT-Ci17-6k_8JW5zE,5484
+vellum_ai-0.14.62.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_ai-0.14.62.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
+vellum_ai-0.14.62.dist-info/RECORD,,

vellum_ee/workflows/display/nodes/vellum/prompt_deployment_node.py

@@ -1,8 +1,7 @@
 from uuid import UUID
-from typing import Generic, Optional, TypeVar, cast
+from typing import Generic, Optional, TypeVar
 
 from vellum.workflows.nodes.displayable.prompt_deployment_node import PromptDeploymentNode
-from vellum.workflows.references import OutputReference
 from vellum.workflows.types.core import JsonObject
 from vellum_ee.workflows.display.nodes.base_node_display import BaseNodeDisplay
 from vellum_ee.workflows.display.nodes.utils import raise_if_descriptor
@@ -40,8 +39,9 @@ class BasePromptDeploymentNodeDisplay(BaseNodeDisplay[_PromptDeploymentNodeType]
             else []
         )
 
-        _, output_display = display_context.global_node_output_displays[cast(OutputReference, node.Outputs.text)]
-        _, array_display = display_context.global_node_output_displays[cast(OutputReference, node.Outputs.results)]
+        output_display = self.output_display[node.Outputs.text]
+        array_display = self.output_display[node.Outputs.results]
+        json_display = self.output_display[node.Outputs.json]
 
         deployment = display_context.client.deployments.retrieve(
             id=str(raise_if_descriptor(node.deployment)),
@@ -68,4 +68,9 @@ class BasePromptDeploymentNodeDisplay(BaseNodeDisplay[_PromptDeploymentNodeType]
             "base": self.get_base().dict(),
             "definition": self.get_definition().dict(),
             "ports": self.serialize_ports(display_context),
+            "outputs": [
+                {"id": str(json_display.id), "name": "json", "type": "JSON", "value": None},
+                {"id": str(output_display.id), "name": "text", "type": "STRING", "value": None},
+                {"id": str(array_display.id), "name": "results", "type": "ARRAY", "value": None},
+            ],
         }

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):
@@ -103,30 +106,26 @@ def test_serialize_workflow(vellum_client):
                 "id": "947d7ead-0fad-4e5f-aa3a-d06029ac94bc",
                 "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": "3deebdd7-2900-4d8c-93f2-e5b90649ac42",
                 "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",
                 },
             },
         ],
@@ -142,21 +141,21 @@ 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": 0.0, "y": 0.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"],
         },
         "ports": [{"id": "2f26c7e0-283d-4f04-b639-adebb56bc679", "name": "default", "type": "DEFAULT"}],
+        "outputs": [
+            {"id": "180355a8-e67c-4ce6-9ac3-e5dbb75a6629", "name": "json", "type": "JSON", "value": None},
+            {"id": "4d38b850-79e3-4b85-9158-a41d0c535410", "name": "text", "type": "STRING", "value": None},
+            {"id": "0cf47d33-6d5f-466f-b826-e814f1d0348b", "name": "results", "type": "ARRAY", "value": None},
+        ],
     }
 
     final_output_node = workflow_raw_data["nodes"][2]
@@ -251,3 +250,323 @@ 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": 0.0}},
+        "base": None,
+        "definition": None,
+    }
+
+    prompt_node = workflow_raw_data["nodes"][1]
+    assert prompt_node == {
+        "id": "56c74024-19a3-4c0d-a5f5-23e1e9f11b21",
+        "type": "PROMPT",
+        "inputs": [
+            {
+                "id": "947d7ead-0fad-4e5f-aa3a-d06029ac94bc",
+                "key": "city",
+                "value": {
+                    "rules": [
+                        {
+                            "type": "INPUT_VARIABLE",
+                            "data": {"input_variable_id": "ad577aca-0797-4deb-a484-574ef1a1f0c7"},
+                        }
+                    ],
+                    "combinator": "OR",
+                },
+            },
+            {
+                "id": "3deebdd7-2900-4d8c-93f2-e5b90649ac42",
+                "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": "4d38b850-79e3-4b85-9158-a41d0c535410",
+            "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",
+            "variant": "DEPLOYMENT",
+            "prompt_deployment_id": deployment.id,
+            "release_tag": "LATEST",
+            "ml_model_fallbacks": None,
+        },
+        "display_data": {"position": {"x": 0.0, "y": 0.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",
+            ],
+        },
+        "ports": [{"id": "2f26c7e0-283d-4f04-b639-adebb56bc679", "name": "default", "type": "DEFAULT"}],
+        "outputs": [
+            {"id": "180355a8-e67c-4ce6-9ac3-e5dbb75a6629", "name": "json", "type": "JSON", "value": None},
+            {"id": "4d38b850-79e3-4b85-9158-a41d0c535410", "name": "text", "type": "STRING", "value": None},
+            {"id": "0cf47d33-6d5f-466f-b826-e814f1d0348b", "name": "results", "type": "ARRAY", "value": None},
+        ],
+    }
+
+    templating_node = workflow_raw_data["nodes"][2]
+    assert templating_node == {
+        "id": "51cbe21d-0232-4362-bc54-5bc283297aa6",
+        "type": "TEMPLATING",
+        "inputs": [
+            {
+                "id": "7c775379-d589-4d79-b876-dcd224d72966",
+                "key": "template",
+                "value": {
+                    "rules": [
+                        {
+                            "type": "CONSTANT_VALUE",
+                            "data": {
+                                "type": "STRING",
+                                "value": "The weather in {{ city }} on {{ date }} is {{ weather }}.",
+                            },
+                        }
+                    ],
+                    "combinator": "OR",
+                },
+            },
+            {
+                "id": "dec1317a-6900-4858-a5fb-c849254b2c91",
+                "key": "city",
+                "value": {
+                    "rules": [
+                        {
+                            "type": "INPUT_VARIABLE",
+                            "data": {"input_variable_id": "ad577aca-0797-4deb-a484-574ef1a1f0c7"},
+                        }
+                    ],
+                    "combinator": "OR",
+                },
+            },
+            {
+                "id": "4cc5b9f1-075d-45fd-a978-f530c29c5682",
+                "key": "date",
+                "value": {
+                    "rules": [
+                        {
+                            "type": "INPUT_VARIABLE",
+                            "data": {"input_variable_id": "066124c4-42bd-4764-aa75-6f230dbbed4a"},
+                        }
+                    ],
+                    "combinator": "OR",
+                },
+            },
+            {
+                "id": "77b352e0-1b77-4d48-9f6f-04ce17fe7286",
+                "key": "weather",
+                "value": {
+                    "rules": [
+                        {
+                            "type": "NODE_OUTPUT",
+                            "data": {
+                                "node_id": "56c74024-19a3-4c0d-a5f5-23e1e9f11b21",
+                                "output_id": "180355a8-e67c-4ce6-9ac3-e5dbb75a6629",
+                            },
+                        }
+                    ],
+                    "combinator": "OR",
+                },
+            },
+        ],
+        "data": {
+            "label": "Example Templating Node",
+            "output_id": "6834cae4-8173-4fa6-88f7-bc09d335bdd1",
+            "error_output_id": None,
+            "source_handle_id": "39317827-df43-4f5a-bfbc-20bffc839748",
+            "target_handle_id": "58427684-3848-498a-8299-c6b0fc70265d",
+            "template_node_input_id": "7c775379-d589-4d79-b876-dcd224d72966",
+            "output_type": "STRING",
+        },
+        "display_data": {"position": {"x": 0.0, "y": 0.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",
+            ],
+        },
+        "ports": [{"id": "39317827-df43-4f5a-bfbc-20bffc839748", "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": "51cbe21d-0232-4362-bc54-5bc283297aa6",
+                                "output_id": "6834cae4-8173-4fa6-88f7-bc09d335bdd1",
+                            },
+                        }
+                    ],
+                    "combinator": "OR",
+                },
+            }
+        ],
+        "display_data": {"position": {"x": 0.0, "y": 0.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": "2c49ffa6-7b9a-49a0-a932-009534556480",
+            "source_node_id": "32c7f398-277c-456b-9279-aa1f867fb637",
+            "source_handle_id": "cc0f4028-1039-4063-971d-7dacbb01b379",
+            "target_node_id": "56c74024-19a3-4c0d-a5f5-23e1e9f11b21",
+            "target_handle_id": "b7605c48-0937-4ecc-914e-0d1058130e65",
+            "type": "DEFAULT",
+        },
+        {
+            "id": "a46909ec-9572-43c6-a134-0bd7e2c09f99",
+            "source_node_id": "56c74024-19a3-4c0d-a5f5-23e1e9f11b21",
+            "source_handle_id": "2f26c7e0-283d-4f04-b639-adebb56bc679",
+            "target_node_id": "51cbe21d-0232-4362-bc54-5bc283297aa6",
+            "target_handle_id": "58427684-3848-498a-8299-c6b0fc70265d",
+            "type": "DEFAULT",
+        },
+        {
+            "id": "1f720900-e5e1-49b7-9910-6ede79f6afd2",
+            "source_node_id": "51cbe21d-0232-4362-bc54-5bc283297aa6",
+            "source_handle_id": "39317827-df43-4f5a-bfbc-20bffc839748",
+            "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",
+        ],
+    }