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

fix: Fix code execution and multimodal tool handling in Agent class

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

40
src/crewai/agent.py
View File

@@ -338,21 +338,33 @@ class Agent(BaseAgent):
return tools return tools
def get_multimodal_tools(self) -> Sequence[BaseTool]: def get_multimodal_tools(self) -> Sequence[BaseTool]:
from crewai.tools.agent_tools.add_image_tool import AddImageTool """Get multimodal tools for the agent.
Returns:
List[BaseTool]: List containing the AddImageTool if agent is multimodal
"""
if self.multimodal:
from crewai.tools.agent_tools.add_image_tool import AddImageTool
return [AddImageTool()]
return []
return [AddImageTool()] def get_code_execution_tools(self) -> Sequence[BaseTool]:
"""Get code execution tools for the agent.
def get_code_execution_tools(self):
try: Returns:
from crewai_tools import CodeInterpreterTool # type: ignore List[BaseTool]: List containing the CodeInterpreterTool if code execution is allowed
"""
# Set the unsafe_mode based on the code_execution_mode attribute if self.allow_code_execution:
unsafe_mode = self.code_execution_mode == "unsafe" try:
return [CodeInterpreterTool(unsafe_mode=unsafe_mode)] from crewai_tools import CodeInterpreterTool # type: ignore
except ModuleNotFoundError: # Set the unsafe_mode based on the code_execution_mode attribute
self._logger.log( unsafe_mode = self.code_execution_mode == "unsafe"
"info", "Coding tools not available. Install crewai_tools. " return [CodeInterpreterTool(unsafe_mode=unsafe_mode)]
) except ModuleNotFoundError:
self._logger.log(
"warning", "Coding tools not available. Install crewai_tools.", color="yellow"
)
return []
def get_output_converter(self, llm, text, model, instructions): def get_output_converter(self, llm, text, model, instructions):
return Converter(llm=llm, text=text, model=model, instructions=instructions) return Converter(llm=llm, text=text, model=model, instructions=instructions)