vellum-workflow-server 1.1.4__tar.gz → 1.2.0.post1__tar.gz

{vellum_workflow_server-1.1.4 → vellum_workflow_server-1.2.0.post1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 1.1.4
+Version: 1.2.0.post1
 Summary: 
 License: AGPL
 Requires-Python: >=3.9.0,<4
@@ -29,7 +29,7 @@ Requires-Dist: pyjwt (==2.10.0)
 Requires-Dist: python-dotenv (==1.0.1)
 Requires-Dist: retrying (==1.3.4)
 Requires-Dist: sentry-sdk[flask] (==2.20.0)
-Requires-Dist: vellum-ai (==1.1.4)
+Requires-Dist: vellum-ai (==1.2.0)
 Description-Content-Type: text/markdown
 
 # Vellum Workflow Runner Server

{vellum_workflow_server-1.1.4 → vellum_workflow_server-1.2.0.post1}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-workflow-server"
 
 [tool.poetry]
 name = "vellum-workflow-server"
-version = "1.1.4"
+version = "1.2.0.post1"
 description = ""
 readme = "README.md"
 authors = []
@@ -45,7 +45,7 @@ flask = "2.3.3"
 orderly-set = "5.2.2"
 pebble = "5.0.7"
 gunicorn = "23.0.0"
-vellum-ai = "1.1.4"
+vellum-ai = "1.2.0"
 python-dotenv = "1.0.1"
 retrying = "1.3.4"
 sentry-sdk = {extras = ["flask"], version = "2.20.0"}

{vellum_workflow_server-1.1.4 → vellum_workflow_server-1.2.0.post1}/src/workflow_server/api/tests/test_workflow_view.py

@@ -59,11 +59,11 @@ class TestNode(BaseNode):
         "base": {"module": ["vellum", "workflows", "nodes", "bases", "base"], "name": "BaseNode"},
         "definition": {"module": ["vellum_custom_nodes", "test_node"], "name": "TestNode"},
         "display_data": {
-            "comment": {"value": "A test node for processing data."},
+            "comment": {"expanded": True, "value": "A test node for processing data."},
             "position": {"x": 0.0, "y": 0.0},
         },
         "id": "7a8b251d-f5ca-462a-b293-071d219460fb",
-        "label": "TestNode",
+        "label": "Test Node",
         "outputs": [],
         "ports": [{"id": "a3a0eefd-45d0-4f13-8c58-a836a9f7f9ed", "name": "default", "type": "DEFAULT"}],
         "trigger": {"id": "a022e36c-9852-4772-9be3-3c6c147fd811", "merge_behavior": "AWAIT_ATTRIBUTES"},
@@ -122,9 +122,12 @@ class SomeOtherNode(BaseNode):
         "attributes": [],
         "base": {"module": ["vellum", "workflows", "nodes", "bases", "base"], "name": "BaseNode"},
         "definition": {"module": ["vellum_custom_nodes", "first_node"], "name": "SomeNode"},
-        "display_data": {"comment": {"value": "This is Some Node."}, "position": {"x": 0.0, "y": 0.0}},
+        "display_data": {
+            "comment": {"expanded": True, "value": "This is Some Node."},
+            "position": {"x": 0.0, "y": 0.0},
+        },
         "id": "1e559c2e-db82-41f0-9ceb-5e89b0c5a0a3",
-        "label": "SomeNode",
+        "label": "Some Node",
         "outputs": [],
         "ports": [{"id": "48e39e97-5fd4-471e-b4f2-51d3baf06456", "name": "default", "type": "DEFAULT"}],
         "trigger": {"id": "e3381fb7-61fc-4c46-ae8e-51fc463b6a59", "merge_behavior": "AWAIT_ATTRIBUTES"},
@@ -142,9 +145,12 @@ class SomeOtherNode(BaseNode):
         "attributes": [],
         "base": {"module": ["vellum", "workflows", "nodes", "bases", "base"], "name": "BaseNode"},
         "definition": {"module": ["vellum_custom_nodes", "second_node"], "name": "SomeOtherNode"},
-        "display_data": {"comment": {"value": "This is Some Other Node."}, "position": {"x": 0.0, "y": 0.0}},
+        "display_data": {
+            "comment": {"expanded": True, "value": "This is Some Other Node."},
+            "position": {"x": 0.0, "y": 0.0},
+        },
         "id": "7aee541b-b245-4c8a-9137-3e4631d5100c",
-        "label": "SomeOtherNode",
+        "label": "Some Other Node",
         "outputs": [],
         "ports": [{"id": "fb66b46a-d970-4bc9-83ea-70c154c57ddd", "name": "default", "type": "DEFAULT"}],
         "trigger": {"id": "13fa2714-20b3-4bc3-ab79-621a188e3bfa", "merge_behavior": "AWAIT_ATTRIBUTES"},
@@ -211,9 +217,12 @@ class HelperClass:
             "attributes": [],
             "base": {"module": ["vellum", "workflows", "nodes", "bases", "base"], "name": "BaseNode"},
             "definition": {"module": ["vellum_custom_nodes", "multiple_nodes"], "name": "ProcessingNode"},
-            "display_data": {"comment": {"value": "Processes input data."}, "position": {"x": 0.0, "y": 0.0}},
+            "display_data": {
+                "comment": {"expanded": True, "value": "Processes input data."},
+                "position": {"x": 0.0, "y": 0.0},
+            },
             "id": "f92c09f0-0434-46cb-829d-a73f801d6343",
-            "label": "ProcessingNode",
+            "label": "Processing Node",
             "outputs": [],
             "ports": [{"id": "abaa2984-b312-4491-b069-e689759f72c8", "name": "default", "type": "DEFAULT"}],
             "trigger": {"id": "35378c2b-f089-44af-ac37-efe4ea42c817", "merge_behavior": "AWAIT_ATTRIBUTES"},
@@ -226,9 +235,12 @@ class HelperClass:
             "attributes": [],
             "base": {"module": ["vellum", "workflows", "nodes", "bases", "base"], "name": "BaseNode"},
             "definition": {"module": ["vellum_custom_nodes", "multiple_nodes"], "name": "TransformationNode"},
-            "display_data": {"comment": {"value": "Transforms data format."}, "position": {"x": 0.0, "y": 0.0}},
+            "display_data": {
+                "comment": {"expanded": True, "value": "Transforms data format."},
+                "position": {"x": 0.0, "y": 0.0},
+            },
             "id": "09ca32f7-c8f2-4469-97e5-1f288f85127a",
-            "label": "TransformationNode",
+            "label": "Transformation Node",
             "outputs": [],
             "ports": [{"id": "88778117-fbfc-4b44-964b-5a4994aa2f24", "name": "default", "type": "DEFAULT"}],
             "trigger": {"id": "5d096263-7fbf-490a-83b7-e441852b5fb6", "merge_behavior": "AWAIT_ATTRIBUTES"},
@@ -289,9 +301,12 @@ class BrokenNode(BaseNode)
         "attributes": [],
         "base": {"module": ["vellum", "workflows", "nodes", "bases", "base"], "name": "BaseNode"},
         "definition": {"module": ["vellum_custom_nodes", "valid_node"], "name": "SomeNode"},
-        "display_data": {"comment": {"value": "This is Some Node."}, "position": {"x": 0.0, "y": 0.0}},
+        "display_data": {
+            "comment": {"expanded": True, "value": "This is Some Node."},
+            "position": {"x": 0.0, "y": 0.0},
+        },
         "id": "1e559c2e-db82-41f0-9ceb-5e89b0c5a0a3",
-        "label": "SomeNode",
+        "label": "Some Node",
         "outputs": [],
         "ports": [{"id": "48e39e97-5fd4-471e-b4f2-51d3baf06456", "name": "default", "type": "DEFAULT"}],
         "trigger": {"id": "e3381fb7-61fc-4c46-ae8e-51fc463b6a59", "merge_behavior": "AWAIT_ATTRIBUTES"},
@@ -368,11 +383,11 @@ class MyAdditionNode(BaseNode):
         "base": {"module": ["vellum", "workflows", "nodes", "bases", "base"], "name": "BaseNode"},
         "definition": {"module": ["vellum_custom_nodes", "addition_node"], "name": "MyAdditionNode"},
         "display_data": {
-            "comment": {"value": "Custom node that performs simple addition."},
+            "comment": {"expanded": True, "value": "Custom node that performs simple addition."},
             "position": {"x": 0.0, "y": 0.0},
         },
         "id": "195cd69d-3d2d-41e4-a432-16c433cb8d34",
-        "label": "MyAdditionNode",
+        "label": "My Addition Node",
         "outputs": [{"id": "3d8e40cb-2aa8-44bd-ae6a-708a9fbc4779", "name": "result", "type": "NUMBER", "value": None}],
         "ports": [{"id": "9a9e4ef6-febf-4093-a515-217bbb1373db", "name": "default", "type": "DEFAULT"}],
         "trigger": {"id": "a5298668-d808-4a45-a62e-790943948e8a", "merge_behavior": "AWAIT_ATTRIBUTES"},

{vellum_workflow_server-1.1.4 → vellum_workflow_server-1.2.0.post1}/src/workflow_server/api/workflow_view.py

@@ -25,7 +25,6 @@ from vellum_ee.workflows.server.virtual_file_loader import VirtualFileFinder
 
 from vellum.workflows.exceptions import WorkflowInitializationException
 from vellum.workflows.nodes import BaseNode
-from vellum.workflows.utils.names import pascal_to_title_case
 from workflow_server.config import ENABLE_PROCESS_WRAPPER, MEMORY_LIMIT_MB
 from workflow_server.core.events import (
     SPAN_ID_EVENT,
@@ -493,7 +492,6 @@ def get_version_route() -> tuple[dict, int]:
                             exec_config_raw = node_display_class().serialize(WorkflowDisplayContext())
                             exec_config = cast(Dict[str, Any], exec_config_raw)
                             config_module = exec_config["definition"]["module"]
-                            label = exec_config["label"]
                             description = (
                                 exec_config["display_data"]["comment"]["value"]
                                 if "comment" in exec_config["display_data"]
@@ -504,7 +502,7 @@ def get_version_route() -> tuple[dict, int]:
                                     "id": str(uuid4()),
                                     "module": config_module,
                                     "name": obj.__name__,
-                                    "label": pascal_to_title_case(label),
+                                    "label": exec_config["label"],
                                     "description": description,
                                     "exec_config": exec_config,
                                 }