vellum-ai 0.10.1__py3-none-any.whl → 0.10.2__py3-none-any.whl

vellum/client/core/client_wrapper.py

@@ -17,7 +17,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.10.1",
+            "X-Fern-SDK-Version": "0.10.2",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

{vellum_ai-0.10.1.dist-info → vellum_ai-0.10.2.dist-info}/METADATA

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

{vellum_ai-0.10.1.dist-info → vellum_ai-0.10.2.dist-info}/RECORD

@@ -1,12 +1,12 @@
-vellum_cli/__init__.py,sha256=i9jXzrSFcTGQJ5SgOv7_BZaMTgX_CSmNr5JoFIaTvJs,2204
+vellum_cli/__init__.py,sha256=t62kIq0vDia8TTJqAmqFpYCo40buWNDQmjIDTOvmwwE,2416
 vellum_cli/aliased_group.py,sha256=ugW498j0yv4ALJ8vS9MsO7ctDW7Jlir9j6nE_uHAP8c,3363
 vellum_cli/config.py,sha256=urqMGQUkTntzdx-JV0P32FZVbqRZqgI-yFVmGPVblNI,3613
 vellum_cli/image_push.py,sha256=lCQhkQu-IlIMHNSynmEl5KUzO_dHW27NTAFWnyvPpcs,4419
 vellum_cli/logger.py,sha256=PuRFa0WCh4sAGFS5aqWB0QIYpS6nBWwPJrIXpWxugV4,1022
-vellum_cli/pull.py,sha256=nIjb0Ejv9IXDzxjTKj-G8J4V6wCh2gyd0AwXH_lW6TE,3065
+vellum_cli/pull.py,sha256=op8HQmtiKa4TlrgEvhLiIac0Zhal6LETW22rhSkNHGE,3483
 vellum_cli/push.py,sha256=HhXHqLvWKf3CUZJ3yIm-KeoBkgk73EXqVtGgWYcBcVk,5089
 vellum_cli/tests/test_config.py,sha256=uvKGDc8BoVyT9_H0Z-g8469zVxomn6Oi3Zj-vK7O_wU,2631
-vellum_cli/tests/test_pull.py,sha256=osrjSMVs61rsfw6Ui7ZnzxSX4-VI64vKOQS7BY8zfIo,5126
+vellum_cli/tests/test_pull.py,sha256=24sHYWNViS-G_UHmPC_-KXb6l3OOs2AiDGv2kKp4AHY,5815
 vellum_ee/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum_ee/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum_ee/workflows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -23,7 +23,7 @@ vellum_ee/workflows/display/nodes/utils.py,sha256=sloya5TpXsnot1HURc9L51INwflRqU
 vellum_ee/workflows/display/nodes/vellum/__init__.py,sha256=WQ_KbMp__S41AUFDHt9dZU6IQLK9AizdGMi2M6eK2_8,1362
 vellum_ee/workflows/display/nodes/vellum/api_node.py,sha256=5mMO1rKDafoPtnFZciuFGGfbu8Ei-j4zRiEDx9oQG4I,8572
 vellum_ee/workflows/display/nodes/vellum/code_execution_node.py,sha256=VWFzPYzzNy4dSju0Gyegxx81A9cYjkN-dYqPTK3hCr4,3403
-vellum_ee/workflows/display/nodes/vellum/conditional_node.py,sha256=-DoU5wdR4rYYxkCXBGE0re0iOqLTOV-JY4D9UDEYoh8,10130
+vellum_ee/workflows/display/nodes/vellum/conditional_node.py,sha256=3pVtCukiILTSZPUWuYLIwzwXGBSSUV4q--i0zlYxGKQ,10850
 vellum_ee/workflows/display/nodes/vellum/final_output_node.py,sha256=kR_z9ichhBtK_CLetqGjewmFmIFalfSiG2tQlM2T4yM,2772
 vellum_ee/workflows/display/nodes/vellum/guardrail_node.py,sha256=CiGZL1nxgK0KxLUgKilpollRYFIqqGHjSGXtvP1miqs,2168
 vellum_ee/workflows/display/nodes/vellum/inline_prompt_node.py,sha256=NuNoPnvwbdSRkTQA3clLySNSzqkK0nmU4L0zYqgIfFQ,7397
