vellum-workflow-server 1.11.0.post1__py3-none-any.whl → 1.11.21__py3-none-any.whl

{vellum_workflow_server-1.11.0.post1.dist-info → vellum_workflow_server-1.11.21.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 1.11.0.post1
+Version: 1.11.21
 Summary: 
 License: AGPL
 Requires-Python: >=3.9.0,<4
@@ -30,13 +30,15 @@ Requires-Dist: pyjwt (==2.10.0)
 Requires-Dist: python-dotenv (==1.2.1)
 Requires-Dist: retrying (==1.3.4)
 Requires-Dist: sentry-sdk[flask] (==2.20.0)
-Requires-Dist: vellum-ai (==1.11.0)
+Requires-Dist: vellum-ai (==1.11.21)
 Description-Content-Type: text/markdown
 
 # Vellum Workflow Runner Server
+
 This package is meant for installing on container images in order to use custom docker images when using Vellum Workflows.
 
 ## Example Dockerfile Usage:
+
 ```
 FROM python:3.11.6-slim-bookworm
 
@@ -49,7 +51,6 @@ RUN pip install --upgrade pip
 RUN pip --no-cache-dir install vellum-workflow-server==0.13.2
 
 ENV PYTHONUNBUFFERED 1
-ENV PYTHONDONTWRITEBYTECODE 1
 COPY ./base-image/code_exec_entrypoint.sh .
 RUN chmod +x /code_exec_entrypoint.sh
 
@@ -57,5 +58,6 @@ CMD ["vellum_start_server"]
 ```
 
 ## Skipping Publishes
+
 If you wish to automatically skip publishing a new version when merging to main you can add a [skip-publish] to your commit message. This is useful if your changes are not time sensitive and can just go out with the next release. This avoids causing new services being created causing extra cold starts for our customers and also keeps our public versioning more tidy.
 

{vellum_workflow_server-1.11.0.post1.dist-info → vellum_workflow_server-1.11.21.dist-info}/RECORD

@@ -5,16 +5,16 @@ workflow_server/api/healthz_view.py,sha256=itiRvBDBXncrw8Kbbc73UZLwqMAhgHOR3uSre
 workflow_server/api/status_view.py,sha256=Jah8dBAVL4uOcRfsjKAOyfVONFyk9HQjXeRfjcIqhmA,514
 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=B6B8mCirt3FvpPKRP_AyzPJ199k_gwLzAcQuWRkzEfA,32343
+workflow_server/api/tests/test_workflow_view.py,sha256=I2sd11ptKDqbylzB9rKqkMXeZoh8ttad3zIhNus86vk,32491
 workflow_server/api/tests/test_workflow_view_async_exec.py,sha256=eP_H2xI9SRfJdoJ6HPeynQecnxR50I_8aDCooF-YzIw,11952
-workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=16ZxP_nuAIP1Cg4-4z6EQttn07SRY1GkVq2m53z6XaE,42389
-workflow_server/api/workflow_view.py,sha256=gYT1lWSYadiG5OLQpg09mJT4xz_foSmsIMvtO7GNbAo,24669
+workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=QilhiYv3_iaF-xvbmqUqqQqqVIZC4KgBe1u8Ku1q52s,42511
+workflow_server/api/workflow_view.py,sha256=1VSHGY0VXPnJQTz4XIX_sBiYFlTdByHH7RSAoOHBdtk,26650
 workflow_server/code_exec_runner.py,sha256=vJlCQ8FkcG8RfCZ34Ea2Xt6J7dNkU5EqA-KxRkbVOeo,2219
 workflow_server/config.py,sha256=I4hfTsjIbHxoSKylPCjKnrysPV0jO5nfRKwpKvEcfAE,2193
 workflow_server/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-workflow_server/core/cancel_workflow.py,sha256=QcEeYUIrxq4pub-z9BlGi5fLI3gVRml-56rMCW7j5Hc,2212
+workflow_server/core/cancel_workflow.py,sha256=uMPZg_rQ6iKZBLuxgwla1NYwhkcbO0gLh8QYBfU_2_0,2371
 workflow_server/core/events.py,sha256=24MA66DVQuaLJJcZrS8IL1Zq4Ohi9CoouKZ5VgoH3Cs,1402
-workflow_server/core/executor.py,sha256=Drw6TcegjKFIoowDzeQzCf2QwblSa-stm70YBfFItRU,16968
+workflow_server/core/executor.py,sha256=7mJDuP8uedCAygNorp2Vscy6w7xHHqNZ_jxJXkJZzeY,17112
 workflow_server/core/utils.py,sha256=mecVPqQkthrC4mpop3r8J3IWnBmKbDgqfCrSagyzVEg,2021
 workflow_server/core/workflow_executor_context.py,sha256=8faOdpU4cBeIbmOvg9VzD3eS5i_PKcH7tyNGzx_rehg,3899
 workflow_server/logging_config.py,sha256=Hvx1t8uhqMMinl-5qcef7ufUvzs6x14VRnCb7YZxEAg,1206
@@ -30,8 +30,8 @@ workflow_server/utils/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
 workflow_server/utils/tests/test_sentry_integration.py,sha256=14PfuW8AaQNNtqLmBs16EPe5T3f_iTI7YJMCRtiboZk,4502
 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=m7iMJtor5SQLWu7jlJw-X5Q3nmbq69BCxTMv6qnFYrA,4835
-vellum_workflow_server-1.11.0.post1.dist-info/METADATA,sha256=sU_57FzsmfgTgFLvjdhj_4qSa5hFOlBPrnTtRndDQnc,2308
-vellum_workflow_server-1.11.0.post1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-1.11.0.post1.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-1.11.0.post1.dist-info/RECORD,,
+workflow_server/utils/utils.py,sha256=dwUBP_0ngq6q-y2IFrjEWbGvao2roDR-VAekN6z57kE,5078
+vellum_workflow_server-1.11.21.dist-info/METADATA,sha256=WEJ5yhlrLvZjT0HHImwf0MlRjS6FwAUBTlghLEdbeWY,2277
+vellum_workflow_server-1.11.21.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-1.11.21.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-1.11.21.dist-info/RECORD,,

workflow_server/api/tests/test_workflow_view.py

@@ -389,7 +389,15 @@ class MyAdditionNode(BaseNode):
         },
         "id": "2464b610-fb6d-495b-b17c-933ee147f19f",
         "label": "My Addition Node",
