vellum-workflow-server 0.14.73.post5__tar.gz → 0.14.73.post7__tar.gz

{vellum_workflow_server-0.14.73.post5 → vellum_workflow_server-0.14.73.post7}/PKG-INFO

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

{vellum_workflow_server-0.14.73.post5 → vellum_workflow_server-0.14.73.post7}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-workflow-server"
 
 [tool.poetry]
 name = "vellum-workflow-server"
-version = "0.14.73.post5"
+version = "0.14.73.post7"
 description = ""
 readme = "README.md"
 authors = []

{vellum_workflow_server-0.14.73.post5 → vellum_workflow_server-0.14.73.post7}/src/workflow_server/api/tests/test_workflow_view_stream_workflow_route.py

@@ -3,6 +3,7 @@ from contextlib import redirect_stdout
 from importlib.metadata import version
 import io
 import json
+from queue import Empty
 from unittest import mock
 from uuid import uuid4
 
@@ -825,3 +826,43 @@ class Workflow(BaseWorkflow):
     outputs = events[2]["body"]["outputs"]
     assert outputs["env_var_value"] == "test_value"
     assert outputs["another_var_value"] == "another_value"
+
+
+@mock.patch("workflow_server.api.workflow_view.Queue")
+def test_stream_workflow_route__queue_get_timeout(mock_queue_class):
+    # GIVEN a valid request body
+    span_id = uuid4()
+    request_body = {
+        "execution_id": str(span_id),
+        "inputs": [],
+        "environment_api_key": "test",
+        "module": "workflow",
+        "timeout": 360,
+        "files": {
+            "__init__.py": "",
+            "workflow.py": """\
+from vellum.workflows import BaseWorkflow
+
+class Workflow(BaseWorkflow):
+    class Outputs(BaseWorkflow.Outputs):
+        foo = "hello"
+""",
+        },
+    }
+
+    # AND the queue.get method raises Empty exception
+    mock_queue_instance = mock_queue_class.return_value
+    mock_queue_instance.get.side_effect = Empty()
+
+    # WHEN we call the stream route
+    flask_app = create_app()
+    with flask_app.test_client() as test_client:
+        response = test_client.post("/workflow/stream", json=request_body)
+        status_code = response.status_code
+        response_data = response.get_json()
+
+    # THEN we get a 408 response
+    assert status_code == 408
+
+    # AND we get the expected timeout error message
+    assert response_data == {"detail": "Request timed out trying to initiate the Workflow"}

{vellum_workflow_server-0.14.73.post5 → vellum_workflow_server-0.14.73.post7}/src/workflow_server/api/workflow_view.py

@@ -52,6 +52,7 @@ set_start_method("fork", force=True)
 logger = logging.getLogger(__name__)
 
 CUSTOM_NODES_DIRECTORY = "vellum_custom_nodes"
+WORKFLOW_INITIATION_TIMEOUT_SECONDS = 60
 
 
 @bp.route("/stream", methods=["POST"])
@@ -118,7 +119,17 @@ def stream_workflow_route() -> Response:
 
         process_output_queue.put(create_vembda_rejected_event(context, traceback.format_exc()))
 
-    first_item = process_output_queue.get(timeout=0.1)
+    try:
+        first_item = process_output_queue.get(timeout=WORKFLOW_INITIATION_TIMEOUT_SECONDS)
+    except Empty:
+        logger.error("Request timed out trying to initiate the Workflow")
+        return Response(
+            json.dumps({"detail": "Request timed out trying to initiate the Workflow"}),
+            status=408,
+            content_type="application/json",
+            headers=headers,
+        )
+
     if isinstance(first_item, str) and first_item.startswith(SPAN_ID_EVENT):
         span_id = first_item.split(":")[1]
         headers["X-Vellum-Workflow-Span-Id"] = span_id