vellum-workflow-server 1.3.9.post2__tar.gz → 1.3.10.post1__tar.gz

{vellum_workflow_server-1.3.9.post2 → vellum_workflow_server-1.3.10.post1}/PKG-INFO

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

{vellum_workflow_server-1.3.9.post2 → vellum_workflow_server-1.3.10.post1}/pyproject.toml

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

{vellum_workflow_server-1.3.9.post2 → vellum_workflow_server-1.3.10.post1}/src/workflow_server/core/executor.py

@@ -20,6 +20,7 @@ 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 import Vellum
 from vellum.workflows import BaseWorkflow
 from vellum.workflows.emitters.base import BaseWorkflowEmitter
 from vellum.workflows.emitters.vellum_emitter import VellumEmitter
@@ -233,6 +234,7 @@ def stream_workflow(
                 yield _dump_event(
                     event=event,
                     executor_context=executor_context,
+                    client=workflow.context.vellum_client,
                 )
         except Exception as e:
             logger.exception("Failed to generate event from Workflow Stream")
@@ -396,9 +398,9 @@ def _get_file_namespace(executor_context: BaseExecutorContext) -> str:
     )
 
 
-def _dump_event(event: BaseEvent, executor_context: BaseExecutorContext) -> dict:
+def _dump_event(event: BaseEvent, executor_context: BaseExecutorContext, client: Vellum) -> dict:
     module_base = executor_context.module.split(".")
-    dump = event.model_dump(mode="json", context={"event_enricher": event_enricher})
+    dump = event.model_dump(mode="json", context={"event_enricher": lambda event: event_enricher(event, client)})
     if dump["name"] in {
         "workflow.execution.initiated",
         "workflow.execution.fulfilled",