andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-11 00:58:30 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix: Step callback issue

Browse Source
This commit is contained in:
Eduardo Chiarotti
2024-11-13 10:03:11 -03:00
parent bcfcf88e78
commit d8f271daeb
1 changed files with 18 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
src/crewai/agents/crew_agent_executor.py
View File

@@ -145,25 +145,25 @@ class CrewAgentExecutor(CrewAgentExecutorMixin):
formatted_answer.result = action_result formatted_answer.result = action_result
self._show_logs(formatted_answer) self._show_logs(formatted_answer)
if self.step_callback: if self.step_callback:
self.step_callback(formatted_answer) self.step_callback(formatted_answer)
if self._should_force_answer(): if self._should_force_answer():
if self.have_forced_answer: if self.have_forced_answer:
return AgentFinish( return AgentFinish(
output=self._i18n.errors( output=self._i18n.errors(
"force_final_answer_error" "force_final_answer_error"
).format(formatted_answer.text), ).format(formatted_answer.text),
text=formatted_answer.text, text=formatted_answer.text,
) )
else: else:
formatted_answer.text += ( formatted_answer.text += (
f'\n{self._i18n.errors("force_final_answer")}' f'\n{self._i18n.errors("force_final_answer")}'
) )
self.have_forced_answer = True self.have_forced_answer = True
self.messages.append( self.messages.append(
self._format_msg(formatted_answer.text, role="assistant") self._format_msg(formatted_answer.text, role="assistant")
) )
except OutputParserException as e: except OutputParserException as e:
self.messages.append({"role": "user", "content": e.error}) self.messages.append({"role": "user", "content": e.error})