vellum-workflow-server 1.1.2__py3-none-any.whl → 1.1.2.post1__py3-none-any.whl

{vellum_workflow_server-1.1.2.dist-info → vellum_workflow_server-1.1.2.post1.dist-info}/METADATA

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

{vellum_workflow_server-1.1.2.dist-info → vellum_workflow_server-1.1.2.post1.dist-info}/RECORD

@@ -6,15 +6,15 @@ workflow_server/api/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 workflow_server/api/tests/test_input_display_mapping.py,sha256=drBZqMudFyB5wgiUOcMgRXz7E7ge-Qgxbstw4E4f0zE,2211
 workflow_server/api/tests/test_workflow_view.py,sha256=UxOwN9J3tI2YEXCGKsyMf890NiPnTNo0Rxyeidmx3oY,18620
 workflow_server/api/tests/test_workflow_view_stream_workflow_route.py,sha256=7q5AiBzfXX-npCrp-Da2OWk6-M1Ai8QZPDLhLlkxb34,26051
-workflow_server/api/workflow_view.py,sha256=qY-0aTB-NeYNJ4RUSpmx7onKP_uODFWx6OeXB2Ns0_4,20631
+workflow_server/api/workflow_view.py,sha256=i19Ksgg_2EbPhPRAlwYRqmAKI_8VHvAGc61Q9k90Iow,21189
 workflow_server/code_exec_runner.py,sha256=lBnMIorPZL8zZBye6TjeCIs06WTJM7P2HR07B1fjJJI,2533
 workflow_server/config.py,sha256=DyTty8NrAwvtx-esM3KthnpsNh-nKdWNlovWQOgiGpg,1417
 workflow_server/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 workflow_server/core/cancel_workflow.py,sha256=Ffkc3mzmrdMEUcD-sHfEhX4IwVrka-E--SxKA1dUfIU,2185
 workflow_server/core/events.py,sha256=24MA66DVQuaLJJcZrS8IL1Zq4Ohi9CoouKZ5VgoH3Cs,1402
-workflow_server/core/executor.py,sha256=-LXwbM9NwuyU55EaGK6O-FJnFoWzlLyMGgKx-IS2GBM,16954
+workflow_server/core/executor.py,sha256=68AINc5SlmJTrb0vvDxiPgc-qqgX1oVCKKKrVCKzfPQ,18046
 workflow_server/core/utils.py,sha256=lgzxkAEjEXPxGXXQlUYTYuCdHht-eDJJmHj5AhEb3_o,1500
-workflow_server/core/workflow_executor_context.py,sha256=kqbHDAteyeyB5-ISJKS2d3Bafm7mndhIjovCqNNnFPE,1486
+workflow_server/core/workflow_executor_context.py,sha256=SMbCRhYsjrxnlEUZ2lJg2WqnhxDBrZzlQwjpqnBzbOg,1538
 workflow_server/server.py,sha256=QBU12AaAfAgLqfCDBd24qIJl_mbheiq0-hfcWV7rZM4,1234
 workflow_server/start.py,sha256=qpIg0SgIgz8RNyc8Cu9LxyzXdOXZRv9qq3M3uSBbgD0,2180
 workflow_server/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -28,7 +28,7 @@ workflow_server/utils/tests/test_sentry_integration.py,sha256=LGmWiaLhFrx-jslrRj
 workflow_server/utils/tests/test_system_utils.py,sha256=_4GwXvVvU5BrATxUEWwQIPg0bzQXMWBtiBmjP8MTxJM,4314
 workflow_server/utils/tests/test_utils.py,sha256=qwK5Rmy3RQyjtlUrYAuGuDlBeRzZKsf1yS-y2IpUizQ,6452
 workflow_server/utils/utils.py,sha256=Wqqn-1l2ugkGgy5paWWdt0AVxAyPMQCYcnRSSOMjXlA,4355
-vellum_workflow_server-1.1.2.dist-info/METADATA,sha256=VBP71cRJ7UEXHzkobqV4X1VDZp55p63Pce0J18Rv4YI,2267
-vellum_workflow_server-1.1.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-1.1.2.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-1.1.2.dist-info/RECORD,,
+vellum_workflow_server-1.1.2.post1.dist-info/METADATA,sha256=6fJa1OW8XEwsBnu6axLZTDNzRiXkP_AoZAhonFZt3Yo,2273
+vellum_workflow_server-1.1.2.post1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-1.1.2.post1.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-1.1.2.post1.dist-info/RECORD,,

