vellum-workflow-server 1.1.5.post1__py3-none-any.whl → 1.2.0.post2__py3-none-any.whl

{vellum_workflow_server-1.1.5.post1.dist-info → vellum_workflow_server-1.2.0.post2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 1.1.5.post1
+Version: 1.2.0.post2
 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.5)
+Requires-Dist: vellum-ai (==1.2.0)
 Description-Content-Type: text/markdown
 
 # Vellum Workflow Runner Server

{vellum_workflow_server-1.1.5.post1.dist-info → vellum_workflow_server-1.2.0.post2.dist-info}/RECORD

@@ -4,9 +4,9 @@ workflow_server/api/auth_middleware.py,sha256=IlZaCiwZ5nwQqk5sYQorvOFj7lt0p1ZSSE
 workflow_server/api/healthz_view.py,sha256=itiRvBDBXncrw8Kbbc73UZLwqMAhgHOR3uSre_dAfgY,404
 workflow_server/api/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 workflow_server/api/tests/test_input_display_mapping.py,sha256=drBZqMudFyB5wgiUOcMgRXz7E7ge-Qgxbstw4E4f0zE,2211
-workflow_server/api/tests/test_workflow_view.py,sha256=Ummbd-18XN8jc3bbIDa7j3jifu4lTiHIaRhnTDnpjAI,18945
+workflow_server/api/tests/test_workflow_view.py,sha256=H1tYWSA9nj0pXJ987pxQw5ERn1O9mPTc6WJEXA5ptV0,19094
 workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=7q5AiBzfXX-npCrp-Da2OWk6-M1Ai8QZPDLhLlkxb34,26051
-workflow_server/api/workflow_view.py,sha256=i19Ksgg_2EbPhPRAlwYRqmAKI_8VHvAGc61Q9k90Iow,21189
+workflow_server/api/workflow_view.py,sha256=WDjVxMW1H9R5CQjHZgSdi4mbM1dMp3NTurtSVqtCvOU,21037
 workflow_server/code_exec_runner.py,sha256=lBnMIorPZL8zZBye6TjeCIs06WTJM7P2HR07B1fjJJI,2533
 workflow_server/config.py,sha256=DyTty8NrAwvtx-esM3KthnpsNh-nKdWNlovWQOgiGpg,1417
 workflow_server/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -28,7 +28,7 @@ workflow_server/utils/tests/test_sentry_integration.py,sha256=LGmWiaLhFrx-jslrRj
 workflow_server/utils/tests/test_system_utils.py,sha256=_4GwXvVvU5BrATxUEWwQIPg0bzQXMWBtiBmjP8MTxJM,4314
 workflow_server/utils/tests/test_utils.py,sha256=qwK5Rmy3RQyjtlUrYAuGuDlBeRzZKsf1yS-y2IpUizQ,6452
 workflow_server/utils/utils.py,sha256=Wqqn-1l2ugkGgy5paWWdt0AVxAyPMQCYcnRSSOMjXlA,4355
-vellum_workflow_server-1.1.5.post1.dist-info/METADATA,sha256=nBEmEeICCRLkYMyab6en6v8HSy-pCgRod-caY6qT-a8,2273
-vellum_workflow_server-1.1.5.post1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-1.1.5.post1.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-1.1.5.post1.dist-info/RECORD,,
+vellum_workflow_server-1.2.0.post2.dist-info/METADATA,sha256=9QvVFnyxfKSFOJfVSJj-IU_SRdk0ZJIZU9M_r1gDfL4,2273
+vellum_workflow_server-1.2.0.post2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-1.2.0.post2.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-1.2.0.post2.dist-info/RECORD,,

workflow_server/api/tests/test_workflow_view.py

@@ -63,7 +63,7 @@ class TestNode(BaseNode):
             "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"},
@@ -127,7 +127,7 @@ class SomeOtherNode(BaseNode):
             "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"},
@@ -150,7 +150,7 @@ class SomeOtherNode(BaseNode):
             "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"},
@@ -222,7 +222,7 @@ class HelperClass:
                 "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"},
@@ -240,7 +240,7 @@ class HelperClass:
                 "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"},
@@ -306,7 +306,7 @@ class BrokenNode(BaseNode)
             "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"},
@@ -387,7 +387,7 @@ class MyAdditionNode(BaseNode):
             "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"},
@@ -456,8 +456,10 @@ def test_serialize_route__with__workflow():
     # THEN we should get a successful response
     assert response.status_code == 200
 
-    # AND the response should contain exec_config
+    # AND the response should contain the full WorkflowSerializationResult
     assert "exec_config" in response.json
+    assert "errors" in response.json
+    assert "dataset" in response.json or response.json.get("dataset") is None
     exec_config = response.json["exec_config"]
 
     # AND the exec_config should have workflow_raw_data

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,
@@ -447,10 +446,10 @@ def serialize_route() -> Response:
 
         try:
             sys.meta_path.append(virtual_finder)
-            exec_config = BaseWorkflowDisplay.serialize_module(namespace).exec_config
+            result = BaseWorkflowDisplay.serialize_module(namespace)
 
             return Response(
-                json.dumps({"exec_config": exec_config}),
+                json.dumps(result.model_dump()),
                 status=200,
                 content_type="application/json",
             )
@@ -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,
                                 }