@@ -42,7 +42,7 @@ vellum_ee/workflows/display/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeR
 vellum_ee/workflows/display/tests/workflow_serialization/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_api_node_serialization.py,sha256=bLdJ0VVwAPDRvwaJMiMf0fRCuIWvvuRjAxKzddGPFyA,16871
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_code_execution_node_serialization.py,sha256=JnaxdIDMudwgO4zzB0uJlCDYvoD1lZaAxFQ9OYxUFoA,22716
-vellum_ee/workflows/display/tests/workflow_serialization/test_basic_conditional_node_serialization.py,sha256=ojMpjIq89GqMN8rDSGgjNDReYOD2WvAofRooKVv7TU8,50681
+vellum_ee/workflows/display/tests/workflow_serialization/test_basic_conditional_node_serialization.py,sha256=0NxRo7KM_z8A1vgRgPISjCVFbYyKaeWCkgNldCkF6XY,53075
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_guardrail_node_serialization.py,sha256=FNmGO1hQjQOOejfSY-ErcMJXxvrxMyR7GZq4UX26UHA,8271
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_inline_subworkflow_serialization.py,sha256=2X8MzxZ3rf1lkx_e7Oez-qp4fvPZrdjulGAEOhbCMnY,21546
 vellum_ee/workflows/display/tests/workflow_serialization/test_basic_map_node_serialization.py,sha256=ESalw3PsfGy42w85FJr9VDWg8Rpv9NT25OMHpye27bs,14967
@@ -68,7 +68,7 @@ vellum/client/README.md,sha256=8cKUE1kSuhz-U87n3-clA7o9_zlP0Avr4qKbvRbdT2k,5984
 vellum/client/__init__.py,sha256=Fu-wzw_MiTtqAQOFhcFcrLKAPkEfUhf6K4ZuFspfKUw,115305
 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=KTpqKvCVAPkVRUzzAODEnMHWvg_Pq-qDtpBSZW5qBxU,1890
+vellum/client/core/client_wrapper.py,sha256=mQRFKZD2WpUTWabsoFG3brPhS2qJ4398Akys9PcuMV4,1890
 vellum/client/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/client/core/file.py,sha256=X9IbmkZmB2bB_DpmZAO3crWdXagOakAyn6UCOCImCPg,2322
 vellum/client/core/http_client.py,sha256=R0pQpCppnEtxccGvXl4uJ76s7ro_65Fo_erlNNLp_AI,19228
@@ -1343,8 +1343,8 @@ vellum/workflows/utils/vellum_variables.py,sha256=JA7hhbJ_9m7KEYFaCv5R6FqYeTRWDp
 vellum/workflows/vellum_client.py,sha256=ODrq_TSl-drX2aezXegf7pizpWDVJuTXH-j6528t75s,683
 vellum/workflows/workflows/__init__.py,sha256=KY45TqvavCCvXIkyCFMEc0dc6jTMOUci93U2DUrlZYc,66
 vellum/workflows/workflows/base.py,sha256=NbQ3jR9veKdfK5S9qPkiJvy16l19YyQxzops_Y09wMw,12872
-vellum_ai-0.10.1.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
-vellum_ai-0.10.1.dist-info/METADATA,sha256=O42wdOBlrjLWVYPaen5pKq9eyquqvkNWdl1-m3TjrI4,5011
-vellum_ai-0.10.1.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-vellum_ai-0.10.1.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
-vellum_ai-0.10.1.dist-info/RECORD,,
+vellum_ai-0.10.2.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
+vellum_ai-0.10.2.dist-info/METADATA,sha256=UPsNCcm_oVqPV4zmvlxPX4hSC_VUjpPJ7z5LcCF4rOY,5011
+vellum_ai-0.10.2.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+vellum_ai-0.10.2.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
+vellum_ai-0.10.2.dist-info/RECORD,,

vellum_cli/__init__.py

@@ -43,9 +43,10 @@ def push(
 @main.command()
 @click.argument("module", required=False)
 @click.option("--legacy-module", is_flag=True, help="Pull the workflow as a legacy module")
-def pull(module: Optional[str], legacy_module: Optional[bool]) -> None:
+@click.option("--include-json", is_flag=True, help="Include the JSON representation of the Workflow in the pull response. Should only be used for debugging purposes.")
+def pull(module: Optional[str], legacy_module: Optional[bool], include_json: Optional[bool]) -> None:
     """Pull Workflow from Vellum"""
-    pull_command(module, legacy_module)
+    pull_command(module, legacy_module, include_json)
 
 
 @main.group(aliases=["images", "image"])

vellum_cli/pull.py

@@ -6,12 +6,14 @@ from typing import Optional
 
 from dotenv import load_dotenv
 
+from vellum.workflows.vellum_client import create_vellum_client
 from vellum_cli.config import load_vellum_cli_config
 from vellum_cli.logger import load_cli_logger
-from vellum.workflows.vellum_client import create_vellum_client
 
 
-def pull_command(module: Optional[str], legacy_module: Optional[bool] = None) -> None:
+def pull_command(
+    module: Optional[str], legacy_module: Optional[bool] = None, include_json: Optional[bool] = None
+) -> None:
     load_dotenv()
     logger = load_cli_logger()
     config = load_vellum_cli_config()
@@ -22,7 +24,9 @@ def pull_command(module: Optional[str], legacy_module: Optional[bool] = None) ->
     if len(config.workflows) > 1 and not module:
         raise ValueError("Multiple workflows found in project to pull. Pulling only a single workflow is supported.")
 
-    workflow_config = next((w for w in config.workflows if w.module == module), None) if module else config.workflows[0]
+    workflow_config = (
+        next((w for w in config.workflows if w.module == module), None) if module else config.workflows[0]
+    )
     if workflow_config is None:
         raise ValueError(f"No workflow config for '{module}' found in project to push.")
 
@@ -31,9 +35,15 @@ def pull_command(module: Optional[str], legacy_module: Optional[bool] = None) ->
 
     logger.info(f"Pulling workflow into {workflow_config.module}")
     client = create_vellum_client()
+    query_parameters = {}
+    if legacy_module:
+        query_parameters["legacyModule"] = legacy_module
+    if include_json:
+        query_parameters["include_json"] = include_json
+
     response = client.workflows.pull(
         workflow_config.workflow_sandbox_id,
-        request_options={"additional_query_parameters": {"legacyModule": legacy_module} if legacy_module else {}},
+        request_options={"additional_query_parameters": query_parameters},
     )
 
     zip_bytes = b"".join(response)
@@ -70,4 +80,7 @@ def pull_command(module: Optional[str], legacy_module: Optional[bool] = None) ->
                 logger.info(f"Writing to {target_file}...")
                 target.write(source.read().decode("utf-8"))
 
+    if include_json:
+        logger.warning("The pulled JSON representation of the Workflow should be used for debugging purposely only. Its schema should be considered unstable and subject to change at any time.")
+
     logger.info(f"Successfully pulled Workflow into {workflow_config.module}")

vellum_cli/tests/test_pull.py

@@ -150,3 +150,21 @@ def test_pull__remove_missing_files__ignore_pattern(vellum_client, mock_module):
 
     # AND the tests/test_workflow.py file is untouched
     assert os.path.exists(test_file_path)
+
+
+def test_pull__include_json(vellum_client, mock_module):
+    # GIVEN a module on the user's filesystem
+    _, module = mock_module
+
+    # AND the workflow pull API call returns a zip file
+    vellum_client.workflows.pull.return_value = iter(
+        [zip_file_map({"workflow.py": "print('hello')", "workflow.json": "{}"})]
+    )
+
+    # WHEN the user runs the pull command
+    pull_command(module, include_json=True)
+
+    # THEN the pull api is called with include_json=True
+    vellum_client.workflows.pull.assert_called_once()
+    call_args = vellum_client.workflows.pull.call_args.kwargs
+    assert call_args["request_options"]["additional_query_parameters"] == {"include_json": True}

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

@@ -40,10 +40,16 @@ class RuleIdMap:
     lhs: Optional["RuleIdMap"]
     rhs: Optional["RuleIdMap"]
 
+@dataclass
+class ConditionId:
+    id: UUID
+    rule_group_id: UUID
+
 
 class BaseConditionalNodeDisplay(BaseNodeVellumDisplay[_ConditionalNodeType], Generic[_ConditionalNodeType]):
     source_handle_ids: ClassVar[Dict[int, UUID]]
     rule_ids: ClassVar[List[RuleIdMap]]
+    condition_ids: ClassVar[list[ConditionId]]
 
     def serialize(self, display_context: WorkflowDisplayContext, **kwargs: Any) -> JsonObject:
         node = self._node
@@ -51,6 +57,13 @@ class BaseConditionalNodeDisplay(BaseNodeVellumDisplay[_ConditionalNodeType], Ge
 
         node_inputs: List[NodeInput] = []
         source_handle_ids = self._get_source_handle_ids() or {}
+        condition_ids = self._get_condition_ids()
+
+        ports_size = sum(1 for _ in node.Ports)
+
+        if len(condition_ids) > ports_size:
+            raise ValueError(
+                f"Too many defined condition ids. Ports are size {ports_size} but the defined conditions have length {len(condition_ids)}")
 
         def serialize_rule(
             descriptor: BaseDescriptor, path: List[int], rule_id_map: Optional[RuleIdMap]
@@ -132,7 +145,9 @@ class BaseConditionalNodeDisplay(BaseNodeVellumDisplay[_ConditionalNodeType], Ge
             if port._condition is None or port._condition_type is None:
                 continue
 
-            condition_id = str(uuid4_from_hash(f"{node_id}|conditions|{idx}"))
+
+            condition_id = str(condition_ids[idx].id if condition_ids else uuid4_from_hash(f"{node_id}|conditions|{idx}"))
+            rule_group_id = str(condition_ids[idx].rule_group_id if condition_ids else uuid4_from_hash(f"{condition_id}|rule_group"))
             source_handle_id = str(source_handle_ids.get(idx) or uuid4_from_hash(f"{node_id}|handles|{idx}"))
 
             rule_ids = self._get_rule_ids()
@@ -145,7 +160,7 @@ class BaseConditionalNodeDisplay(BaseNodeVellumDisplay[_ConditionalNodeType], Ge
                     "type": port._condition_type.value,
                     "source_handle_id": source_handle_id,
                     "data": {
-                        "id": str(uuid4_from_hash(f"{condition_id}|rule_group")),
+                        "id": rule_group_id,
                         "rules": rules,
                         "combinator": "AND",
                         "negated": False,
@@ -215,3 +230,6 @@ class BaseConditionalNodeDisplay(BaseNodeVellumDisplay[_ConditionalNodeType], Ge
 
     def _get_rule_ids(self) -> List[RuleIdMap]:
         return self._get_explicit_node_display_attr("rule_ids", List[RuleIdMap]) or []
+
+    def _get_condition_ids(self)-> List[ConditionId]:
+        return self._get_explicit_node_display_attr("condition_ids", List[ConditionId]) or []

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

@@ -4,10 +4,16 @@ from unittest import mock
 from deepdiff import DeepDiff
 
 from tests.workflows.basic_conditional_node.workflow import CategoryWorkflow
-from tests.workflows.basic_conditional_node.workflow_with_only_one_conditional_node import create_simple_workflow
-from vellum_ee.workflows.display.nodes.base_node_vellum_display import BaseNodeVellumDisplay
+from tests.workflows.basic_conditional_node.workflow_with_only_one_conditional_node import (
+    create_simple_workflow,
+)
+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 vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import (
+    get_workflow_display,
+)
 from vellum.workflows.expressions.begins_with import BeginsWithExpression
 from vellum.workflows.expressions.between import BetweenExpression
 from vellum.workflows.expressions.contains import ContainsExpression
@@ -18,12 +24,16 @@ from vellum.workflows.expressions.does_not_equal import DoesNotEqualExpression
 from vellum.workflows.expressions.ends_with import EndsWithExpression
 from vellum.workflows.expressions.equals import EqualsExpression
 from vellum.workflows.expressions.greater_than import GreaterThanExpression
-from vellum.workflows.expressions.greater_than_or_equal_to import GreaterThanOrEqualToExpression
+from vellum.workflows.expressions.greater_than_or_equal_to import (
+    GreaterThanOrEqualToExpression,
+)
 from vellum.workflows.expressions.in_ import InExpression
 from vellum.workflows.expressions.is_not_null import IsNotNullExpression
 from vellum.workflows.expressions.is_null import IsNullExpression
 from vellum.workflows.expressions.less_than import LessThanExpression
-from vellum.workflows.expressions.less_than_or_equal_to import LessThanOrEqualToExpression
+from vellum.workflows.expressions.less_than_or_equal_to import (
+    LessThanOrEqualToExpression,
+)
 from vellum.workflows.expressions.not_between import NotBetweenExpression
 from vellum.workflows.expressions.not_in import NotInExpression
 
@@ -31,7 +41,9 @@ from vellum.workflows.expressions.not_in import NotInExpression
 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
@@ -69,11 +81,31 @@ def test_serialize_workflow():
     assert len(output_variables) == 5
     assert not DeepDiff(
         [
-            {"id": "c05f7d96-59a0-4d58-93d7-d451afd3f630", "key": "question", "type": "STRING"},
-            {"id": "93f2cb75-6fa2-4e46-9488-c0bcd29153c0", "key": "compliment", "type": "STRING"},
-            {"id": "f936ae31-ba15-4864-8961-86231022a4d7", "key": "complaint", "type": "STRING"},
-            {"id": "cdbe2adf-9951-409a-b9a8-b8b349037f4f", "key": "statement", "type": "STRING"},
-            {"id": "62ad462f-f819-4940-99ab-b3f145507f57", "key": "fallthrough", "type": "STRING"},
+            {
+                "id": "c05f7d96-59a0-4d58-93d7-d451afd3f630",
+                "key": "question",
+                "type": "STRING",
+            },
+            {
+                "id": "93f2cb75-6fa2-4e46-9488-c0bcd29153c0",
+                "key": "compliment",
+                "type": "STRING",
+            },
+            {
+                "id": "f936ae31-ba15-4864-8961-86231022a4d7",
+                "key": "complaint",
+                "type": "STRING",
+            },
+            {
+                "id": "cdbe2adf-9951-409a-b9a8-b8b349037f4f",
+                "key": "statement",
+                "type": "STRING",
+            },
+            {
+                "id": "62ad462f-f819-4940-99ab-b3f145507f57",
+                "key": "fallthrough",
+                "type": "STRING",
+            },
         ],
         output_variables,
         ignore_order=True,
@@ -124,7 +156,9 @@ 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",
@@ -134,7 +168,12 @@ def test_serialize_workflow():
                     "id": "1fb4cf46-f8b3-418f-be30-f7ec57f92285",
                     "key": "708bb538-4c77-4ae9-8e87-0706346e2947.value",
                     "value": {
-                        "rules": [{"type": "CONSTANT_VALUE", "data": {"type": "STRING", "value": "question"}}],
+                        "rules": [
+                            {
+                                "type": "CONSTANT_VALUE",
+                                "data": {"type": "STRING", "value": "question"},
+                            }
+                        ],
                         "combinator": "OR",
                     },
                 },
@@ -145,7 +184,9 @@ 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",
@@ -155,7 +196,12 @@ def test_serialize_workflow():
                     "id": "40957176-de6e-4131-bfa7-55c633312af0",
                     "key": "ddee5d1d-46e9-4ae8-b0a8-311747ebadd4.value",
                     "value": {
-                        "rules": [{"type": "CONSTANT_VALUE", "data": {"type": "STRING", "value": "complaint"}}],
+                        "rules": [
+                            {
+                                "type": "CONSTANT_VALUE",
+                                "data": {"type": "STRING", "value": "complaint"},
+                            }
+                        ],
                         "combinator": "OR",
                     },
                 },
@@ -166,7 +212,9 @@ 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",
@@ -176,7 +224,12 @@ def test_serialize_workflow():
                     "id": "93f06582-aff7-4ce5-8c60-f923090ffebc",
                     "key": "73157578-205a-4816-8985-cf726063647c.value",
                     "value": {
-                        "rules": [{"type": "CONSTANT_VALUE", "data": {"type": "STRING", "value": "compliment"}}],
+                        "rules": [
+                            {
+                                "type": "CONSTANT_VALUE",
+                                "data": {"type": "STRING", "value": "compliment"},
+                            }
+                        ],
                         "combinator": "OR",
                     },
                 },
@@ -187,7 +240,9 @@ 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",
@@ -197,7 +252,12 @@ def test_serialize_workflow():
                     "id": "e759091b-3609-4581-9014-5f46f438a4c9",
                     "key": "e805add5-7f7f-443d-b9bc-11ad15eeb49c.value",
                     "value": {
-                        "rules": [{"type": "CONSTANT_VALUE", "data": {"type": "STRING", "value": "statement"}}],
+                        "rules": [
+                            {
+                                "type": "CONSTANT_VALUE",
+                                "data": {"type": "STRING", "value": "statement"},
+                            }
+                        ],
                         "combinator": "OR",
                     },
                 },
@@ -208,7 +268,9 @@ 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",
@@ -218,7 +280,12 @@ def test_serialize_workflow():
                     "id": "e915cd85-ae55-48be-b31c-f2285db9db10",
                     "key": "f47d72ff-665f-4143-ada3-6fa66f5bda42.value",
                     "value": {
-                        "rules": [{"type": "CONSTANT_VALUE", "data": {"type": "STRING", "value": "statement"}}],
+                        "rules": [
+                            {
+                                "type": "CONSTANT_VALUE",
+                                "data": {"type": "STRING", "value": "statement"},
+                            }
+                        ],
                         "combinator": "OR",
                     },
                 },
@@ -229,7 +296,9 @@ 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",
@@ -239,7 +308,12 @@ def test_serialize_workflow():
                     "id": "e5d75ae4-cd46-437e-9695-9df2d79578b4",
                     "key": "d3359d60-9bb4-4c6e-8009-b7ea46ab28a7.value",
                     "value": {
-                        "rules": [{"type": "CONSTANT_VALUE", "data": {"type": "STRING", "value": "statement"}}],
+                        "rules": [
+                            {
+                                "type": "CONSTANT_VALUE",
+                                "data": {"type": "STRING", "value": "statement"},
+                            }
+                        ],
                         "combinator": "OR",
                     },
                 },
@@ -377,8 +451,11 @@ def test_serialize_workflow():
             },
             "display_data": {"position": {"x": 0.0, "y": 0.0}},
             "definition": {
+                "name": "CategoryConditionalNode",
+                "module": ["tests", "workflows", "basic_conditional_node", "workflow"],
                 "bases": [
                     {
+                        "name": "ConditionalNode",
                         "module": [
                             "vellum",
                             "workflows",
@@ -387,16 +464,8 @@ def test_serialize_workflow():
                             "conditional_node",
                             "node",
                         ],
-                        "name": "ConditionalNode",
                     }
                 ],
-                "module": [
-                    "tests",
-                    "workflows",
-                    "basic_conditional_node",
-                    "workflow",
-                ],
-                "name": "CategoryConditionalNode",
             },
         },
         conditional_node,
