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

fix: Fix multimodal and code execution tool handling

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-14 21:20:46 +00:00
parent 4832b35779
commit c6153cc1b4
1 changed files with 36 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
src/crewai/crew.py
View File

@@ -1014,13 +1014,45 @@ class Crew(BaseModel):
return delegation_tools
return self._merge_tools(tools, delegation_tools)
def _add_multimodal_tools(self, agent: BaseAgent, tools: List[Tool]):
def _add_multimodal_tools(self, agent: BaseAgent, tools: List[Tool]) -> List[Tool]:
"""Add multimodal tools for an agent.
Args:
agent: The agent that will use the tools
tools: The current set of tools
Returns:
List[Tool]: The tools with multimodal tools added
"""
multimodal_tools = agent.get_multimodal_tools()
return self._merge_tools(tools, multimodal_tools)
if multimodal_tools:
self._logger.log(
"debug",
f"Adding multimodal tools for agent {agent.role}: {[tool.name for tool in multimodal_tools]}",
color="blue",
)
return multimodal_tools
return []
def _add_code_execution_tools(self, agent: BaseAgent, tools: List[Tool]):
def _add_code_execution_tools(self, agent: BaseAgent, tools: List[Tool]) -> List[Tool]:
"""Add code execution tools for an agent.
Args:
agent: The agent that will use the tools
tools: The current set of tools
Returns:
List[Tool]: The tools with code execution tools added
"""
code_tools = agent.get_code_execution_tools()
return self._merge_tools(tools, code_tools)
if code_tools:
self._logger.log(
"debug",
f"Adding code execution tools for agent {agent.role}: {[tool.name for tool in code_tools]}",
color="blue",
)
return code_tools
return []
def _add_delegation_tools(self, task: Task, tools: List[Tool]) -> List[Tool]:
"""Add delegation tools for the task's agent.