vellum-workflow-server 1.8.2.post1__tar.gz → 1.8.3__tar.gz

{vellum_workflow_server-1.8.2.post1 → vellum_workflow_server-1.8.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-workflow-server
-Version: 1.8.2.post1
+Version: 1.8.3
 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.8.2)
+Requires-Dist: vellum-ai (==1.8.3)
 Description-Content-Type: text/markdown
 
 # Vellum Workflow Runner Server

{vellum_workflow_server-1.8.2.post1 → vellum_workflow_server-1.8.3}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-workflow-server"
 
 [tool.poetry]
 name = "vellum-workflow-server"
-version = "1.8.2.post1"
+version = "1.8.3"
 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.8.2"
+vellum-ai = "1.8.3"
 python-dotenv = "1.0.1"
 retrying = "1.3.4"
 sentry-sdk = {extras = ["flask"], version = "2.20.0"}

{vellum_workflow_server-1.8.2.post1 → vellum_workflow_server-1.8.3}/src/workflow_server/api/tests/test_workflow_view_stream_workflow_route.py

@@ -174,6 +174,8 @@ class Workflow(BaseWorkflow):
     assert "sdk_version" in server_metadata
     assert "memory_usage_mb" in server_metadata
     assert isinstance(server_metadata["memory_usage_mb"], (int, float))
+    assert "is_new_server" in server_metadata
+    assert server_metadata["is_new_server"] is False
 
     assert events[2]["name"] == "workflow.execution.fulfilled", events[2]
     assert events[2]["body"]["workflow_definition"]["module"] == ["test", "workflow"]

{vellum_workflow_server-1.8.2.post1 → vellum_workflow_server-1.8.3}/src/workflow_server/api/workflow_view.py

@@ -90,12 +90,13 @@ def stream_workflow_route() -> Response:
     )
 
     # Create this event up here so timestamps are fully from the start to account for any unknown overhead
+    version_data = get_version()
     vembda_initiated_event = VembdaExecutionInitiatedEvent(
         id=uuid4(),
         timestamp=datetime.now(),
         trace_id=context.trace_id,
         span_id=context.execution_id,
-        body=VembdaExecutionInitiatedBody.model_validate(get_version()),
+        body=VembdaExecutionInitiatedBody.model_validate(version_data),
         parent=None,
     )
 
@@ -374,12 +375,13 @@ def stream_node_route() -> Response:
         )
 
     # Create this event up here so timestamps are fully from the start to account for any unknown overhead
+    version_data = get_version()
     vembda_initiated_event = VembdaExecutionInitiatedEvent(
         id=uuid4(),
         timestamp=datetime.now(),
         trace_id=context.trace_id,
         span_id=context.execution_id,
-        body=VembdaExecutionInitiatedBody.model_validate(get_version()),
+        body=VembdaExecutionInitiatedBody.model_validate(version_data),
         parent=None,
     )
 

{vellum_workflow_server-1.8.2.post1 → vellum_workflow_server-1.8.3}/src/workflow_server/core/executor.py

@@ -411,7 +411,7 @@ def get_random_namespace() -> str:
     return "workflow_tmp_" + "".join(random.choice(string.ascii_letters + string.digits) for i in range(14))
 
 
-def _enrich_event(event: WorkflowEvent, vellum_client: Optional[Any]) -> WorkflowEvent:
+def _enrich_event(event: WorkflowEvent, executor_context: Optional[BaseExecutorContext] = None) -> WorkflowEvent:
     """
     Enrich an event with metadata based on the event type.
 
@@ -431,6 +431,9 @@ def _enrich_event(event: WorkflowEvent, vellum_client: Optional[Any]) -> Workflo
             memory_mb = get_memory_in_use_mb()
             if memory_mb is not None:
                 metadata["memory_usage_mb"] = memory_mb
+
+            if executor_context is not None:
+                metadata["is_new_server"] = executor_context.is_new_server
         elif event.name == "workflow.execution.fulfilled" and is_deployment:
             metadata = {}
             memory_mb = get_memory_in_use_mb()
@@ -439,6 +442,7 @@ def _enrich_event(event: WorkflowEvent, vellum_client: Optional[Any]) -> Workflo
     except Exception:
         pass
 
+    vellum_client = executor_context.vellum_client if executor_context else None
     return event_enricher(event, vellum_client, metadata=metadata)
 
 
@@ -447,7 +451,7 @@ def _dump_event(event: BaseEvent, executor_context: BaseExecutorContext) -> dict
 
     dump = event.model_dump(
         mode="json",
-        context={"event_enricher": lambda event: _enrich_event(event, executor_context.vellum_client)},
+        context={"event_enricher": lambda event: _enrich_event(event, executor_context)},
     )
     if dump["name"] in {
         "workflow.execution.initiated",