-        "outputs": [{"id": "f39d85c9-e7bf-45e1-bb67-f16225db0118", "name": "result", "type": "NUMBER", "value": None}],
+        "outputs": [
+            {
+                "id": "f39d85c9-e7bf-45e1-bb67-f16225db0118",
+                "name": "result",
+                "type": "NUMBER",
+                "value": None,
+                "schema": {"type": "integer"},
+            }
+        ],
         "ports": [{"id": "bc489295-cd8a-4aa2-88bb-34446374100d", "name": "default", "type": "DEFAULT"}],
         "trigger": {"id": "ff580cad-73d6-44fe-8f2c-4b8dc990ee70", "merge_behavior": "AWAIT_ATTRIBUTES"},
         "type": "GENERIC",

workflow_server/api/tests/test_workflow_view_stream_workflow_route.py

@@ -549,7 +549,10 @@ class Inputs(BaseInputs):
     # 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"]
-    assert "Required input variables foo should have defined value" in events[2]["body"]["error"]["message"]
+    actual_error_message = events[2]["body"]["error"]["message"]
+    assert "Required input variables" in actual_error_message
+    assert "foo" in actual_error_message
+    assert "should have defined value" in actual_error_message
 
     # AND the fourth event should be vembda execution fulfilled
     assert events[3]["name"] == "vembda.execution.fulfilled"

workflow_server/api/workflow_view.py

@@ -22,6 +22,13 @@ 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.errors import WorkflowError, WorkflowErrorCode
+from vellum.workflows.events.workflow import (
+    WorkflowExecutionInitiatedBody,
+    WorkflowExecutionInitiatedEvent,
+    WorkflowExecutionRejectedBody,
+    WorkflowExecutionRejectedEvent,
+)
 from vellum.workflows.exceptions import WorkflowInitializationException
 from vellum.workflows.nodes import BaseNode
 from vellum.workflows.vellum_client import create_vellum_client
@@ -174,6 +181,43 @@ def stream_workflow_route() -> Response:
     return resp
 
 
+def _emit_async_error_events(
+    context: WorkflowExecutorContext, error_message: str, stacktrace: Optional[str] = None
+) -> None:
+    """
+    Emit workflow execution error events when async execution fails before or during workflow startup.
+
+    This ensures that errors in async mode are properly reported to Vellum's events API,
+    making them visible in the executions UI.
+    """
+    try:
+        workflow_span_id = context.workflow_span_id or str(uuid4())
+
+        initiated_event = WorkflowExecutionInitiatedEvent[Any, Any](
+            trace_id=context.trace_id,
+            span_id=workflow_span_id,
+            body=WorkflowExecutionInitiatedBody(inputs=context.inputs),
+            parent=context.execution_context.parent_context if context.execution_context else None,
+        )
+
+        rejected_event = WorkflowExecutionRejectedEvent(
+            trace_id=context.trace_id,
+            span_id=workflow_span_id,
+            body=WorkflowExecutionRejectedBody(
+                error=WorkflowError(
+                    message=error_message,
+                    code=WorkflowErrorCode.INTERNAL_ERROR,
+                ),
+                stacktrace=stacktrace,
+            ),
+            parent=context.execution_context.parent_context if context.execution_context else None,
+        )
+
+        context.vellum_client.events.create(request=[initiated_event, rejected_event])  # type: ignore[list-item]
+    except Exception as e:
+        logger.exception(f"Failed to emit async error events: {e}")
+
+
 @bp.route("/async-exec", methods=["POST"])
 def async_exec_workflow() -> Response:
     data = request.get_json()
