vellum-ai 0.10.9__py3-none-any.whl → 0.11.1__py3-none-any.whl

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

@@ -4,17 +4,18 @@ from uuid import uuid4
 from deepdiff import DeepDiff
 
 from vellum import WorkspaceSecretRead
-
-from tests.workflows.basic_vellum_api_node.workflow import SimpleAPIWorkflow
 from vellum_ee.workflows.display.workflows import VellumWorkflowDisplay
 from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
 
+from tests.workflows.basic_vellum_api_node.workflow import SimpleAPIWorkflow
+
 
 def test_serialize_workflow(vellum_client):
     # GIVEN a Workflow that uses a vellum API node
     # AND stubbed out API calls
+    workspace_secret_id = str(uuid4())
     workspace_secret = WorkspaceSecretRead(
-        id=str(uuid4()),
+        id=workspace_secret_id,
         modified=datetime.now(),
         name="MY_SECRET",
         label="My Secret",
@@ -118,6 +119,16 @@ def test_serialize_workflow(vellum_client):
                         "combinator": "OR",
                     },
                 },
+                {
+                    "id": "fee5e3c9-442a-4922-ba80-5ee07361cea7",
+                    "key": "bearer_token_value",
+                    "value": {
+                        "rules": [
+                            {"type": "WORKSPACE_SECRET", "data": {"type": "STRING", "workspace_secret_id": None}}
+                        ],
+                        "combinator": "OR",
+                    },
+                },
                 {
                     "id": "2fcdfbc3-8095-4277-bb4a-a201fd326b54",
                     "key": "api_key_header_key",
@@ -133,12 +144,33 @@ def test_serialize_workflow(vellum_client):
                         "rules": [
                             {
                                 "type": "WORKSPACE_SECRET",
-                                "data": {"type": "STRING", "workspace_secret_id": workspace_secret.id},
+                                "data": {
+                                    "type": "STRING",
+                                    "workspace_secret_id": f"{workspace_secret_id}",
+                                },
                             }
                         ],
                         "combinator": "OR",
                     },
                 },
+                {
+                    "id": "57c31247-998a-430d-bb62-bf50eca7df35",
+                    "key": "additional_header_key",
+                    "value": {
+                        "rules": [{"type": "CONSTANT_VALUE", "data": {"type": "STRING", "value": "additional_header"}}],
+                        "combinator": "OR",
+                    },
+                },
+                {
+                    "id": "47b32274-f19b-4c15-b788-55c069c311c5",
+                    "key": "additional_header_value",
+                    "value": {
+                        "rules": [
+                            {"type": "CONSTANT_VALUE", "data": {"type": "STRING", "value": "additional header value"}}
+                        ],
+                        "combinator": "OR",
+                    },
+                },
             ],
             "data": {
                 "label": "Simple A P I Node",
@@ -149,13 +181,13 @@ def test_serialize_workflow(vellum_client):
                 "method_input_id": "fd61b5ac-39f9-4cfe-a839-f8ce78c202df",
                 "body_input_id": "c3a17ceb-e201-4025-b18a-9162aac7705e",
                 "authorization_type_input_id": "3092bf23-3202-4f3e-874c-9a33ccc73459",
-                "bearer_token_value_input_id": None,
+                "bearer_token_value_input_id": "fee5e3c9-442a-4922-ba80-5ee07361cea7",
                 "api_key_header_key_input_id": "2fcdfbc3-8095-4277-bb4a-a201fd326b54",
                 "api_key_header_value_input_id": "d794bb51-a419-4fd8-be63-dfaf4166e831",
                 "additional_headers": [
                     {
-                        "header_key_input_id": "cff297b8-8ef9-46ab-9ef4-1f1eb43a2298",
-                        "header_value_input_id": "0d41f210-f428-4849-9c21-0cde2cf54657",
+                        "header_key_input_id": "57c31247-998a-430d-bb62-bf50eca7df35",
+                        "header_value_input_id": "47b32274-f19b-4c15-b788-55c069c311c5",
                     }
                 ],
                 "text_output_id": "17342c21-12bb-49ab-88ce-f144e0376b32",
@@ -164,26 +196,11 @@ def test_serialize_workflow(vellum_client):
             },
             "display_data": {"position": {"x": 0.0, "y": 0.0}},
             "definition": {
+                "name": "SimpleAPINode",
+                "module": ["tests", "workflows", "basic_vellum_api_node", "workflow"],
                 "bases": [
-                    {
-                        "module": [
-                            "vellum",
-                            "workflows",
-                            "nodes",
-                            "displayable",
-                            "api_node",
-                            "node",
-                        ],
-                        "name": "APINode",
-                    }
-                ],
-                "module": [
-                    "tests",
-                    "workflows",
-                    "basic_vellum_api_node",
-                    "workflow",
+                    {"name": "APINode", "module": ["vellum", "workflows", "nodes", "displayable", "api_node", "node"]}
                 ],
-                "name": "SimpleAPINode",
             },
         },
         api_node,

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

