vellum-workflow-server 0.14.75__tar.gz → 0.14.75.post1__tar.gz

{vellum_workflow_server-0.14.75 → vellum_workflow_server-0.14.75.post1}/PKG-INFO

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

{vellum_workflow_server-0.14.75 → vellum_workflow_server-0.14.75.post1}/pyproject.toml

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

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

@@ -3,7 +3,7 @@ import importlib
 import inspect
 import json
 import logging
-from multiprocessing import Queue, set_start_method
+from multiprocessing import Process, Queue, set_start_method
 import os
 import pkgutil
 from queue import Empty
@@ -11,7 +11,7 @@ import sys
 import time
 import traceback
 from uuid import uuid4
-from typing import Any, Dict, Generator, Iterator, Union, cast
+from typing import Any, Dict, Generator, Iterator, Optional, Union, cast
 
 from flask import Blueprint, Response, current_app as app, request, stream_with_context
 from pydantic import ValidationError
@@ -108,6 +108,7 @@ def stream_workflow_route() -> Response:
             headers=headers,
         )
 
+    process: Optional[Process] = None
     try:
         process = stream_workflow_process_timeout(
             executor_context=context,
@@ -123,6 +124,11 @@ def stream_workflow_route() -> Response:
         first_item = process_output_queue.get(timeout=WORKFLOW_INITIATION_TIMEOUT_SECONDS)
     except Empty:
         logger.error("Request timed out trying to initiate the Workflow")
+
+        if process and process.is_alive():
+            process.kill()
+        increment_process_count(-1)
+
         return Response(
             json.dumps({"detail": "Request timed out trying to initiate the Workflow"}),
             status=408,
@@ -250,6 +256,12 @@ def stream_workflow_route() -> Response:
             # being exceeded.
             app.logger.error("Client disconnected in the middle of the Workflow Stream")
             return
+        except Exception as e:
+            logger.exception("Error during workflow response stream generator", extra={"error": e})
+            yield "\n"
+            yield "END"
+            yield "\n"
+            return
         finally:
             try:
                 if process and process.is_alive():