@@ -860,11 +929,15 @@ 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
@@ -894,7 +967,12 @@ def test_conditional_node_serialize_all_operators_with_lhs_and_rhs(descriptor, o
                     "id": "2262b7b4-a2f2-408b-9d4d-362940ca1ed3",
                     "key": "abe7afac-952f-4cfc-ab07-47b47f34105f.field",
                     "value": {
-                        "rules": [{"type": "CONSTANT_VALUE", "data": {"type": "STRING", "value": "123"}}],
+                        "rules": [
+                            {
+                                "type": "CONSTANT_VALUE",
+                                "data": {"type": "STRING", "value": "123"},
+                            }
+                        ],
                         "combinator": "OR",
                     },
                 },
@@ -902,7 +980,12 @@ def test_conditional_node_serialize_all_operators_with_lhs_and_rhs(descriptor, o
                     "id": "aadade8a-c253-483a-8620-31fe8171c0fd",
                     "key": "abe7afac-952f-4cfc-ab07-47b47f34105f.value",
                     "value": {
-                        "rules": [{"type": "CONSTANT_VALUE", "data": {"type": "STRING", "value": "123"}}],
+                        "rules": [
+                            {
+                                "type": "CONSTANT_VALUE",
+                                "data": {"type": "STRING", "value": "123"},
+                            }
+                        ],
                         "combinator": "OR",
                     },
                 },
