vellum-workflow-server 1.7.0__py3-none-any.whl → 1.7.0.post1__py3-none-any.whl

{vellum_workflow_server-1.7.0.dist-info → vellum_workflow_server-1.7.0.post1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 1.7.0
+Version: 1.7.0.post1
 Summary: 
 License: AGPL
 Requires-Python: >=3.9.0,<4

{vellum_workflow_server-1.7.0.dist-info → vellum_workflow_server-1.7.0.post1.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=d46UNMZUJbIKLiTJkjIsVpgqCJMwCe4LL6RjFkfx_Y4,29178
+workflow_server/api/tests/test_workflow_view.py,sha256=XfsW8_hFKN5rzp2UFVviVoLW-8FD-nvD6mSZHnIyrWM,30483
 workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=K_T-AGeVZ-UOe5Cy8n8-yRof4eZPmLjEqvlEYalaIBo,33847
-workflow_server/api/workflow_view.py,sha256=pqqgeQMVU3YjozjP26Q_f9qSsL_pjkpOQctsq_xwHfE,20447
+workflow_server/api/workflow_view.py,sha256=TBwrtWeQtrRaOUNT_lUTPS_6MqK3-JUrkd91hIDByPA,20179
 workflow_server/code_exec_runner.py,sha256=5uTjHJ231aX6vMwPKRvLcQdbsIxhjTsGMk8G0MEnasU,2152
 workflow_server/config.py,sha256=qmmTr6ty3ZN5LDOFs3TfUxYshYe6Mmn_LanplHHeE9Q,1796
 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=0Nq6du8o-iBtTrip9_wgHES53JSiJbVdSXaBnPobw3s,6930
 workflow_server/utils/utils.py,sha256=ZPoM1Suhid22dpB8oEFLux8wx-9iyzmSfWuYxSCrgWk,4774
-vellum_workflow_server-1.7.0.dist-info/METADATA,sha256=I5OXyosyGAtBVxLGNuQaUBgzOMd0PnJr0PujFy3tCYI,2267
-vellum_workflow_server-1.7.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-1.7.0.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-1.7.0.dist-info/RECORD,,
+vellum_workflow_server-1.7.0.post1.dist-info/METADATA,sha256=QgS_WIUlRyosFr7Bkm-epkharNF-Hi7gtRDLcASzlUc,2273
+vellum_workflow_server-1.7.0.post1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-1.7.0.post1.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-1.7.0.post1.dist-info/RECORD,,

workflow_server/api/tests/test_workflow_view.py

@@ -429,7 +429,7 @@ class BrokenNode(BaseNode)  # Missing colon
     with flask_app.test_client() as test_client:
         response = test_client.post("/workflow/serialize", json={"files": {"broken_node.py": invalid_content}})
 
-    # THEN we should get a 4xx response
+    # THEN we should get a 400 response
     assert response.status_code == 400
 
     # AND the response should contain an error message
@@ -729,3 +729,45 @@ class Workflow(BaseWorkflow):
     assert events[0]["name"] == "vembda.execution.initiated"
     assert events[1]["name"] == "vembda.execution.fulfilled"
     assert len(events) == 2
+
+
+def test_serialize_route__with_invalid_nested_set_graph():
+    """
+    Tests that a workflow with an invalid nested set graph structure raises a clear user-facing exception.
+    """
+    # GIVEN a Flask application
+    flask_app = create_app()
+
+    invalid_workflow_content = """
+from vellum.workflows import BaseWorkflow
+from vellum.workflows.nodes import BaseNode
+
+class TestNode(BaseNode):
+    class Outputs(BaseNode.Outputs):
+        value = "test"
+
+class InvalidWorkflow(BaseWorkflow):
+    graph = {TestNode, {TestNode}}
+
+    class Outputs(BaseWorkflow.Outputs):
+        result = TestNode.Outputs.value
+"""
+
+    workflow_files = {
+        "__init__.py": "",
+        "workflow.py": invalid_workflow_content,
+    }
+
+    # WHEN we make a request to the serialize route
+    with flask_app.test_client() as test_client:
+        response = test_client.post("/workflow/serialize", json={"files": workflow_files})
+
+    # THEN we should get a 400 response
+    assert response.status_code == 400
+
+    # AND the response should contain a user-friendly error message
+    assert "detail" in response.json
+    error_detail = response.json["detail"]
+    assert "Serialization failed" in error_detail
+    assert "Invalid graph structure detected" in error_detail
+    assert "contact Vellum support" in error_detail

workflow_server/api/workflow_view.py

@@ -23,7 +23,6 @@ from vellum_ee.workflows.display.types import WorkflowDisplayContext
 from vellum_ee.workflows.display.workflows import BaseWorkflowDisplay
 from vellum_ee.workflows.server.virtual_file_loader import VirtualFileFinder
 
-from vellum.workflows import BaseWorkflow
 from vellum.workflows.exceptions import WorkflowInitializationException
 from vellum.workflows.nodes import BaseNode
 from workflow_server.config import ENABLE_PROCESS_WRAPPER, MEMORY_LIMIT_MB
@@ -446,13 +445,7 @@ def serialize_route() -> Response:
 
         try:
             sys.meta_path.append(virtual_finder)
-            try:
-                result = BaseWorkflowDisplay.serialize_module(namespace, client=client)
-            except Exception as e:
-                raise WorkflowInitializationException(
-                    message=str(e),
-                    workflow_definition=BaseWorkflow,
-                ) from e
+            result = BaseWorkflowDisplay.serialize_module(namespace, client=client)
 
             return Response(
                 json.dumps(result.model_dump()),