vellum-workflow-server 1.7.2.post2__tar.gz → 1.7.2.post3__tar.gz

{vellum_workflow_server-1.7.2.post2 → vellum_workflow_server-1.7.2.post3}/PKG-INFO

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

{vellum_workflow_server-1.7.2.post2 → vellum_workflow_server-1.7.2.post3}/pyproject.toml

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

{vellum_workflow_server-1.7.2.post2 → vellum_workflow_server-1.7.2.post3}/src/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"}