vellum-workflow-server 1.7.9__tar.gz → 1.7.9.post2__tar.gz

{vellum_workflow_server-1.7.9 → vellum_workflow_server-1.7.9.post2}/PKG-INFO

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

{vellum_workflow_server-1.7.9 → vellum_workflow_server-1.7.9.post2}/pyproject.toml

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

{vellum_workflow_server-1.7.9 → vellum_workflow_server-1.7.9.post2}/src/workflow_server/api/tests/test_workflow_view_stream_workflow_route.py

@@ -117,7 +117,7 @@ def test_stream_workflow_route__happy_path(both_stream_types):
         "execution_id": str(span_id),
         "inputs": [],
         "environment_api_key": "test",
-        "module": "workflow",
+        "module": "test",
         "timeout": 360,
         "files": {
             "__init__.py": "",
@@ -154,6 +154,7 @@ class Workflow(BaseWorkflow):
     }
 
     assert events[1]["name"] == "workflow.execution.initiated", events[1]
+    assert events[1]["body"]["workflow_definition"]["module"] == ["test", "workflow"]
     assert "display_context" in events[1]["body"], events[1]["body"]
     display_context = events[1]["body"]["display_context"]
     assert "node_displays" in display_context
@@ -164,6 +165,7 @@ class Workflow(BaseWorkflow):
     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"]["workflow_definition"]["module"] == ["test", "workflow"]
 
     assert events[3] == {
         "id": mock.ANY,

{vellum_workflow_server-1.7.9 → vellum_workflow_server-1.7.9.post2}/src/workflow_server/core/executor.py

@@ -156,7 +156,6 @@ def stream_workflow(
             if executor_context.state
             else None
         )
-        run_from_node = _get_run_from_node(executor_context, workflow)
         node_output_mocks = MockNodeExecution.validate_all(
             executor_context.node_output_mocks,
             workflow.__class__,
@@ -170,7 +169,7 @@ def stream_workflow(
             node_output_mocks=node_output_mocks,
             event_filter=all_workflow_event_filter,
             cancel_signal=cancel_signal,
-            entrypoint_nodes=[run_from_node] if run_from_node else None,
+            entrypoint_nodes=[executor_context.node_id] if executor_context.node_id else None,
             previous_execution_id=executor_context.previous_execution_id,
         )
     except WorkflowInitializationException as e:
@@ -462,14 +461,3 @@ but found {inputs_module.Inputs.__class__.__name__}""",
         )
 
     return inputs_module.Inputs(**executor_context.inputs)
-
-
-def _get_run_from_node(executor_context: WorkflowExecutorContext, workflow: BaseWorkflow) -> Optional[Type[BaseNode]]:
-    if not executor_context.node_id:
-        return None
-
-    for node in workflow.get_nodes():
-        if node.__id__ == executor_context.node_id:
-            return node
-
-    return None