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

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

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

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

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

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

@@ -421,7 +421,7 @@ from vellum.workflows.inputs import BaseInputs
     assert events[2]["name"] == "workflow.execution.rejected"
     assert events[2]["span_id"] == events[1]["span_id"]
     assert (
-        "Failed to initialize workflow: Syntax Error raised while loading Workflow: "
+        "Syntax Error raised while loading Workflow: "
         "unexpected indent (inputs.py, line 3)" in events[2]["body"]["error"]["message"]
     )
 
@@ -1178,7 +1178,7 @@ class InvalidWorkflow(BaseWorkflow):
     # AND the error message should contain information about the invalid graph structure
     error_message = events[2]["body"]["error"]["message"]
     expected_message = (
-        "Failed to initialize workflow: Invalid graph structure detected. "
+        "Invalid graph structure detected. "
         "Nested sets or unsupported graph types are not allowed. "
         "Please contact Vellum support for assistance with Workflow configuration."
     )

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

@@ -145,7 +145,7 @@ def stream_workflow(
 ) -> tuple[Iterator[dict], UUID]:
     cancel_watcher_kill_switch = ThreadingEvent()
     try:
-        workflow, namespace = _gather_workflow(executor_context)
+        workflow, namespace = _create_workflow(executor_context)
         workflow_inputs = _get_workflow_inputs(executor_context, workflow.__class__)
 
         workflow_state = (
@@ -247,7 +247,7 @@ def stream_node(
     executor_context: NodeExecutorContext,
     disable_redirect: bool = True,
 ) -> Iterator[dict]:
-    workflow, namespace = _gather_workflow(executor_context)
+    workflow, namespace = _create_workflow(executor_context)
     Node: Optional[Type[BaseNode]] = None
 
     for workflow_node in workflow.get_nodes():
@@ -331,7 +331,11 @@ def _call_stream(
         yield vembda_fulfilled_event.model_dump(mode="json")
 
 
-def _create_workflow(executor_context: BaseExecutorContext, namespace: str) -> BaseWorkflow:
+def _create_workflow(executor_context: BaseExecutorContext) -> Tuple[BaseWorkflow, str]:
+    namespace = _get_file_namespace(executor_context)
+    if namespace != LOCAL_WORKFLOW_MODULE:
+        sys.meta_path.append(VirtualFileFinder(executor_context.files, namespace))
+
     workflow_context = _create_workflow_context(executor_context)
     Workflow = BaseWorkflow.load_from_module(namespace)
     VembdaExecutionFulfilledEvent.model_rebuild(
@@ -363,7 +367,7 @@ def _create_workflow(executor_context: BaseExecutorContext, namespace: str) -> B
         resolvers=resolvers,
     )
 
-    return workflow
+    return workflow, namespace
 
 
 def _create_workflow_context(executor_context: BaseExecutorContext) -> WorkflowContext:
@@ -469,21 +473,3 @@ def _get_run_from_node(executor_context: WorkflowExecutorContext, workflow: Base
             return node
 
     return None
-
-
-def _gather_workflow(context: BaseExecutorContext) -> Tuple[BaseWorkflow, str]:
-    try:
-        namespace = _get_file_namespace(context)
-        if namespace != LOCAL_WORKFLOW_MODULE:
-            sys.meta_path.append(VirtualFileFinder(context.files, namespace))
-        workflow = _create_workflow(
-            executor_context=context,
-            namespace=namespace,
-        )
-        return workflow, namespace
-    except Exception as e:
-        logger.exception("Failed to initialize Workflow")
-        raise WorkflowInitializationException(
-            message=f"Failed to initialize workflow: {e}",
-            workflow_definition=BaseWorkflow,
-        ) from e