vellum-workflow-server 1.7.4.post2__tar.gz → 1.7.5__tar.gz

{vellum_workflow_server-1.7.4.post2 → vellum_workflow_server-1.7.5}/PKG-INFO

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

{vellum_workflow_server-1.7.4.post2 → vellum_workflow_server-1.7.5}/pyproject.toml

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

{vellum_workflow_server-1.7.4.post2 → vellum_workflow_server-1.7.5}/src/workflow_server/api/tests/test_workflow_view_stream_workflow_route.py

@@ -645,10 +645,13 @@ class TimeoutWorkflow(BaseWorkflow):
     assert "workflow.execution.initiated" in event_names
     assert "node.execution.initiated" in event_names
 
-    # TODO: Re-enable once solved SDK-side
-    # assert "node.execution.rejected" in event_names, "Should emit node.execution.rejected on timeout"
+    assert "node.execution.rejected" in event_names, "Should emit node.execution.rejected on timeout"
+    node_execution_rejected = next(e for e in events if e["name"] == "node.execution.rejected")
+    assert "vellum/workflows/runner/runner.py" in node_execution_rejected["body"]["stacktrace"]
 
     assert "workflow.execution.rejected" in event_names, "Should emit workflow.execution.rejected on timeout"
+    workflow_execution_rejected = next(e for e in events if e["name"] == "workflow.execution.rejected")
+    assert "vellum/workflows/runner/runner.py" in workflow_execution_rejected["body"]["stacktrace"]
 
     assert "vembda.execution.fulfilled" in event_names
     vembda_fulfilled = next(e for e in events if e["name"] == "vembda.execution.fulfilled")

{vellum_workflow_server-1.7.4.post2 → vellum_workflow_server-1.7.5}/src/workflow_server/core/cancel_workflow.py

@@ -6,7 +6,7 @@ from typing import Optional
 
 import requests
 
-from workflow_server.core.utils import CancelSignal
+from vellum.workflows.types import CancelSignal
 
 _TIMER_INTERVAL = 5
 

{vellum_workflow_server-1.7.4.post2 → vellum_workflow_server-1.7.5}/src/workflow_server/core/executor.py

@@ -32,6 +32,7 @@ from vellum.workflows.resolvers.base import BaseWorkflowResolver
 from vellum.workflows.resolvers.resolver import VellumResolver
 from vellum.workflows.state.context import WorkflowContext
 from vellum.workflows.state.store import EmptyStore
+from vellum.workflows.types import CancelSignal
 from vellum.workflows.workflows.event_filters import all_workflow_event_filter
 from workflow_server.config import LOCAL_DEPLOYMENT, LOCAL_WORKFLOW_MODULE
 from workflow_server.core.cancel_workflow import CancelWorkflowWatcherThread
@@ -42,7 +43,6 @@ from workflow_server.core.events import (
     VembdaExecutionFulfilledEvent,
 )
 from workflow_server.core.utils import (
-    CancelSignal,
     create_vembda_rejected_event,
     is_events_emitting_enabled,
     serialize_vembda_rejected_event,
@@ -172,8 +172,7 @@ def stream_workflow(
             state=workflow_state,
             node_output_mocks=node_output_mocks,
             event_filter=all_workflow_event_filter,
-            # TODO: Fix type in SDK
-            cancel_signal=cancel_signal,  # type: ignore[arg-type]
+            cancel_signal=cancel_signal,
             entrypoint_nodes=[run_from_node] if run_from_node else None,
             previous_execution_id=executor_context.previous_execution_id,
         )

{vellum_workflow_server-1.7.4.post2 → vellum_workflow_server-1.7.5}/src/workflow_server/core/utils.py

@@ -1,14 +1,10 @@
 from datetime import datetime
-from multiprocessing.synchronize import Event as MultiprocessingEvent
-from threading import Event as ThreadingEvent
 from uuid import uuid4
-from typing import Optional, Union
+from typing import Optional
 
 from workflow_server.core.events import VembdaExecutionFulfilledBody, VembdaExecutionFulfilledEvent
 from workflow_server.core.workflow_executor_context import BaseExecutorContext
 
-CancelSignal = Union[MultiprocessingEvent, ThreadingEvent]
-
 
 def _create_vembda_rejected_event_base(
     executor_context: Optional[BaseExecutorContext], error_message: str, timed_out: bool