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

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

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

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

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

{vellum_workflow_server-1.7.8 → vellum_workflow_server-1.7.8.post2}/src/workflow_server/api/workflow_view.py

@@ -319,7 +319,15 @@ def stream_workflow_route() -> Response:
             # These can happen either from Vembda disconnects (possibily from predict disconnects) or
             # from knative activator gateway timeouts which are caused by idleTimeout or responseStartSeconds
             # being exceeded.
-            app.logger.error("Client disconnected in the middle of the Workflow Stream")
+            app.logger.error(
+                "Client disconnected in the middle of the Workflow Stream",
+                extra={
+                    "sentry_tags": {
+                        "server_version": vembda_initiated_event.body.server_version,
+                        "sdk_version": vembda_initiated_event.body.sdk_version,
+                    }
+                },
+            )
             return
         except Exception as e:
             logger.exception("Error during workflow response stream generator", extra={"error": e})
@@ -535,5 +543,13 @@ def startup_error_generator(
 
         logger.error("Workflow stream could not start from resource constraints")
     except GeneratorExit:
-        app.logger.error("Client disconnected in the middle of the Startup Error Stream")
+        app.logger.error(
+            "Client disconnected in the middle of the Startup Error Stream",
+            extra={
+                "sentry_tags": {
+                    "server_version": vembda_initiated_event.body.server_version,
+                    "sdk_version": vembda_initiated_event.body.sdk_version,
+                }
+            },
+        )
         return

{vellum_workflow_server-1.7.8 → vellum_workflow_server-1.7.8.post2}/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
@@ -150,7 +148,6 @@ def stream_workflow(
         workflow, namespace = _gather_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,
@@ -492,11 +487,3 @@ def _gather_workflow(context: BaseExecutorContext) -> Tuple[BaseWorkflow, str]:
             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 → vellum_workflow_server-1.7.8.post2}/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: