vellum-workflow-server 1.5.3__py3-none-any.whl → 1.5.4__py3-none-any.whl

{vellum_workflow_server-1.5.3.dist-info → vellum_workflow_server-1.5.4.dist-info}/METADATA

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

{vellum_workflow_server-1.5.3.dist-info → vellum_workflow_server-1.5.4.dist-info}/RECORD

@@ -12,7 +12,7 @@ workflow_server/config.py,sha256=qmmTr6ty3ZN5LDOFs3TfUxYshYe6Mmn_LanplHHeE9Q,179
 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=zr5iruCsAMI5yHoW_yWzAhCo1U4Si9Kr9tjOmBKQf4A,20102
+workflow_server/core/executor.py,sha256=Bw6luelJs84-_fBIG9VvxYFxeha4DmwxDpslVeqr2I8,18411
 workflow_server/core/utils.py,sha256=si0NB4Suurc-mn8NYdn59xM9CkPrfOP1aWEVrZvifDI,1929
 workflow_server/core/workflow_executor_context.py,sha256=w0nFhtu54GX36ZzmaqLTD3-ssPtoFd9QX02dPgeIbBM,3615
 workflow_server/server.py,sha256=QBU12AaAfAgLqfCDBd24qIJl_mbheiq0-hfcWV7rZM4,1234
@@ -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=0Nq6du8o-iBtTrip9_wgHES53JSiJbVdSXaBnPobw3s,6930
 workflow_server/utils/utils.py,sha256=ZPoM1Suhid22dpB8oEFLux8wx-9iyzmSfWuYxSCrgWk,4774
-vellum_workflow_server-1.5.3.dist-info/METADATA,sha256=C5KSsVZqP-H0jzKalUsiOnYdxcGIf6GcxJwcm1SPWvI,2267
-vellum_workflow_server-1.5.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_workflow_server-1.5.3.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
-vellum_workflow_server-1.5.3.dist-info/RECORD,,
+vellum_workflow_server-1.5.4.dist-info/METADATA,sha256=4dW7Mupn81DeVG3BVZcETkVlr7lkX_cUG6AnHwhGQB0,2267
+vellum_workflow_server-1.5.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_workflow_server-1.5.4.dist-info/entry_points.txt,sha256=uB_0yPkr7YV6RhEXzvFReUM8P4OQBlVXD6TN6eb9-oc,277
+vellum_workflow_server-1.5.4.dist-info/RECORD,,

workflow_server/core/executor.py

@@ -23,14 +23,6 @@ from vellum.workflows import BaseWorkflow
 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.node import (
-    NodeExecutionFulfilledBody,
-    NodeExecutionFulfilledEvent,
-    NodeExecutionInitiatedBody,
-    NodeExecutionInitiatedEvent,
-    NodeExecutionStreamingBody,
-    NodeExecutionStreamingEvent,
-)
 from vellum.workflows.events.types import BaseEvent
 from vellum.workflows.events.workflow import WorkflowEventDisplayContext
 from vellum.workflows.exceptions import WorkflowInitializationException
@@ -290,51 +282,11 @@ def stream_node(
             workflow_definition=workflow.__class__,
         )
 
-    workflow_inputs = _get_workflow_inputs(executor_context, workflow.__class__)
-    workflow_state = _get_workflow_state(executor_context, workflow_inputs=workflow_inputs)
-    node = Node(
-        state=workflow_state,
-        context=workflow._context,
-    )
-
     def call_node() -> Generator[dict[str, Any], Any, None]:
         executor_context.stream_start_time = time.time_ns()
-        span_id = uuid4()
-        yield NodeExecutionInitiatedEvent(
-            id=uuid4(),
-            timestamp=datetime.now(),
-            trace_id=executor_context.trace_id,
-            span_id=span_id,
-            body=NodeExecutionInitiatedBody(
-                node_definition=Node,
-                inputs=node._inputs,
-            ),
-        ).model_dump(mode="json")
-        node_outputs = node.run()
-
-        if isinstance(node_outputs, (Iterator)):
-            for node_output in node_outputs:
-                yield NodeExecutionStreamingEvent(
-                    id=uuid4(),
-                    timestamp=datetime.now(),
-                    trace_id=executor_context.trace_id,
-                    span_id=span_id,
-                    body=NodeExecutionStreamingBody(
-                        node_definition=Node,
-                        output=node_output,
-                    ),
-                ).model_dump(mode="json")
-        else:
-            yield NodeExecutionFulfilledEvent(
-                id=uuid4(),
-                timestamp=datetime.now(),
-                trace_id=executor_context.trace_id,
-                span_id=span_id,
-                body=NodeExecutionFulfilledBody(
-                    node_definition=Node,
-                    outputs=node_outputs,
-                ),
-            ).model_dump(mode="json")
+
+        for event in workflow.run_node(Node, inputs=executor_context.inputs):  # type: ignore[arg-type]
+            yield event.model_dump(mode="json")
 
     return _call_stream(
         executor_context=executor_context,