vellum-workflow-server 1.7.2.post1__py3-none-any.whl → 1.7.2.post3__py3-none-any.whl

{vellum_workflow_server-1.7.2.post1.dist-info → vellum_workflow_server-1.7.2.post3.dist-info}/METADATA

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

{vellum_workflow_server-1.7.2.post1.dist-info → vellum_workflow_server-1.7.2.post3.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=XfsW8_hFKN5rzp2UFVviVoLW-8FD-nvD6mSZHnIyrWM,30483
-workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=1tLpUx6dBOJ7BSlKGEJ8sB6eDEuyzy6eK2vbM_IsQuQ,35558
+workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=kCNcouRXA_zW1WliLDHCeqiLvQInSg70jLsBMFRk2PY,36960
 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/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 workflow_server/core/cancel_workflow.py,sha256=zsP703UOPe_ulPXdbn6osiw-xTFa7IsaWzvXZJWHcKw,2216
 workflow_server/core/events.py,sha256=24MA66DVQuaLJJcZrS8IL1Zq4Ohi9CoouKZ5VgoH3Cs,1402
-workflow_server/core/executor.py,sha256=jXX5OLv0iacDoB_TAyJs-1xIIvsFkZHdHIn4ncWuIoQ,17893
+workflow_server/core/executor.py,sha256=yuWD3swkJh-pPQvcvBG1xkKij-1DH5WhLx8dDfB2Fxg,17662
 workflow_server/core/utils.py,sha256=C59WjckMGDwAyeKe62RmmaMJPi42RZobDVJVptE0k7c,2112
 workflow_server/core/workflow_executor_context.py,sha256=rSjgkWenRGpFOwioRXNnUhGnEI7isOB-Qk7Tw-2JwnM,4521
 workflow_server/server.py,sha256=QBU12AaAfAgLqfCDBd24qIJl_mbheiq0-hfcWV7rZM4,1234
@@ -28,7 +28,7 @@ workflow_server/utils/tests/test_sentry_integration.py,sha256=LGmWiaLhFrx-jslrRj
 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.2.post1.dist-info/METADATA,sha256=66m30H2Oto9Z3uRwJMo-pZRimqVv0eetIQ8RPwQKLnU,2273
-vellum_workflow_server-1.7.2.post1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-1.7.2.post1.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-1.7.2.post1.dist-info/RECORD,,
+vellum_workflow_server-1.7.2.post3.dist-info/METADATA,sha256=Bysb6xz6RmFmLOhj-9rOighhBrq7YEi_tK_hK7QuD1s,2273
+vellum_workflow_server-1.7.2.post3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-1.7.2.post3.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-1.7.2.post3.dist-info/RECORD,,

workflow_server/api/tests/test_workflow_view_stream_workflow_route.py

@@ -155,6 +155,14 @@ class Workflow(BaseWorkflow):
 
     assert events[1]["name"] == "workflow.execution.initiated", events[1]
     assert "display_context" in events[1]["body"], events[1]["body"]
+    display_context = events[1]["body"]["display_context"]
+    assert "node_displays" in display_context
+    assert "workflow_inputs" in display_context
+    assert "workflow_outputs" in display_context
+    assert isinstance(display_context["node_displays"], dict)
+    assert isinstance(display_context["workflow_inputs"], dict)
+    assert isinstance(display_context["workflow_outputs"], dict)
+    assert "foo" in display_context["workflow_outputs"]
     assert events[2]["name"] == "workflow.execution.fulfilled", events[2]
 
     assert events[3] == {
@@ -234,6 +242,15 @@ class Inputs(BaseInputs):
 
     assert events[1]["name"] == "workflow.execution.initiated", events[1]
     assert "display_context" in events[1]["body"], events[1]["body"]
+    display_context = events[1]["body"]["display_context"]
+    assert "node_displays" in display_context
+    assert "workflow_inputs" in display_context
+    assert "workflow_outputs" in display_context
+    assert isinstance(display_context["node_displays"], dict)
+    assert isinstance(display_context["workflow_inputs"], dict)
+    assert isinstance(display_context["workflow_outputs"], dict)
+    assert "foo" in display_context["workflow_inputs"]
+    assert "foo" in display_context["workflow_outputs"]
     assert events[2]["name"] == "workflow.execution.fulfilled", events[2]
 
     assert events[3] == {
@@ -311,6 +328,14 @@ class State(BaseState):
 
     assert events[1]["name"] == "workflow.execution.initiated", events[1]
     assert "display_context" in events[1]["body"], events[1]["body"]
+    display_context = events[1]["body"]["display_context"]
+    assert "node_displays" in display_context
+    assert "workflow_inputs" in display_context
+    assert "workflow_outputs" in display_context
+    assert isinstance(display_context["node_displays"], dict)
+    assert isinstance(display_context["workflow_inputs"], dict)
+    assert isinstance(display_context["workflow_outputs"], dict)
+    assert "foo" in display_context["workflow_outputs"]
     assert events[2]["name"] == "workflow.execution.fulfilled", events[2]
     assert events[2]["body"]["outputs"] == {"foo": "bar"}
 

workflow_server/core/executor.py

@@ -8,7 +8,6 @@ import os
 import random
 import string
 import sys
-import threading
 from threading import Event as ThreadingEvent
 import time
 from traceback import format_exc
@@ -237,12 +236,6 @@ def stream_workflow(
         finally:
             cancel_watcher_kill_switch.set()
 
-        emitter_thread = next(
-            (t for t in threading.enumerate() if t.name.endswith(".background_thread") and t.is_alive()), None
-        )
-        if emitter_thread:
-            emitter_thread.join()
-
         workflow.join()
 
     return (