@@ -208,8 +252,8 @@ def async_exec_workflow() -> Response:
         try:
             start_workflow_result = _start_workflow(context)
             if isinstance(start_workflow_result, Response):
-                # TODO same here, should return this response as en event or it will get yeeted to the nether
-                # return start_workflow_result
+                error_detail = start_workflow_result.get_json().get("detail", "Unknown error during workflow startup")
+                _emit_async_error_events(context, error_detail)
                 return
 
             workflow_events, vembda_initiated_event, process, span_id, headers = start_workflow_result
@@ -223,6 +267,7 @@ def async_exec_workflow() -> Response:
             )
         except Exception as e:
             logger.exception("Error during workflow async background worker", e)
+            _emit_async_error_events(context, str(e), traceback.format_exc())
         finally:
             if ENABLE_PROCESS_WRAPPER:
                 try:
@@ -531,11 +576,18 @@ def serialize_route() -> Response:
     is_new_server = data.get("is_new_server", False)
     module = data.get("module")
 
+    headers = {
+        "X-Vellum-Is-New-Server": str(is_new_server).lower(),
+    }
+
     if not files:
+        error_message = "No files received"
+        logger.warning(error_message)
         return Response(
-            json.dumps({"detail": "No files received"}),
+            json.dumps({"detail": error_message}),
             status=400,
             content_type="application/json",
+            headers=headers,
         )
 
     client = create_vellum_client(api_key=workspace_api_key)
@@ -544,10 +596,6 @@ def serialize_route() -> Response:
     namespace = get_random_namespace()
     virtual_finder = VirtualFileFinder(files, namespace, source_module=module)
 
-    headers = {
-        "X-Vellum-Is-New-Server": str(is_new_server).lower(),
-    }
-
     try:
         sys.meta_path.append(virtual_finder)
         result = BaseWorkflowDisplay.serialize_module(namespace, client=client, dry_run=True)

workflow_server/core/cancel_workflow.py

@@ -14,14 +14,18 @@ logger = logging.getLogger(__name__)
 
 
 def get_is_workflow_cancelled(execution_id: UUID, vembda_public_url: Optional[str]) -> bool:
-    response = requests.get(
-        f"{vembda_public_url}/vembda-public/cancel-workflow-execution-status/{execution_id}",
-        headers={"Accept": "application/json"},
-        timeout=5,
-    )
-    response.raise_for_status()
+    try:
+        response = requests.get(
+            f"{vembda_public_url}/vembda-public/cancel-workflow-execution-status/{execution_id}",
+            headers={"Accept": "application/json"},
+            timeout=5,
+        )
+        response.raise_for_status()
 
-    return response.json().get("cancelled")
+        return response.json().get("cancelled", False)
+    except Exception:
+        logger.exception("Error checking workflow cancellation status")
+        return False
 
 
 class CancelWorkflowWatcherThread(Thread):

workflow_server/core/executor.py

@@ -14,6 +14,7 @@ from typing import Any, Callable, Generator, Iterator, Optional, Tuple
 
 import orjson
 from vellum_ee.workflows.display.utils.events import event_enricher
+from vellum_ee.workflows.display.utils.expressions import base_descriptor_validator
 from vellum_ee.workflows.server.virtual_file_loader import VirtualFileFinder
 
 from vellum.workflows import BaseWorkflow
@@ -177,6 +178,7 @@ def stream_workflow(
         node_output_mocks = MockNodeExecution.validate_all(
             executor_context.node_output_mocks,
             workflow.__class__,
+            descriptor_validator=base_descriptor_validator,
         )
 
         cancel_signal = cancel_signal or ThreadingEvent()

workflow_server/utils/utils.py

@@ -59,10 +59,19 @@ def convert_json_inputs_to_vellum(inputs: List[dict]) -> dict:
 
 
 def get_version() -> dict:
+    # Return hotswappable lock file so we can save it and reuse it
+    lock_file = None
+    try:
+        with open("/app/uv.lock", "r") as f:
+            lock_file = f.read()
+    except Exception:
+        pass
+
     return {
         "sdk_version": version("vellum-ai"),
         "server_version": "local" if is_development() else version("vellum-workflow-server"),
         "container_image": CONTAINER_IMAGE,
+        "lock_file": lock_file,
     }