From a9da07dc5dd36c54ee0df8a03525c00128528021 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Moura?= Date: Fri, 9 Feb 2024 02:00:41 -0800 Subject: [PATCH] include agentFinish as part of step callback --- src/crewai/agents/executor.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/crewai/agents/executor.py b/src/crewai/agents/executor.py index dab080d3a..1dbe224d6 100644 --- a/src/crewai/agents/executor.py +++ b/src/crewai/agents/executor.py @@ -64,14 +64,15 @@ class CrewAgentExecutor(AgentExecutor): intermediate_steps, run_manager=run_manager, ) + + if self.step_callback: + self.step_callback(next_step_output) + if isinstance(next_step_output, AgentFinish): return self._return( next_step_output, intermediate_steps, run_manager=run_manager ) - if self.step_callback: - self.step_callback(next_step_output) - intermediate_steps.extend(next_step_output) if len(next_step_output) == 1: next_step_action = next_step_output[0]