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

fixed hierarchial manager tools when assigned an agent

Browse Source
This commit is contained in:
Lorenze Jay
2024-07-16 14:00:25 -07:00
parent b0c2b15a3e
commit e8cbdb7881
4 changed files with 19297 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/crewai/crew.py
View File

@@ -712,7 +712,8 @@ class Crew(BaseModel):
def _prepare_task(self, task: Task, manager: Optional[BaseAgent]):
if self.process == Process.hierarchical:
self._update_manager_tools(task, manager)
if manager:
self._update_manager_tools(task, manager)
elif task.agent and task.agent.allow_delegation:
self._add_delegation_tools(task)
@@ -750,10 +751,10 @@ class Crew(BaseModel):
if self.output_log_file:
self._file_handler.log(agent=role, task=task.description, status="started")
def _update_manager_tools(self, task: Task, manager: Optional[BaseAgent]):
if task.agent and manager:
def _update_manager_tools(self, task: Task, manager: BaseAgent):
if task.agent:
manager.tools = task.agent.get_delegation_tools([task.agent])
if manager:
else:
manager.tools = manager.get_delegation_tools(self.agents)
def _get_context(self, task: Task, task_outputs: List[TaskOutput]):