@@ -972,7 +1055,9 @@ 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
@@ -1002,7 +1087,12 @@ def test_conditional_node_serialize_all_operators_with_expression(descriptor, op
                     "id": "2262b7b4-a2f2-408b-9d4d-362940ca1ed3",
                     "key": "abe7afac-952f-4cfc-ab07-47b47f34105f.field",
                     "value": {
-                        "rules": [{"type": "CONSTANT_VALUE", "data": {"type": "STRING", "value": "123"}}],
+                        "rules": [
+                            {
+                                "type": "CONSTANT_VALUE",
+                                "data": {"type": "STRING", "value": "123"},
+                            }
+                        ],
                         "combinator": "OR",
                     },
                 }
@@ -1067,12 +1157,18 @@ 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
@@ -1102,7 +1198,12 @@ def test_conditional_node_serialize_all_operators_with_value_and_start_and_end(d
                     "id": "2262b7b4-a2f2-408b-9d4d-362940ca1ed3",
                     "key": "abe7afac-952f-4cfc-ab07-47b47f34105f.field",
                     "value": {
-                        "rules": [{"type": "CONSTANT_VALUE", "data": {"type": "STRING", "value": "123"}}],
+                        "rules": [
+                            {
+                                "type": "CONSTANT_VALUE",
+                                "data": {"type": "STRING", "value": "123"},
+                            }
+                        ],
                         "combinator": "OR",
                     },
                 },
@@ -1110,7 +1211,12 @@ def test_conditional_node_serialize_all_operators_with_value_and_start_and_end(d
                     "id": "aadade8a-c253-483a-8620-31fe8171c0fd",
                     "key": "abe7afac-952f-4cfc-ab07-47b47f34105f.value",
                     "value": {
-                        "rules": [{"type": "CONSTANT_VALUE", "data": {"type": "STRING", "value": "123,123"}}],
+                        "rules": [
+                            {
+                                "type": "CONSTANT_VALUE",
+                                "data": {"type": "STRING", "value": "123,123"},
+                            }
+                        ],
                         "combinator": "OR",
                     },
                 },