@@ -32,9 +32,7 @@ from tests.workflows.basic_conditional_node.workflow_with_only_one_conditional_n
 def test_serialize_workflow():
     # GIVEN a Workflow that uses a ConditionalNode
     # WHEN we serialize it
-    workflow_display = get_workflow_display(
-        base_display_class=VellumWorkflowDisplay, workflow_class=CategoryWorkflow
-    )
+    workflow_display = get_workflow_display(base_display_class=VellumWorkflowDisplay, workflow_class=CategoryWorkflow)
 
     # TODO: Support serialization of BaseNode
     # https://app.shortcut.com/vellum/story/4871/support-serialization-of-base-node
@@ -147,9 +145,7 @@ def test_serialize_workflow():
                         "rules": [
                             {
                                 "type": "INPUT_VARIABLE",
-                                "data": {
-                                    "input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf"
-                                },
+                                "data": {"input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf"},
                             }
                         ],
                         "combinator": "OR",
@@ -175,9 +171,7 @@ def test_serialize_workflow():
                         "rules": [
                             {
                                 "type": "INPUT_VARIABLE",
-                                "data": {
-                                    "input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf"
-                                },
+                                "data": {"input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf"},
                             }
                         ],
                         "combinator": "OR",
@@ -203,9 +197,7 @@ def test_serialize_workflow():
                         "rules": [
                             {
                                 "type": "INPUT_VARIABLE",
-                                "data": {
-                                    "input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf"
-                                },
+                                "data": {"input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf"},
                             }
                         ],
                         "combinator": "OR",
@@ -231,9 +223,7 @@ def test_serialize_workflow():
                         "rules": [
                             {
                                 "type": "INPUT_VARIABLE",
-                                "data": {
-                                    "input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf"
-                                },
+                                "data": {"input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf"},
                             }
                         ],
                         "combinator": "OR",
@@ -259,9 +249,7 @@ def test_serialize_workflow():
                         "rules": [
                             {
                                 "type": "INPUT_VARIABLE",
-                                "data": {
-                                    "input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf"
-                                },
+                                "data": {"input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf"},
                             }
                         ],
                         "combinator": "OR",
@@ -287,9 +275,7 @@ def test_serialize_workflow():
                         "rules": [
                             {
                                 "type": "INPUT_VARIABLE",
-                                "data": {
-                                    "input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf"
-                                },
+                                "data": {"input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf"},
                             }
                         ],
                         "combinator": "OR",
@@ -926,15 +912,11 @@ def descriptors_with_value_and_start_and_end():
 
 
 @pytest.mark.parametrize("descriptor, operator", descriptors_with_lhs_and_rhs())
-def test_conditional_node_serialize_all_operators_with_lhs_and_rhs(
-    descriptor, operator
-):
+def test_conditional_node_serialize_all_operators_with_lhs_and_rhs(descriptor, operator):
     # GIVEN a simple workflow with one conditional node
     workflow_cls = create_simple_workflow(descriptor)
 
-    workflow_display = get_workflow_display(
-        base_display_class=VellumWorkflowDisplay, workflow_class=workflow_cls
-    )
+    workflow_display = get_workflow_display(base_display_class=VellumWorkflowDisplay, workflow_class=workflow_cls)
 
     # TODO: Support serialization of BaseNode
     # https://app.shortcut.com/vellum/story/4871/support-serialization-of-base-node
