andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-01 23:32:39 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

feat: support to initialize a tool from defined Tool attributes

Browse Source
This commit is contained in:
Lucas Gomide
2025-06-17 18:22:48 -03:00
parent db1e9e9b9a
commit b63a8f738c
2 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/crewai/utilities/agent_utils.py
View File

@@ -476,7 +476,7 @@ def load_agent_from_repository(from_repository: str) -> Dict[str, Any]:
try: try:
module = importlib.import_module(tool["module"]) module = importlib.import_module(tool["module"])
tool_class = getattr(module, tool["name"]) tool_class = getattr(module, tool["name"])
attributes[key].append(tool_class()) attributes[key].append(tool_class(**tool["init_params"]))
except Exception as e: except Exception as e:
raise AgentRepositoryError( raise AgentRepositoryError(
f"Tool {tool['name']} could not be loaded: {e}" f"Tool {tool['name']} could not be loaded: {e}"

16
tests/agent_test.py
View File

@@ -2099,7 +2099,7 @@ def mock_get_auth_token():
@patch("crewai.cli.plus_api.PlusAPI.get_agent") @patch("crewai.cli.plus_api.PlusAPI.get_agent")
def test_agent_from_repository(mock_get_agent, mock_get_auth_token): def test_agent_from_repository(mock_get_agent, mock_get_auth_token):
from crewai_tools import SerperDevTool, XMLSearchTool from crewai_tools import SerperDevTool, XMLSearchTool, CSVSearchTool
mock_get_response = MagicMock() mock_get_response = MagicMock()
mock_get_response.status_code = 200 mock_get_response.status_code = 200
@@ -2108,8 +2108,9 @@ def test_agent_from_repository(mock_get_agent, mock_get_auth_token):
"goal": "test goal", "goal": "test goal",
"backstory": "test backstory", "backstory": "test backstory",
"tools": [ "tools": [
{"module": "crewai_tools", "name": "SerperDevTool"}, {"module": "crewai_tools", "name": "SerperDevTool", "init_params": {"n_results": 30}},
{"module": "crewai_tools", "name": "XMLSearchTool"}, {"module": "crewai_tools", "name": "XMLSearchTool", "init_params": {"summarize": True}},
{"module": "crewai_tools", "name": "CSVSearchTool", "init_params": {}},
], ],
} }
mock_get_agent.return_value = mock_get_response mock_get_agent.return_value = mock_get_response
@@ -2118,9 +2119,14 @@ def test_agent_from_repository(mock_get_agent, mock_get_auth_token):
assert agent.role == "test role" assert agent.role == "test role"
assert agent.goal == "test goal" assert agent.goal == "test goal"
assert agent.backstory == "test backstory" assert agent.backstory == "test backstory"
assert len(agent.tools) == 2 assert len(agent.tools) == 3
assert isinstance(agent.tools[0], SerperDevTool) assert isinstance(agent.tools[0], SerperDevTool)
assert agent.tools[0].n_results == 30
assert isinstance(agent.tools[1], XMLSearchTool) assert isinstance(agent.tools[1], XMLSearchTool)
assert agent.tools[1].summarize
assert isinstance(agent.tools[2], CSVSearchTool)
assert not agent.tools[2].summarize
@patch("crewai.cli.plus_api.PlusAPI.get_agent") @patch("crewai.cli.plus_api.PlusAPI.get_agent")
@@ -2133,7 +2139,7 @@ def test_agent_from_repository_override_attributes(mock_get_agent, mock_get_auth
"role": "test role", "role": "test role",
"goal": "test goal", "goal": "test goal",
"backstory": "test backstory", "backstory": "test backstory",
"tools": [{"name": "SerperDevTool", "module": "crewai_tools"}], "tools": [{"name": "SerperDevTool", "module": "crewai_tools", "init_params": {}}],
} }
mock_get_agent.return_value = mock_get_response mock_get_agent.return_value = mock_get_response
agent = Agent(from_repository="test_agent", role="Custom Role") agent = Agent(from_repository="test_agent", role="Custom Role")