mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-08 15:48:29 +00:00
7acf0b2107
* Exploring output being passed to tool selector to see if we can better format data * WIP. Adding JSON repair functionality * Almost done implementing JSON repair. Testing fixes vs current base case. * More action cleanup with additional tests * WIP. Trying to figure out what is going on with tool descriptions * Update tool description generation * WIP. Trying to find out what is causing the tools to duplicate * Replacing tools properly instead of duplicating them accidentally * Fixing issues for MR * Update dependencies for JSON_REPAIR * More cleaning up pull request * preppering for call * Fix type-checking issues --------- Co-authored-by: João Moura <joaomdmoura@gmail.com>
87 lines
3.5 KiB
Python
87 lines
3.5 KiB
Python
from abc import ABC, abstractmethod
|
|
from typing import List, Optional, Union
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from crewai.agents.agent_builder.base_agent import BaseAgent
|
|
from crewai.task import Task
|
|
from crewai.utilities import I18N
|
|
|
|
|
|
class BaseAgentTools(BaseModel, ABC):
|
|
"""Default tools around agent delegation"""
|
|
|
|
agents: List[BaseAgent] = Field(description="List of agents in this crew.")
|
|
i18n: I18N = Field(default=I18N(), description="Internationalization settings.")
|
|
|
|
@abstractmethod
|
|
def tools(self):
|
|
pass
|
|
|
|
def _get_coworker(self, coworker: Optional[str], **kwargs) -> Optional[str]:
|
|
coworker = coworker or kwargs.get("co_worker") or kwargs.get("coworker")
|
|
if coworker:
|
|
is_list = coworker.startswith("[") and coworker.endswith("]")
|
|
if is_list:
|
|
coworker = coworker[1:-1].split(",")[0]
|
|
|
|
return coworker
|
|
|
|
def delegate_work(
|
|
self, task: str, context: str, coworker: Optional[str] = None, **kwargs
|
|
):
|
|
"""Useful to delegate a specific task to a coworker passing all necessary context and names."""
|
|
coworker = self._get_coworker(coworker, **kwargs)
|
|
return self._execute(coworker, task, context)
|
|
|
|
def ask_question(
|
|
self, question: str, context: str, coworker: Optional[str] = None, **kwargs
|
|
):
|
|
"""Useful to ask a question, opinion or take from a coworker passing all necessary context and names."""
|
|
coworker = self._get_coworker(coworker, **kwargs)
|
|
return self._execute(coworker, question, context)
|
|
|
|
def _execute(
|
|
self, agent_name: Union[str, None], task: str, context: Union[str, None]
|
|
):
|
|
"""Execute the command."""
|
|
try:
|
|
if agent_name is None:
|
|
agent_name = ""
|
|
|
|
# It is important to remove the quotes from the agent name.
|
|
# The reason we have to do this is because less-powerful LLM's
|
|
# have difficulty producing valid JSON.
|
|
# As a result, we end up with invalid JSON that is truncated like this:
|
|
# {"task": "....", "coworker": "....
|
|
# when it should look like this:
|
|
# {"task": "....", "coworker": "...."}
|
|
agent_name = agent_name.casefold().replace('"', "").replace("\n", "")
|
|
|
|
agent = [ # type: ignore # Incompatible types in assignment (expression has type "list[BaseAgent]", variable has type "str | None")
|
|
available_agent
|
|
for available_agent in self.agents
|
|
if available_agent.role.casefold().replace("\n", "") == agent_name
|
|
]
|
|
except Exception as _:
|
|
return self.i18n.errors("agent_tool_unexsiting_coworker").format(
|
|
coworkers="\n".join(
|
|
[f"- {agent.role.casefold()}" for agent in self.agents]
|
|
)
|
|
)
|
|
|
|
if not agent:
|
|
return self.i18n.errors("agent_tool_unexsiting_coworker").format(
|
|
coworkers="\n".join(
|
|
[f"- {agent.role.casefold()}" for agent in self.agents]
|
|
)
|
|
)
|
|
|
|
agent = agent[0]
|
|
task_with_assigned_agent = Task( # type: ignore # Incompatible types in assignment (expression has type "Task", variable has type "str")
|
|
description=task,
|
|
agent=agent,
|
|
expected_output="Your best answer to your coworker asking you this, accounting for the context shared.",
|
|
)
|
|
return agent.execute_task(task_with_assigned_agent, context)
|