vellum-workflow-server 1.5.3__tar.gz → 1.5.3.post1__tar.gz

{vellum_workflow_server-1.5.3 → vellum_workflow_server-1.5.3.post1}/PKG-INFO

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

{vellum_workflow_server-1.5.3 → vellum_workflow_server-1.5.3.post1}/pyproject.toml

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

{vellum_workflow_server-1.5.3 → vellum_workflow_server-1.5.3.post1}/src/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,