From 3613bbc9e8bbb68427a644b173f47e402fbd3c20 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Moura?= Date: Fri, 9 Feb 2024 01:52:28 -0800 Subject: [PATCH] recreating executor upon setting new step_callback --- src/crewai/agent.py | 6 +++--- src/crewai/crew.py | 1 + 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/crewai/agent.py b/src/crewai/agent.py index e1cf749ee..495e5e15b 100644 --- a/src/crewai/agent.py +++ b/src/crewai/agent.py @@ -175,7 +175,7 @@ class Agent(BaseModel): """ self.cache_handler = cache_handler self.tools_handler = ToolsHandler(cache=self.cache_handler) - self._create_agent_executor() + self.create_agent_executor() def set_rpm_controller(self, rpm_controller: RPMController) -> None: """Set the rpm controller for the agent. @@ -185,9 +185,9 @@ class Agent(BaseModel): """ if not self._rpm_controller: self._rpm_controller = rpm_controller - self._create_agent_executor() + self.create_agent_executor() - def _create_agent_executor(self) -> None: + def create_agent_executor(self) -> None: """Create an agent executor for the agent. Returns: diff --git a/src/crewai/crew.py b/src/crewai/crew.py index e1609ccd9..804a003a1 100644 --- a/src/crewai/crew.py +++ b/src/crewai/crew.py @@ -177,6 +177,7 @@ class Crew(BaseModel): agent.i18n = I18N(language=self.language) if (self.step_callback) and (not agent.step_callback): agent.step_callback = self.step_callback + agent.create_agent_executor() if self.process == Process.sequential: return self._run_sequential_process()