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

recreating executor upon setting new step_callback

Browse Source
This commit is contained in:
João Moura
2024-02-09 01:52:28 -08:00
parent a1e0596450
commit d36c2a74cb
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/crewai/agent.py
View File

@@ -175,7 +175,7 @@ class Agent(BaseModel):
""" """
self.cache_handler = cache_handler self.cache_handler = cache_handler
self.tools_handler = ToolsHandler(cache=self.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: def set_rpm_controller(self, rpm_controller: RPMController) -> None:
"""Set the rpm controller for the agent. """Set the rpm controller for the agent.
@@ -185,9 +185,9 @@ class Agent(BaseModel):
""" """
if not self._rpm_controller: if not self._rpm_controller:
self._rpm_controller = 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. """Create an agent executor for the agent.
Returns: Returns:

1
src/crewai/crew.py
View File

@@ -177,6 +177,7 @@ class Crew(BaseModel):
agent.i18n = I18N(language=self.language) agent.i18n = I18N(language=self.language)
if (self.step_callback) and (not agent.step_callback): if (self.step_callback) and (not agent.step_callback):
agent.step_callback = self.step_callback agent.step_callback = self.step_callback
agent.create_agent_executor()
if self.process == Process.sequential: if self.process == Process.sequential:
return self._run_sequential_process() return self._run_sequential_process()