vellum-workflow-server 1.8.0__py3-none-any.whl → 1.8.0.post1__py3-none-any.whl

{vellum_workflow_server-1.8.0.dist-info → vellum_workflow_server-1.8.0.post1.dist-info}/METADATA

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

{vellum_workflow_server-1.8.0.dist-info → vellum_workflow_server-1.8.0.post1.dist-info}/RECORD

@@ -5,14 +5,14 @@ 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=_9SHdK1t-wwWILZFrnoYe5NhkoLazJnt7K1oH4bnJU0,30355
-workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=43p-uY5KUSUH3KiiLup4PlIx2tlPR5vankeDLN9-i_c,38452
+workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=GpOBFnbc54KotEAwwXaCSM2CQMnvXCl64v5IwN_ZVm0,38658
 workflow_server/api/workflow_view.py,sha256=tSepIRP0LcJUKae_wSHGw6P5Ho8K20_YTj3J-9ZL6bs,20463
 workflow_server/code_exec_runner.py,sha256=DLNNrinCRbnkSvlqVvSZ1wv_etI7r_kKAXNPGMj3jBk,2196
 workflow_server/config.py,sha256=cUdI_lEovV7e7lwCkGJ1eM9R4OZVJw5R5zT1eG1SzQQ,2122
 workflow_server/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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=uhjDFFko0evGFnqjVB1qJdioPkO23QxIQCmAhzAzYRk,17380
+workflow_server/core/executor.py,sha256=8EDEyAFonQ5cIV9-orbgSbDErPTe7sj5-fN4WKZR2VM,17426
 workflow_server/core/utils.py,sha256=si0NB4Suurc-mn8NYdn59xM9CkPrfOP1aWEVrZvifDI,1929
 workflow_server/core/workflow_executor_context.py,sha256=Q2R0T2KkYZ1z52v8erDMysJfxSODbzPhDxBxX--k4Zw,3202
 workflow_server/logging_config.py,sha256=Hvx1t8uhqMMinl-5qcef7ufUvzs6x14VRnCb7YZxEAg,1206
@@ -29,7 +29,7 @@ workflow_server/utils/tests/test_sentry_integration.py,sha256=14PfuW8AaQNNtqLmBs
 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.8.0.dist-info/METADATA,sha256=iUzqIEyJfwiziCGvs7GyWLcLykGEI6BcMUG_lZ56zaI,2267
-vellum_workflow_server-1.8.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-1.8.0.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-1.8.0.dist-info/RECORD,,
+vellum_workflow_server-1.8.0.post1.dist-info/METADATA,sha256=gZjZJqVnpCz8ge-f2cDaLHeT70Hu-sxe6Xb4L2dXq8k,2273
+vellum_workflow_server-1.8.0.post1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-1.8.0.post1.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-1.8.0.post1.dist-info/RECORD,,

workflow_server/api/tests/test_workflow_view_stream_workflow_route.py

@@ -671,7 +671,10 @@ class TimeoutWorkflow(BaseWorkflow):
 
     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 workflow_execution_rejected["body"]["error"]["code"] == "WORKFLOW_TIMEOUT"
+    # TODO: Uncomment once version 1.8.1 is released
+    # assert "stacktrace" in workflow_execution_rejected["body"]
+    # 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/core/executor.py

@@ -174,6 +174,7 @@ def stream_workflow(
             cancel_signal=cancel_signal,
             entrypoint_nodes=[executor_context.node_id] if executor_context.node_id else None,
             previous_execution_id=executor_context.previous_execution_id,
+            timeout=executor_context.timeout,
         )
     except WorkflowInitializationException as e:
         cancel_watcher_kill_switch.set()