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

adding more specific guidelines to agent delegation tools

Browse Source
This commit is contained in:
João Moura
2023-12-20 22:36:07 -03:00
parent 1e7bda63bc
commit 9d0f41f32a
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
crewai/tools/agent_tools.py
View File

@@ -14,10 +14,10 @@ class AgentTools(BaseModel):
Tool.from_function( Tool.from_function(
func=self.delegate_work, func=self.delegate_work,
name="Delegate Work to Co-Worker", name="Delegate Work to Co-Worker",
description=dedent(f"""Useful to delegate a specific task to one of the description=dedent(f"""Useful to delegate a specific task to one of the
following co-workers: [{', '.join([agent.role for agent in self.agents])}]. following co-workers: [{', '.join([agent.role for agent in self.agents])}].
The input to this tool should be a pipe (|) separated text of length The input to this tool should be a pipe (|) separated text of length
three, representing the role you want to delegate it to, the task and three, representing the role you want to delegate it to, the task and
information necessary. For example, `coworker|task|information`. information necessary. For example, `coworker|task|information`.
""") """)
), ),
@@ -25,9 +25,9 @@ class AgentTools(BaseModel):
func=self.ask_question, func=self.ask_question,
name="Ask Question to Co-Worker", name="Ask Question to Co-Worker",
description=dedent(f"""Useful to ask a question, opinion or take from on description=dedent(f"""Useful to ask a question, opinion or take from on
of the following co-workers: [{', '.join([agent.role for agent in self.agents])}]. of the following co-workers: [{', '.join([agent.role for agent in self.agents])}].
The input to this tool should be a pipe (|) separated text of length The input to this tool should be a pipe (|) separated text of length
three, representing the role you want to ask it to, the question and three, representing the role you want to ask it to, the question and
information necessary. For example, `coworker|question|information`. information necessary. For example, `coworker|question|information`.
""") """)
), ),
@@ -46,16 +46,16 @@ class AgentTools(BaseModel):
try: try:
agent, task, information = command.split("|") agent, task, information = command.split("|")
except ValueError: except ValueError:
return "Error executing tool. Missing exact 3 pipe (|) separated values." return "Error executing tool. Missing exact 3 pipe (|) separated values. For example, `coworker|task|information`."
if not agent or not task or not information: if not agent or not task or not information:
return "Error executing tool. Missing exact 3 pipe (|) separated values." return "Error executing tool. Missing exact 3 pipe (|) separated values. For example, `coworker|question|information`."
agent = [available_agent for available_agent in self.agents if available_agent.role == agent] agent = [available_agent for available_agent in self.agents if available_agent.role == agent]
if len(agent) == 0: if len(agent) == 0:
return "Error executing tool. Co-worker not found, double check the co-worker." return "Error executing tool. Co-worker not found, double check the co-worker."
agent = agent[0] agent = agent[0]
result = agent.execute_task(task, information) result = agent.execute_task(task, information)
return result return result