@@ -1058,9 +1040,7 @@ def test_conditional_node_serialize_all_operators_with_expression(descriptor, op
     # GIVEN a simple workflow with one conditional node
     workflow_cls = create_simple_workflow(descriptor)
 
-    workflow_display = get_workflow_display(
-        base_display_class=VellumWorkflowDisplay, workflow_class=workflow_cls
-    )
+    workflow_display = get_workflow_display(base_display_class=VellumWorkflowDisplay, workflow_class=workflow_cls)
 
     # TODO: Support serialization of BaseNode
     # https://app.shortcut.com/vellum/story/4871/support-serialization-of-base-node
@@ -1166,18 +1146,12 @@ def test_conditional_node_serialize_all_operators_with_expression(descriptor, op
     )
 
 
-@pytest.mark.parametrize(
-    "descriptor, operator", descriptors_with_value_and_start_and_end()
-)
-def test_conditional_node_serialize_all_operators_with_value_and_start_and_end(
-    descriptor, operator
-):
+@pytest.mark.parametrize("descriptor, operator", descriptors_with_value_and_start_and_end())
+def test_conditional_node_serialize_all_operators_with_value_and_start_and_end(descriptor, operator):
     # GIVEN a simple workflow with one conditional node
     workflow_cls = create_simple_workflow(descriptor)
 
-    workflow_display = get_workflow_display(
-        base_display_class=VellumWorkflowDisplay, workflow_class=workflow_cls
-    )
+    workflow_display = get_workflow_display(base_display_class=VellumWorkflowDisplay, workflow_class=workflow_cls)
 
     # TODO: Support serialization of BaseNode
     # https://app.shortcut.com/vellum/story/4871/support-serialization-of-base-node

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

@@ -29,7 +29,7 @@ def test_serialize_workflow():
                 "type": "STRING",
                 "required": True,
                 "default": None,
-                "extensions": { "color": None },
+                "extensions": {"color": None},
             },
             {
                 "id": "545ff95e-e86f-4d06-a991-602781e72605",
@@ -37,7 +37,7 @@ def test_serialize_workflow():
                 "type": "STRING",
                 "required": True,
                 "default": None,
-                "extensions": { "color": None },
+                "extensions": {"color": None},
             },
         ],
         input_variables,

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

@@ -40,7 +40,7 @@ def test_serialize_workflow():
                 "type": "STRING",
                 "required": True,
                 "default": None,
-                "extensions": { "color": None },
+                "extensions": {"color": None},
             },
             {
                 "id": "aba1e6e0-dfa7-4c15-a4e6-aec6feebfaca",
@@ -48,7 +48,7 @@ def test_serialize_workflow():
                 "type": "STRING",
                 "required": True,
                 "default": None,
-                "extensions": { "color": None },
+                "extensions": {"color": None},
             },
         ],
         input_variables,
@@ -339,9 +339,13 @@ def test_serialize_workflow():
         ignore_order=True,
     )
 
-    temperature_terminal_node = next(node for node in workflow_raw_data["nodes"][2:] if node['data']['name'] == 'temperature')
-    reasoning_terminal_node = next(node for node in workflow_raw_data["nodes"][2:] if node['data']['name'] == 'reasoning')
-    
+    temperature_terminal_node = next(
+        node for node in workflow_raw_data["nodes"][2:] if node["data"]["name"] == "temperature"
+    )
+    reasoning_terminal_node = next(
+        node for node in workflow_raw_data["nodes"][2:] if node["data"]["name"] == "reasoning"
+    )
+
     assert not DeepDiff(
         {
             "id": "31b74695-3f1c-47cf-8be8-a4d86cc589e8",
@@ -402,70 +406,70 @@ def test_serialize_workflow():
         ignore_order=True,
         # TODO: Make sure this output ID matches the workflow output ID of the subworkflow node's workflow
         # https://app.shortcut.com/vellum/story/5660/fix-output-id-in-subworkflow-nodes
-        exclude_regex_paths=r"root\['inputs'\]\[0\]\['value'\]\['rules'\]\[0\]\['data'\]\['output_id'\]"
+        exclude_regex_paths=r"root\['inputs'\]\[0\]\['value'\]\['rules'\]\[0\]\['data'\]\['output_id'\]",
     )
 
     assert not DeepDiff(
         {
-                "id": "0779b232-82ab-4dbe-a340-6a85e6ab3368",
-                "type": "TERMINAL",
-                "definition": {
-                    "bases": [
-                        {
-                            "bases": [],
-                            "module": [
-                                "vellum",
-                                "workflows",
-                                "nodes",
-                                "bases",
-                                "base",
-                            ],
-                            "name": "BaseNode",
-                        },
-                    ],
-                    "module": [
-                        "vellum",
-                        "workflows",
-                        "nodes",
-                        "displayable",
-                        "final_output_node",
-                        "node",
-                    ],
-                    "name": "FinalOutputNode",
-                },
-                "data": {
-                    "label": "Final Output",
-                    "name": "temperature",
-                    "target_handle_id": "9e077063-c394-4c7b-b0c6-e6686df67984",
-                    "output_id": "99afb757-2782-465d-ab55-80ccf50552b9",
-                    "output_type": "NUMBER",
-                    "node_input_id": "df0d7e0c-2b37-4059-91c1-0419f950b7fe",
-                },
-                "inputs": [
+            "id": "0779b232-82ab-4dbe-a340-6a85e6ab3368",
+            "type": "TERMINAL",
+            "definition": {
+                "bases": [
                     {
-                        "id": "df0d7e0c-2b37-4059-91c1-0419f950b7fe",
-                        "key": "node_input",
-                        "value": {
-                            "rules": [
-                                {
-                                    "type": "NODE_OUTPUT",
-                                    "data": {
-                                        "node_id": "080e4343-c7ce-4f82-b9dd-e94c8cc92239",
-                                        "output_id": "2fc57139-7420-49e5-96a6-dcbb3ff5d622",
-                                    },
-                                }
-                            ],
-                            "combinator": "OR",
-                        },
-                    }
+                        "bases": [],
+                        "module": [
+                            "vellum",
+                            "workflows",
+                            "nodes",
+                            "bases",
+                            "base",
+                        ],
+                        "name": "BaseNode",
+                    },
+                ],
+                "module": [
+                    "vellum",
+                    "workflows",
+                    "nodes",
+                    "displayable",
+                    "final_output_node",
+                    "node",
                 ],
-                "display_data": {"position": {"x": 0.0, "y": 0.0}},
+                "name": "FinalOutputNode",
             },
+            "data": {
+                "label": "Final Output",
+                "name": "temperature",
+                "target_handle_id": "9e077063-c394-4c7b-b0c6-e6686df67984",
+                "output_id": "99afb757-2782-465d-ab55-80ccf50552b9",
+                "output_type": "NUMBER",
+                "node_input_id": "df0d7e0c-2b37-4059-91c1-0419f950b7fe",
+            },
+            "inputs": [
+                {
+                    "id": "df0d7e0c-2b37-4059-91c1-0419f950b7fe",
+                    "key": "node_input",
+                    "value": {
+                        "rules": [
+                            {
+                                "type": "NODE_OUTPUT",
+                                "data": {
+                                    "node_id": "080e4343-c7ce-4f82-b9dd-e94c8cc92239",
+                                    "output_id": "2fc57139-7420-49e5-96a6-dcbb3ff5d622",
+                                },
+                            }
+                        ],
+                        "combinator": "OR",
+                    },
+                }
+            ],
+            "display_data": {"position": {"x": 0.0, "y": 0.0}},
+        },
         temperature_terminal_node,
         ignore_order=True,
         # TODO: Make sure this output ID matches the workflow output ID of the subworkflow node's workflow
         # https://app.shortcut.com/vellum/story/5660/fix-output-id-in-subworkflow-nodes
-        exclude_regex_paths=r"root\['inputs'\]\[0\]\['value'\]\['rules'\]\[0\]\['data'\]\['output_id'\]"
+        exclude_regex_paths=r"root\['inputs'\]\[0\]\['value'\]\['rules'\]\[0\]\['data'\]\['output_id'\]",
     )
 
     # AND each edge should be serialized correctly

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

@@ -38,7 +38,7 @@ def test_serialize_workflow():
                 "type": "JSON",
                 "required": True,
                 "default": None,
-                "extensions": { "color": None },
+                "extensions": {"color": None},
             },
         ],
         input_variables,
