vellum-workflow-server 1.8.2__tar.gz → 1.8.2.post1__tar.gz

{vellum_workflow_server-1.8.2 → vellum_workflow_server-1.8.2.post1}/PKG-INFO

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

{vellum_workflow_server-1.8.2 → vellum_workflow_server-1.8.2.post1}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-workflow-server"
 
 [tool.poetry]
 name = "vellum-workflow-server"
-version = "1.8.2"
+version = "1.8.2.post1"
 description = ""
 readme = "README.md"
 authors = []

{vellum_workflow_server-1.8.2 → vellum_workflow_server-1.8.2.post1}/src/workflow_server/api/tests/test_workflow_view_stream_workflow_route.py

@@ -384,9 +384,15 @@ class State(BaseState):
 def test_stream_workflow_route__bad_indent_in_inputs_file(both_stream_types):
     # GIVEN a valid request body
     span_id = uuid4()
+    trace_id = uuid4()
+    parent_span_id = uuid4()
     request_body = {
         "timeout": 360,
         "execution_id": str(span_id),
+        "execution_context": {
+            "trace_id": str(trace_id),
+            "parent_context": {"span_id": str(parent_span_id)},
+        },
         "inputs": [
             {"name": "foo", "type": "STRING", "value": "hello"},
         ],
@@ -423,7 +429,7 @@ from vellum.workflows.inputs import BaseInputs
 
     assert events[0] == {
         "id": mock.ANY,
-        "trace_id": mock.ANY,
+        "trace_id": str(trace_id),
         "span_id": str(span_id),
         "timestamp": mock.ANY,
         "api_version": "2024-10-25",
@@ -437,9 +443,19 @@ from vellum.workflows.inputs import BaseInputs
     }
 
     assert events[1]["name"] == "workflow.execution.initiated"
+    assert events[1]["trace_id"] == str(trace_id), "workflow initiated event should use request trace_id"
+    assert events[1]["parent"] is not None, "workflow initiated event should have parent context"
+    assert events[1]["parent"]["span_id"] == str(
+        parent_span_id
+    ), "workflow initiated event parent should match request parent_context"
 
     assert events[2]["name"] == "workflow.execution.rejected"
+    assert events[2]["trace_id"] == str(trace_id), "workflow rejected event should use request trace_id"
     assert events[2]["span_id"] == events[1]["span_id"]
+    assert events[2]["parent"] is not None, "workflow rejected event should have parent context"
+    assert events[2]["parent"]["span_id"] == str(
+        parent_span_id
+    ), "workflow rejected event parent should match request parent_context"
     assert (
         "Syntax Error raised while loading Workflow: "
         "unexpected indent (inputs.py, line 3)" in events[2]["body"]["error"]["message"]

{vellum_workflow_server-1.8.2 → vellum_workflow_server-1.8.2.post1}/src/workflow_server/core/executor.py

@@ -18,6 +18,7 @@ from vellum_ee.workflows.display.utils.events import event_enricher
 from vellum_ee.workflows.server.virtual_file_loader import VirtualFileFinder
 
 from vellum.workflows import BaseWorkflow
+from vellum.workflows.context import execution_context
 from vellum.workflows.emitters.base import BaseWorkflowEmitter
 from vellum.workflows.emitters.vellum_emitter import VellumEmitter
 from vellum.workflows.events.exception_handling import stream_initialization_exception
@@ -178,7 +179,12 @@ def stream_workflow(
         )
     except WorkflowInitializationException as e:
         cancel_watcher_kill_switch.set()
-        initialization_exception_stream = stream_initialization_exception(e)
+
+        with execution_context(
+            parent_context=executor_context.execution_context.parent_context,
+            trace_id=executor_context.execution_context.trace_id,
+        ):
+            initialization_exception_stream = stream_initialization_exception(e)
 
         def _stream_generator() -> Generator[dict[str, Any], Any, None]:
             for event in initialization_exception_stream: