vellum-workflow-server 1.7.4.post2__py3-none-any.whl → 1.7.5.post1__py3-none-any.whl

{vellum_workflow_server-1.7.4.post2.dist-info → vellum_workflow_server-1.7.5.post1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 1.7.4.post2
+Version: 1.7.5.post1
 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.dist-info → vellum_workflow_server-1.7.5.post1.dist-info}/RECORD

@@ -5,15 +5,15 @@ workflow_server/api/healthz_view.py,sha256=itiRvBDBXncrw8Kbbc73UZLwqMAhgHOR3uSre
 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=XfsW8_hFKN5rzp2UFVviVoLW-8FD-nvD6mSZHnIyrWM,30483
-workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=kCNcouRXA_zW1WliLDHCeqiLvQInSg70jLsBMFRk2PY,36960
+workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=l4sfma1uNB2CtFjxKTIQ2SSn-PpyxaSWJZyy7VYSDYM,37311
 workflow_server/api/workflow_view.py,sha256=WJC8wfdgqIDk5LLdRN2rom1M4qKvqHroxXpCjYczmjg,19859
 workflow_server/code_exec_runner.py,sha256=DLNNrinCRbnkSvlqVvSZ1wv_etI7r_kKAXNPGMj3jBk,2196
-workflow_server/config.py,sha256=qmmTr6ty3ZN5LDOFs3TfUxYshYe6Mmn_LanplHHeE9Q,1796
+workflow_server/config.py,sha256=CKU7qKC5k1_M1AFOJn_nUqntA46KPk_Z7fblfQaGC-0,1877
 workflow_server/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-workflow_server/core/cancel_workflow.py,sha256=zsP703UOPe_ulPXdbn6osiw-xTFa7IsaWzvXZJWHcKw,2216
+workflow_server/core/cancel_workflow.py,sha256=QcEeYUIrxq4pub-z9BlGi5fLI3gVRml-56rMCW7j5Hc,2212
 workflow_server/core/events.py,sha256=24MA66DVQuaLJJcZrS8IL1Zq4Ohi9CoouKZ5VgoH3Cs,1402
-workflow_server/core/executor.py,sha256=yuWD3swkJh-pPQvcvBG1xkKij-1DH5WhLx8dDfB2Fxg,17662
-workflow_server/core/utils.py,sha256=C59WjckMGDwAyeKe62RmmaMJPi42RZobDVJVptE0k7c,2112
+workflow_server/core/executor.py,sha256=IazlinBkJ9kTqhB-6iiCozB2mblc0CshDEChVml_CZ4,17630
+workflow_server/core/utils.py,sha256=si0NB4Suurc-mn8NYdn59xM9CkPrfOP1aWEVrZvifDI,1929
 workflow_server/core/workflow_executor_context.py,sha256=rSjgkWenRGpFOwioRXNnUhGnEI7isOB-Qk7Tw-2JwnM,4521
 workflow_server/server.py,sha256=lhHPmK1PhRZd6eCkj1C0acK3YwaApZgoPHghMChw0fc,1461
 workflow_server/start.py,sha256=WHZEsLen8mtGX-wF3xUbB91_NpC_GBRtOEi87OR1rv4,2360
@@ -27,8 +27,8 @@ workflow_server/utils/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
 workflow_server/utils/tests/test_sentry_integration.py,sha256=LGmWiaLhFrx-jslrRjRq9JY6Z5ShLZyx_N_L0-FU6OI,2100
 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=ZPoM1Suhid22dpB8oEFLux8wx-9iyzmSfWuYxSCrgWk,4774
-vellum_workflow_server-1.7.4.post2.dist-info/METADATA,sha256=ExLFKBDGeHWAx7-vmmk21IIWZ6z3FogJdNgwXc12yKo,2273
-vellum_workflow_server-1.7.4.post2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-1.7.4.post2.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-1.7.4.post2.dist-info/RECORD,,
+workflow_server/utils/utils.py,sha256=m7iMJtor5SQLWu7jlJw-X5Q3nmbq69BCxTMv6qnFYrA,4835
+vellum_workflow_server-1.7.5.post1.dist-info/METADATA,sha256=AS79hT9zNyxoVj6nYGoLmTvDLmND6YvsDTgtaaTT_3I,2273
+vellum_workflow_server-1.7.5.post1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-1.7.5.post1.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-1.7.5.post1.dist-info/RECORD,,

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")

workflow_server/config.py

@@ -28,6 +28,7 @@ PORT = os.getenv("PORT", "8000")
 VELLUM_API_URL_HOST = os.getenv("VELLUM_API_URL_HOST", "localhost")
 VELLUM_API_URL_PORT = os.getenv("VELLUM_API_URL_PORT", 8000)
 CONCURRENCY = int(os.getenv("CONCURRENCY", "8"))
+CONTAINER_IMAGE = os.getenv("CONTAINER_IMAGE", "python-workflow-runtime:latest")
 ENABLE_PROCESS_WRAPPER = os.getenv("ENABLE_PROCESS_WRAPPER", "true").lower() == "true"
 
 # This controls if we should just load a module already available in the py path for running

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
 

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,
         )

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

workflow_server/utils/utils.py

@@ -15,7 +15,7 @@ from vellum import (
     VellumImage,
     VellumVideo,
 )
-from workflow_server.config import is_development
+from workflow_server.config import CONTAINER_IMAGE, is_development
 
 
 def convert_json_inputs_to_vellum(inputs: List[dict]) -> dict:
@@ -62,6 +62,7 @@ def get_version() -> dict:
     return {
         "sdk_version": version("vellum-ai"),
         "server_version": "local" if is_development() else version("vellum-workflow-server"),
+        "container_image": CONTAINER_IMAGE,
     }