@@ -228,9 +228,30 @@ def test_serialize_workflow():
                     },
                 },
                 "input_variables": [
-                    {"id": "b29bb546-9bc8-4136-857d-8c7a464ba9d4", "key": "item", "type": "JSON", "required": True, "default": None, "extensions": {"color": None}},
-                    {"id": "17e7ca49-668f-450d-a792-e1f97d13db67", "key": "index", "type": "NUMBER", "required": True, "default": None, "extensions": {"color": None}},
-                    {"id": "d6fc6c7a-235f-4b98-86f3-e258d1198f93", "key": "items", "type": "JSON", "required": True, "default": None, "extensions": {"color": None}},
+                    {
+                        "id": "b29bb546-9bc8-4136-857d-8c7a464ba9d4",
+                        "key": "item",
+                        "type": "JSON",
+                        "required": True,
+                        "default": None,
+                        "extensions": {"color": None},
+                    },
+                    {
+                        "id": "17e7ca49-668f-450d-a792-e1f97d13db67",
+                        "key": "index",
+                        "type": "NUMBER",
+                        "required": True,
+                        "default": None,
+                        "extensions": {"color": None},
+                    },
+                    {
+                        "id": "d6fc6c7a-235f-4b98-86f3-e258d1198f93",
+                        "key": "items",
+                        "type": "JSON",
+                        "required": True,
+                        "default": None,
+                        "extensions": {"color": None},
+                    },
                 ],
                 "output_variables": [{"id": "2a957315-fae0-4366-8a35-f0b315c5eade", "key": "count", "type": "NUMBER"}],
                 "concurrency": None,

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

@@ -2,11 +2,12 @@ from unittest import mock
 
 from deepdiff import DeepDiff
 
-from tests.workflows.basic_merge_node.await_all_workflow import AwaitAllPassingWorkflow
 from vellum_ee.workflows.display.nodes.base_node_vellum_display import BaseNodeVellumDisplay
 from vellum_ee.workflows.display.workflows import VellumWorkflowDisplay
 from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
 
+from tests.workflows.basic_merge_node.await_all_workflow import AwaitAllPassingWorkflow
+
 
 def test_serialize_workflow__await_all():
     # GIVEN a Workflow that uses an await all merge node

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

@@ -48,7 +48,7 @@ def test_serialize_workflow(vellum_client):
                 "type": "STRING",
                 "required": True,
                 "default": None,
-                "extensions": { "color": None },
+                "extensions": {"color": None},
             },
             {
                 "id": "aa3ca842-250c-4a3f-853f-23928c28d0f8",
@@ -56,7 +56,7 @@ def test_serialize_workflow(vellum_client):
                 "type": "STRING",
                 "required": True,
                 "default": None,
-                "extensions": { "color": None },
+                "extensions": {"color": None},
             },
         ],
         input_variables,

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

@@ -48,7 +48,7 @@ def test_serialize_workflow(vellum_client):
                 "type": "STRING",
                 "required": True,
                 "default": None,
-                "extensions": { "color": None },
+                "extensions": {"color": None},
             },
             {
                 "id": "19a78824-9a98-4ae8-a1fc-61f81a422a17",
@@ -56,7 +56,7 @@ def test_serialize_workflow(vellum_client):
                 "type": "STRING",
                 "required": True,
                 "default": None,
-                "extensions": { "color": None },
+                "extensions": {"color": None},
             },
         ],
         input_variables,

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

@@ -29,7 +29,7 @@ def test_serialize_workflow():
             "type": "STRING",
             "required": True,
             "default": None,
-            "extensions": { "color": None },
+            "extensions": {"color": None},
         }
     ]
 

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

@@ -2,11 +2,12 @@ from unittest import mock
 
 from deepdiff import DeepDiff
 
-from tests.workflows.basic_try_node.workflow import SimpleTryExample
 from vellum_ee.workflows.display.nodes.base_node_vellum_display import BaseNodeVellumDisplay
 from vellum_ee.workflows.display.workflows import VellumWorkflowDisplay
 from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
 
+from tests.workflows.basic_try_node.workflow import SimpleTryExample
+
 
 def test_serialize_workflow():
     # GIVEN a Workflow with a TryNode

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

