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

{vellum_workflow_server-1.7.8.post1 → 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.post1
+Version: 1.7.8.post3
 Summary: 
 License: AGPL
 Requires-Python: >=3.9.0,<4

{vellum_workflow_server-1.7.8.post1 → 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.post1"
+version = "1.7.8.post3"
 description = ""
 readme = "README.md"
 authors = []

{vellum_workflow_server-1.7.8.post1 → 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.post1 → vellum_workflow_server-1.7.8.post3}/src/workflow_server/core/executor.py

@@ -15,7 +15,6 @@ from uuid import UUID, uuid4
 from typing import Any, Callable, Generator, Iterator, Optional, Tuple, Type
 
 from vellum_ee.workflows.display.utils.events import event_enricher
-from vellum_ee.workflows.display.workflows import BaseWorkflowDisplay
 from vellum_ee.workflows.server.virtual_file_loader import VirtualFileFinder
 
 from vellum.workflows import BaseWorkflow
@@ -23,7 +22,6 @@ 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
 from vellum.workflows.events.types import BaseEvent
-from vellum.workflows.events.workflow import WorkflowEventDisplayContext
 from vellum.workflows.exceptions import WorkflowInitializationException
 from vellum.workflows.inputs import BaseInputs
 from vellum.workflows.nodes import BaseNode
@@ -147,10 +145,9 @@ 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__)
 
-        display_context = _gather_display_context(workflow, namespace)
         workflow_state = (
             workflow.deserialize_state(
                 executor_context.state,
@@ -222,8 +219,6 @@ def stream_workflow(
                 if first:
                     executor_context.stream_start_time = time.time_ns()
                     first = False
-                    if event.name == "workflow.execution.initiated":
-                        event.body.display_context = display_context
 
                 yield _dump_event(
                     event=event,
@@ -252,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():
@@ -336,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(
@@ -368,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:
@@ -474,29 +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
-
-
-def _gather_display_context(workflow: BaseWorkflow, namespace: str) -> Optional["WorkflowEventDisplayContext"]:
-    try:
-        return BaseWorkflowDisplay.gather_event_display_context(namespace, workflow.__class__)
-    except Exception:
-        logger.exception("Unable to Parse Workflow Display Context")
-        return None

{vellum_workflow_server-1.7.8.post1 → vellum_workflow_server-1.7.8.post3}/src/workflow_server/start.py

@@ -32,11 +32,17 @@ class CustomGunicornLogger(glogging.Logger):
 
         logger = logging.getLogger("gunicorn.access")
         logger.addFilter(HealthCheckFilter())
+        logger.addFilter(SignalFilter())
 
 
 class HealthCheckFilter(logging.Filter):
     def filter(self, record: Any) -> bool:
-        return "GET /healthz" not in record.getMessage()
+        return "Handling signal: winch" not in record.getMessage()
+
+
+class SignalFilter(logging.Filter):
+    def filter(self, record: Any) -> bool:
+        return "SIGTERM" not in record.getMessage()
 
 
 def start() -> None: