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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 1.7.8.post2
+Version: 1.7.9
 Summary: 
 License: AGPL
 Requires-Python: >=3.9.0,<4
@@ -29,7 +29,7 @@ Requires-Dist: pyjwt (==2.10.0)
 Requires-Dist: python-dotenv (==1.0.1)
 Requires-Dist: retrying (==1.3.4)
 Requires-Dist: sentry-sdk[flask] (==2.20.0)
-Requires-Dist: vellum-ai (==1.7.8)
+Requires-Dist: vellum-ai (==1.7.9)
 Description-Content-Type: text/markdown
 
 # Vellum Workflow Runner Server

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

@@ -3,7 +3,7 @@ name = "vellum-workflow-server"
 
 [tool.poetry]
 name = "vellum-workflow-server"
-version = "1.7.8.post2"
+version = "1.7.9"
 description = ""
 readme = "README.md"
 authors = []
@@ -45,7 +45,7 @@ flask = "2.3.3"
 orderly-set = "5.2.2"
 pebble = "5.0.7"
 gunicorn = "23.0.0"
-vellum-ai = "1.7.8"
+vellum-ai = "1.7.9"
 python-dotenv = "1.0.1"
 retrying = "1.3.4"
 sentry-sdk = {extras = ["flask"], version = "2.20.0"}

{vellum_workflow_server-1.7.8.post2 → vellum_workflow_server-1.7.9}/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.9}/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