mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-07 15:18:29 +00:00
82647358b2
* initial fix on delegation tools
* fixing tests for delegations and coding
* Refactor prepare tool and adding initial add images logic
* supporting image tool
* fixing linter
* fix linter
* Making sure multimodal feature support i18n
* fix linter and types
* mixxing translations
* fix types and linter
* Revert "fixing linter"
This reverts commit 2eda5fdeed.
* fix linters
* test
* fix
* fix
* fix linter
* fix
* ignore
* type improvements
33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
from crewai.agents.agent_builder.base_agent import BaseAgent
|
|
from crewai.tools.base_tool import BaseTool
|
|
from crewai.utilities import I18N
|
|
|
|
from .ask_question_tool import AskQuestionTool
|
|
from .delegate_work_tool import DelegateWorkTool
|
|
|
|
|
|
class AgentTools:
|
|
"""Manager class for agent-related tools"""
|
|
|
|
def __init__(self, agents: list[BaseAgent], i18n: I18N = I18N()):
|
|
self.agents = agents
|
|
self.i18n = i18n
|
|
|
|
def tools(self) -> list[BaseTool]:
|
|
"""Get all available agent tools"""
|
|
coworkers = ", ".join([f"{agent.role}" for agent in self.agents])
|
|
|
|
delegate_tool = DelegateWorkTool(
|
|
agents=self.agents,
|
|
i18n=self.i18n,
|
|
description=self.i18n.tools("delegate_work").format(coworkers=coworkers), # type: ignore
|
|
)
|
|
|
|
ask_tool = AskQuestionTool(
|
|
agents=self.agents,
|
|
i18n=self.i18n,
|
|
description=self.i18n.tools("ask_question").format(coworkers=coworkers), # type: ignore
|
|
)
|
|
|
|
return [delegate_tool, ask_tool]
|