andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-01 23:32:39 +00:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

include agentFinish as part of step callback

Browse Source
This commit is contained in:
João Moura
2024-02-09 02:00:41 -08:00
parent 3613bbc9e8
commit a9da07dc5d
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/crewai/agents/executor.py
View File

@@ -64,14 +64,15 @@ class CrewAgentExecutor(AgentExecutor):
intermediate_steps, intermediate_steps,
run_manager=run_manager, run_manager=run_manager,
) )
if self.step_callback:
self.step_callback(next_step_output)
if isinstance(next_step_output, AgentFinish): if isinstance(next_step_output, AgentFinish):
return self._return( return self._return(
next_step_output, intermediate_steps, run_manager=run_manager 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) intermediate_steps.extend(next_step_output)
if len(next_step_output) == 1: if len(next_step_output) == 1:
next_step_action = next_step_output[0] next_step_action = next_step_output[0]