workflow_server/api/workflow_view.py

@@ -61,6 +61,17 @@ CUSTOM_NODES_DIRECTORY = "vellum_custom_nodes"
 WORKFLOW_INITIATION_TIMEOUT_SECONDS = 60
 
 
+def _events_emission_enabled(context: Union[WorkflowExecutorContext, NodeExecutorContext]) -> bool:
+    flags = context.feature_flags
+    if not (flags and flags.get("vembda-event-emitting-enabled")):
+        return False
+    try:
+        module = importlib.import_module("vellum.workflows.events.emitters.vellum")
+        return hasattr(module, "VellumEmitter")
+    except Exception:
+        return False
+
+
 @bp.route("/stream", methods=["POST"])
 def stream_workflow_route() -> Response:
     data = request.get_json()
@@ -97,6 +108,7 @@ def stream_workflow_route() -> Response:
     headers = {
         "X-Vellum-SDK-Version": vembda_initiated_event.body.sdk_version,
         "X-Vellum-Server-Version": vembda_initiated_event.body.server_version,
+        "X-Vellum-Events-Emitted": str(_events_emission_enabled(context)),
     }
 
     # We can exceed the concurrency count currently with long running workflows due to a knative issue. So here
@@ -403,6 +415,7 @@ def stream_node_route() -> Response:
     headers = {
         "X-Vellum-SDK-Version": vembda_initiated_event.body.sdk_version,
         "X-Vellum-Server-Version": vembda_initiated_event.body.server_version,
+        "X-Vellum-Events-Emitted": str(_events_emission_enabled(context)),
     }
 
     resp = Response(

workflow_server/core/executor.py

@@ -12,7 +12,7 @@ from threading import Event as ThreadingEvent
 import time
 from traceback import format_exc
 from uuid import UUID, uuid4
-from typing import Any, Callable, Generator, Iterator, Optional, Tuple, Type
+from typing import TYPE_CHECKING, Any, Callable, Generator, Iterator, Optional, Tuple, Type
 
 from pebble import concurrent
 from vellum_ee.workflows.display.workflows import BaseWorkflowDisplay
@@ -47,6 +47,15 @@ from workflow_server.core.workflow_executor_context import (
 )
 from workflow_server.utils.log_proxy import redirect_log
 
+if TYPE_CHECKING:
+    # Imported for typing only to avoid runtime dependency issues
+    from vellum.workflows.events.emitters.base import BaseWorkflowEmitter
+
+try:
+    from vellum.workflows.events.emitters.vellum import VellumEmitter as _VellumEmitter
+except Exception:
+    _VellumEmitter = None
+
 logger = logging.getLogger(__name__)
 
 
@@ -331,13 +340,35 @@ def _create_workflow(executor_context: WorkflowExecutorContext, namespace: str)
     VembdaExecutionFulfilledEvent.model_rebuild(
         # Not sure why this is needed, but it is required for the VembdaExecutionFulfilledEvent to be
         # properly rebuilt with the recursive types.
+        # use flag here to determine which emitter to use
         _types_namespace={
             "BaseWorkflow": BaseWorkflow,
             "BaseNode": BaseNode,
         },
     )
 
-    return Workflow(context=workflow_context, store=EmptyStore())
+    # 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:
+        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")
+
+    # Explicit constructor call to satisfy typing
+    workflow = Workflow(
+        context=workflow_context,
+        store=EmptyStore(),
+        emitters=emitters or None,
+    )
+
+    return workflow
 
 
 def _create_workflow_context(executor_context: BaseExecutorContext) -> WorkflowContext:

workflow_server/core/workflow_executor_context.py

@@ -24,6 +24,7 @@ class BaseExecutorContext(UniversalBaseModel):
     vembda_public_url: Optional[str] = None
     node_output_mocks: Optional[list[Any]] = None
     environment_variables: Optional[dict[str, str]] = None
+    feature_flags: Optional[dict[str, bool]] = None
 
     @property
     def container_overhead_latency(self) -> int: