diff --git a/src/crewai/agent.py b/src/crewai/agent.py index 02f84f23d..999d1d800 100644 --- a/src/crewai/agent.py +++ b/src/crewai/agent.py @@ -17,6 +17,7 @@ from crewai.memory.contextual.contextual_memory import ContextualMemory from crewai.task import Task from crewai.tools import BaseTool from crewai.tools.agent_tools.agent_tools import AgentTools +from crewai.tools.base_tool import Tool from crewai.utilities import Converter, Prompts from crewai.utilities.constants import TRAINED_AGENTS_DATA_FILE, TRAINING_DATA_FILE from crewai.utilities.converter import generate_model_description @@ -410,7 +411,7 @@ class Agent(BaseAgent): tools = agent_tools.tools() return tools - def get_multimodal_tools(self): + def get_multimodal_tools(self) -> List[Tool]: from crewai.tools.agent_tools.add_image_tool import AddImageTool return [AddImageTool()] diff --git a/src/crewai/crew.py b/src/crewai/crew.py index fe4ccceff..d488783ea 100644 --- a/src/crewai/crew.py +++ b/src/crewai/crew.py @@ -797,7 +797,7 @@ class Crew(BaseModel): return skipped_task_output return None - def _prepare_tools(self, agent: BaseAgent, task: Task, tools: List[Tool]): + def _prepare_tools(self, agent: BaseAgent, task: Task, tools: List[Tool]) -> List[Tool]: # Add delegation tools if agent allows delegation if agent.allow_delegation: if self.process == Process.hierarchical: