mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 08:08:32 +00:00
7954f6b51c
* fixing identation for AgentTools * updating gitignore to exclude quick test script * startingprompt translation * supporting individual task output * adding agent to task output * cutting new version * Updating README example
77 lines
3.0 KiB
Python
77 lines
3.0 KiB
Python
from textwrap import dedent
|
|
from typing import List
|
|
|
|
from langchain.tools import Tool
|
|
from pydantic import BaseModel, Field
|
|
|
|
from crewai.agent import Agent
|
|
|
|
|
|
class AgentTools(BaseModel):
|
|
"""Default tools around agent delegation"""
|
|
|
|
agents: List[Agent] = Field(description="List of agents in this crew.")
|
|
|
|
def tools(self):
|
|
return [
|
|
Tool.from_function(
|
|
func=self.delegate_work,
|
|
name="Delegate work to co-worker",
|
|
description=dedent(
|
|
f"""\
|
|
Useful to delegate a specific task to one 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
|
|
three, representing the co-worker you want to ask it to (one of the options),
|
|
the task and all actual context you have for the task.
|
|
For example, `coworker|task|context`.
|
|
"""
|
|
),
|
|
),
|
|
Tool.from_function(
|
|
func=self.ask_question,
|
|
name="Ask question to co-worker",
|
|
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])}].
|
|
The input to this tool should be a pipe (|) separated text of length
|
|
three, representing the co-worker you want to ask it to (one of the options),
|
|
the question and all actual context you have for the question.
|
|
For example, `coworker|question|context`.
|
|
"""
|
|
),
|
|
),
|
|
]
|
|
|
|
def delegate_work(self, command):
|
|
"""Useful to delegate a specific task to a coworker."""
|
|
return self.__execute(command)
|
|
|
|
def ask_question(self, command):
|
|
"""Useful to ask a question, opinion or take from a coworker."""
|
|
return self.__execute(command)
|
|
|
|
def __execute(self, command):
|
|
"""Execute the command."""
|
|
try:
|
|
agent, task, context = command.split("|")
|
|
except ValueError:
|
|
return "\nError executing tool. Missing exact 3 pipe (|) separated values. For example, `coworker|task|context`. I need to make sure to pass context as context\n"
|
|
|
|
if not agent or not task or not context:
|
|
return "\nError executing tool. Missing exact 3 pipe (|) separated values. For example, `coworker|task|context`. I need to make sure to pass context as context.\n"
|
|
|
|
agent = [
|
|
available_agent
|
|
for available_agent in self.agents
|
|
if available_agent.role == agent
|
|
]
|
|
|
|
if len(agent) == 0:
|
|
return f"\nError executing tool. Co-worker mentioned on the Action Input not found, it must to be one of the following options: {', '.join([agent.role for agent in self.agents])}.\n"
|
|
|
|
agent = agent[0]
|
|
result = agent.execute_task(task, context)
|
|
return result
|