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 tool_utils to pass original_tools to ToolUsage

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-04-22 16:49:23 +00:00
parent 6dc8c1ad74
commit c7d126ade0
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/crewai/utilities/tool_utils.py
View File

@@ -60,6 +60,7 @@ def execute_tool_and_check_finality(
task=task, task=task,
agent=agent, agent=agent,
action=agent_action, action=agent_action,
original_tools=tools, # Pass original tools to ensure custom tools work
) )
# Parse tool calling # Parse tool calling

8
tests/tools/test_custom_tool_invocation.py
View File

@@ -57,12 +57,16 @@ def test_custom_tool_invocation():
) )
i18n = I18N() i18n = I18N()
mock_agent.key = "test_agent"
mock_agent.role = "test_role"
result = execute_tool_and_check_finality( result = execute_tool_and_check_finality(
agent_action=action, agent_action=action,
tools=[custom_tool], tools=[custom_tool],
i18n=i18n, i18n=i18n,
agent_key=mock_agent.key if hasattr(mock_agent, "key") else None, agent_key=mock_agent.key,
agent_role=mock_agent.role if hasattr(mock_agent, "role") else None, agent_role=mock_agent.role,
tools_handler=tools_handler, tools_handler=tools_handler,
task=mock_task, task=mock_task,
agent=mock_agent, agent=mock_agent,