@@ -43,7 +43,7 @@ def test_serialize_workflow__missing_final_output_node():
                 "type": "STRING",
                 "required": True,
                 "default": None,
-                "extensions": { "color": None },
+                "extensions": {"color": None},
             },
             {
                 "id": "a8b6c5d4-a0e9-4457-834b-46b633c466a6",
@@ -51,7 +51,7 @@ def test_serialize_workflow__missing_final_output_node():
                 "type": "STRING",
                 "required": True,
                 "default": None,
-                "extensions": { "color": None },
+                "extensions": {"color": None},
             },
         ],
         input_variables,

vellum_ee/workflows/display/types.py

@@ -1,6 +1,10 @@
 from dataclasses import dataclass, field
 from typing import TYPE_CHECKING, Dict, Generic, Tuple, Type, TypeVar
 
+from vellum.workflows.descriptors.base import BaseDescriptor
+from vellum.workflows.nodes import BaseNode
+from vellum.workflows.ports import Port
+from vellum.workflows.references import OutputReference, WorkflowInputReference
 from vellum_ee.workflows.display.base import (
     EdgeDisplayType,
     EntrypointDisplayType,
@@ -9,10 +13,6 @@ from vellum_ee.workflows.display.base import (
     WorkflowOutputDisplayType,
 )
 from vellum_ee.workflows.display.nodes.types import NodeOutputDisplay, PortDisplay
-from vellum.workflows.descriptors.base import BaseDescriptor
-from vellum.workflows.nodes import BaseNode
-from vellum.workflows.ports import Port
-from vellum.workflows.references import OutputReference, WorkflowInputReference
 
 if TYPE_CHECKING:
     from vellum_ee.workflows.display.nodes.base_node_display import BaseNodeDisplay

vellum_ee/workflows/display/utils/vellum.py

@@ -4,9 +4,7 @@ import typing
 from typing import Any, List, Union, cast
 
 from vellum import ChatMessage, SearchResult, SearchResultRequest, VellumVariableType
-from vellum.client.types.array_vellum_value import ArrayVellumValue
 from vellum.workflows.descriptors.base import BaseDescriptor
-from vellum.workflows.nodes.core.map_node.node import MapNodeItemType
 from vellum.workflows.references import OutputReference, WorkflowInputReference
 from vellum.workflows.references.execution_count import ExecutionCountReference
 from vellum.workflows.references.node import NodeReference
@@ -48,7 +46,7 @@ def infer_vellum_variable_type(value: Any) -> VellumVariableType:
                     f"Expected NodeReference {descriptor.name} to have an instance pointing to a descriptor"
                 )
             descriptor = descriptor.instance
-        
+
         inferred_type = primitive_type_to_vellum_variable_type(descriptor)
     else:
         vellum_variable_value = primitive_to_vellum_value(value)
@@ -68,9 +66,7 @@ def create_node_input_value_pointer_rule(
         )
     if isinstance(value, WorkflowInputReference):
         workflow_input_display = display_context.workflow_input_displays[value]
-        return InputVariablePointer(
-            data=InputVariableData(input_variable_id=str(workflow_input_display.id))
-        )
+        return InputVariablePointer(data=InputVariableData(input_variable_id=str(workflow_input_display.id)))
     if isinstance(value, VellumSecretReference):
         # TODO: Pass through the name instead of retrieving the ID
         # https://app.shortcut.com/vellum/story/5072

vellum_ee/workflows/display/vellum.py

@@ -202,7 +202,7 @@ class InputVariablePointer(UniversalBaseModel):
 
 class WorkspaceSecretData(UniversalBaseModel):
     type: VellumVariableType
-    workspace_secret_id: str
+    workspace_secret_id: Optional[str] = None
 
 
 class WorkspaceSecretPointer(UniversalBaseModel):

vellum_ee/workflows/display/workflows/get_vellum_workflow_display_class.py

@@ -24,4 +24,7 @@ def get_workflow_display(
         except IndexError:
             return base_display_class(workflow_class)
 
-    return workflow_display_class(workflow_class, parent_display_context=parent_display_context) # type: ignore[return-value]
+    return workflow_display_class(  # type: ignore[return-value]
+        workflow_class,
+        parent_display_context=parent_display_context,
+    )