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

code cleanup

Browse Source
This commit is contained in:
Lorenze Jay
2024-07-17 08:39:57 -07:00
parent 76607062f0
commit 2faa2dbddb
1 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/crewai/crew.py
View File

@@ -635,16 +635,13 @@ class Crew(BaseModel):
last_sync_output = task.output last_sync_output = task.output
continue continue
self._prepare_task(task, manager) agent_to_use = self._get_agent_to_use(task, manager)
if self.process == Process.hierarchical:
agent_to_use = manager
else:
agent_to_use = task.agent
if agent_to_use is None: if agent_to_use is None:
raise ValueError( raise ValueError(
f"No agent available for task: {task.description}. Ensure that either the task has an assigned agent or a manager agent is provided." f"No agent available for task: {task.description}. Ensure that either the task has an assigned agent or a manager agent is provided."
) )
self._log_task_start(task, agent_to_use) self._prepare_agent_tools(task, manager)
self._log_task_start(task, agent_to_use.role)
if isinstance(task, ConditionalTask): if isinstance(task, ConditionalTask):
skipped_task_output = self._handle_conditional_task( skipped_task_output = self._handle_conditional_task(
@@ -711,7 +708,7 @@ class Crew(BaseModel):
return skipped_task_output return skipped_task_output
return None return None
def _prepare_task(self, task: Task, manager: Optional[BaseAgent]): def _prepare_agent_tools(self, task: Task, manager: Optional[BaseAgent]):
if self.process == Process.hierarchical: if self.process == Process.hierarchical:
if manager: if manager:
self._update_manager_tools(task, manager) self._update_manager_tools(task, manager)
@@ -720,6 +717,13 @@ class Crew(BaseModel):
elif task.agent and task.agent.allow_delegation: elif task.agent and task.agent.allow_delegation:
self._add_delegation_tools(task) self._add_delegation_tools(task)
def _get_agent_to_use(
self, task: Task, manager: Optional[BaseAgent]
) -> Optional[BaseAgent]:
if self.process == Process.hierarchical:
return manager
return task.agent
def _add_delegation_tools(self, task: Task): def _add_delegation_tools(self, task: Task):
agents_for_delegation = [agent for agent in self.agents if agent != task.agent] agents_for_delegation = [agent for agent in self.agents if agent != task.agent]
if len(self.agents) > 1 and len(agents_for_delegation) > 0 and task.agent: if len(self.agents) > 1 and len(agents_for_delegation) > 0 and task.agent: