@@ -93,11 +93,11 @@ def run_node(self, invocation: BaseInvocation, queue_item: SessionQueueItem):
93
93
)
94
94
95
95
# Invoke the node
96
- outputs = invocation .invoke_internal (context = context , services = self ._services )
97
- # Save outputs and history
98
- queue_item .session .complete (invocation .id , outputs )
96
+ output = invocation .invoke_internal (context = context , services = self ._services )
97
+ # Save output and history
98
+ queue_item .session .complete (invocation .id , output )
99
99
100
- self ._on_after_run_node (invocation , queue_item , outputs )
100
+ self ._on_after_run_node (invocation , queue_item , output )
101
101
102
102
except KeyboardInterrupt :
103
103
# TODO(MM2): Create an event for this
@@ -172,7 +172,7 @@ def _on_before_run_node(self, invocation: BaseInvocation, queue_item: SessionQue
172
172
callback (invocation , queue_item )
173
173
174
174
def _on_after_run_node (
175
- self , invocation : BaseInvocation , queue_item : SessionQueueItem , outputs : BaseInvocationOutput
175
+ self , invocation : BaseInvocation , queue_item : SessionQueueItem , output : BaseInvocationOutput
176
176
):
177
177
"""Run after a node is executed"""
178
178
# Send complete event on successful runs
@@ -183,11 +183,11 @@ def _on_after_run_node(
183
183
graph_execution_state_id = queue_item .session .id ,
184
184
node = invocation .model_dump (),
185
185
source_node_id = queue_item .session .prepared_source_mapping [invocation .id ],
186
- result = outputs .model_dump (),
186
+ result = output .model_dump (),
187
187
)
188
188
189
189
for callback in self ._on_after_run_node_callbacks :
190
- callback (invocation , queue_item , outputs )
190
+ callback (invocation , queue_item , output )
191
191
192
192
def _on_node_error (
193
193
self ,
0 commit comments