Fix issue #2664: Custom tools not being called by the agent

Co-Authored-By: Joe Moura <joao@crewai.com>

tests/tools/test_custom_tool_invocation.py

@@ -0,0 +1,60 @@
+import pytest
+from unittest.mock import MagicMock
+
+from crewai import Agent, Task
+from crewai.agents.crew_agent_executor import CrewAgentExecutor
+from crewai.tools import BaseTool
+from crewai.agents.tools_handler import ToolsHandler
+from crewai.agents.parser import AgentAction
+from pydantic import BaseModel, Field
+
+
+class TestToolInput(BaseModel):
+    test_param: str = Field(..., description="A test parameter")
+
+
+class TestCustomTool(BaseTool):
+    name: str = "Test Custom Tool"
+    description: str = "A test tool to verify custom tool invocation"
+    args_schema: type[BaseModel] = TestToolInput
+
+    def _run(self, test_param: str) -> str:
+        return f"Tool executed with param: {test_param}"
+
+
+def test_custom_tool_invocation():
+    custom_tool = TestCustomTool()
+    
+    mock_agent = MagicMock()
+    mock_task = MagicMock()
+    mock_llm = MagicMock()
+    mock_crew = MagicMock()
+    tools_handler = ToolsHandler()
+    
+    executor = CrewAgentExecutor(
+        llm=mock_llm,
+        task=mock_task,
+        crew=mock_crew,
+        agent=mock_agent,
+        prompt={},
+        max_iter=5,
+        tools=[custom_tool],
+        tools_names="Test Custom Tool",
+        stop_words=[],
+        tools_description="A test tool to verify custom tool invocation",
+        tools_handler=tools_handler,
+        original_tools=[custom_tool]
+    )
+    
+    action = AgentAction(
+        tool="Test Custom Tool",
+        tool_input={"test_param": "test_value"},
+        thought="I'll use the custom tool",
+        text="I'll use the Test Custom Tool to get a result",
+        message_log=[]
+    )
+    
+    result = executor._execute_tool_and_check_finality(action)
+    
+    assert "Tool executed with param: test_value" in result.result
+    assert result.result_as_answer is False