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

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

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

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

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

{vellum_workflow_server-1.3.0 → vellum_workflow_server-1.3.0.post2}/src/workflow_server/core/executor.py

@@ -12,20 +12,25 @@ from threading import Event as ThreadingEvent
 import time
 from traceback import format_exc
 from uuid import UUID, uuid4
-from typing import TYPE_CHECKING, Any, Callable, Generator, Iterator, Optional, Tuple, Type
+from typing import Any, Callable, Generator, Iterator, Optional, Tuple, Type
 
 from pebble import concurrent
+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, VellumEnvironment
 from vellum.workflows import BaseWorkflow
+from vellum.workflows.emitters.base import BaseWorkflowEmitter
+from vellum.workflows.emitters.vellum_emitter import VellumEmitter
 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
 from vellum.workflows.nodes.mocks import MockNodeExecution
+from vellum.workflows.resolvers.base import BaseWorkflowResolver
+from vellum.workflows.resolvers.resolver import VellumResolver
 from vellum.workflows.state.base import BaseState, StateMeta
 from vellum.workflows.state.context import WorkflowContext
 from vellum.workflows.state.store import EmptyStore
@@ -47,10 +52,6 @@ from workflow_server.core.workflow_executor_context import (
 )
 from workflow_server.utils.log_proxy import redirect_log
 
-if TYPE_CHECKING:
-    from vellum.workflows.emitters.base import BaseWorkflowEmitter
-
-
 logger = logging.getLogger(__name__)
 
 
@@ -344,25 +345,21 @@ def _create_workflow(executor_context: WorkflowExecutorContext, namespace: str)
 
     # Determine whether to enable the Vellum Emitter for event publishing
     use_vellum_emitter = bool((executor_context.feature_flags or {}).get("vembda-event-emitting-enabled"))
-
     emitters: list["BaseWorkflowEmitter"] = []
     if use_vellum_emitter:
-        from vellum.workflows.emitters.vellum_emitter import VellumEmitter
+        emitters = [VellumEmitter()]
 
-        if VellumEmitter is not None:
-            try:
-                emitters = [VellumEmitter()]
-            except Exception:
-                emitters = []
-        else:
-            # Emitter is not available at import time; continue without it
-            logger.warning("VellumEmitter not available; continuing without it")
+    use_vellum_resolver = executor_context.previous_execution_id is not None
+    resolvers: list["BaseWorkflowResolver"] = []
+    if use_vellum_resolver:
+        resolvers = [VellumResolver()]
 
     # Explicit constructor call to satisfy typing
     workflow = Workflow(
         context=workflow_context,
         store=EmptyStore(),
-        emitters=emitters or None,
+        emitters=emitters,
+        resolvers=resolvers,
     )
 
     return workflow
@@ -410,7 +407,7 @@ def _get_file_namespace(executor_context: BaseExecutorContext) -> str:
 
 def _dump_event(event: BaseEvent, executor_context: BaseExecutorContext) -> dict:
     module_base = executor_context.module.split(".")
-    dump = event.model_dump(mode="json")
+    dump = event.model_dump(mode="json", context={"event_enricher": event_enricher})
     if dump["name"] in {
         "workflow.execution.initiated",
         "workflow.execution.fulfilled",