vellum-workflow-server 1.6.4.post2__tar.gz → 1.7.0__tar.gz

{vellum_workflow_server-1.6.4.post2 → vellum_workflow_server-1.7.0}/PKG-INFO

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

{vellum_workflow_server-1.6.4.post2 → vellum_workflow_server-1.7.0}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-workflow-server"
 
 [tool.poetry]
 name = "vellum-workflow-server"
-version = "1.6.4.post2"
+version = "1.7.0"
 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.6.4"
+vellum-ai = "1.7.0"
 python-dotenv = "1.0.1"
 retrying = "1.3.4"
 sentry-sdk = {extras = ["flask"], version = "2.20.0"}

{vellum_workflow_server-1.6.4.post2 → vellum_workflow_server-1.7.0}/src/workflow_server/api/tests/test_workflow_view_stream_workflow_route.py

@@ -396,7 +396,8 @@ from vellum.workflows.inputs import BaseInputs
     assert events[2]["name"] == "workflow.execution.rejected"
     assert events[2]["span_id"] == events[1]["span_id"]
     assert (
-        "Failed to initialize workflow: unexpected indent (inputs.py, line 3)" in events[2]["body"]["error"]["message"]
+        "Failed to initialize workflow: Failed to load workflow module: unexpected indent (inputs.py, line 3)"
+        in events[2]["body"]["error"]["message"]
     )
 
     assert events[3] == {
@@ -1031,3 +1032,72 @@ class Workflow(BaseWorkflow):
     assert len(event_names) == 2, "Should include 2 events"
     assert "workflow.execution.initiated" in event_names, "Should include workflow.execution.initiated event"
     assert "workflow.execution.fulfilled" in event_names, "Should include workflow.execution.fulfilled event"
+
+
+def test_stream_workflow_route__with_invalid_nested_set_graph(both_stream_types):
+    """
+    Tests that a workflow with an invalid nested set graph structure raises a clear error in the stream response.
+    """
+    # GIVEN a Flask application and invalid workflow content with nested set graph
+    span_id = uuid4()
+
+    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
+"""
+
+    request_body = {
+        "timeout": 360,
+        "execution_id": str(span_id),
+        "inputs": [],
+        "environment_api_key": "test",
+        "module": "workflow",
+        "files": {
+            "__init__.py": "",
+            "workflow.py": invalid_workflow_content,
+        },
+    }
+
+    # WHEN we call the stream route
+    status_code, events = both_stream_types(request_body)
+
+    # THEN we get a 200 response
+    assert status_code == 200, events
+
+    # THEN we get the expected events: vembda initiated, workflow initiated, workflow rejected, vembda fulfilled
+    assert len(events) == 4
+
+    # AND the first event should be vembda execution initiated
+    assert events[0]["name"] == "vembda.execution.initiated"
+    assert events[0]["span_id"] == str(span_id)
+
+    # AND the second event should be workflow execution initiated
+    assert events[1]["name"] == "workflow.execution.initiated"
+
+    # AND the third event should be workflow execution rejected
+    assert events[2]["name"] == "workflow.execution.rejected"
+    assert events[1]["span_id"] == events[2]["span_id"]
+
+    # AND the error message should contain information about the invalid graph structure
+    error_message = events[2]["body"]["error"]["message"]
+    expected_message = (
+        "Failed to initialize workflow: Invalid graph structure detected. "
+        "Nested sets or unsupported graph types are not allowed. "
+        "Please contact Vellum support for assistance with Workflow configuration."
+    )
+    assert error_message == expected_message
+
+    # AND the fourth event should be vembda execution fulfilled
+    assert events[3]["name"] == "vembda.execution.fulfilled"
+    assert events[3]["span_id"] == str(span_id)
+    assert events[3]["body"]["exit_code"] == 0