24 lines
696 B
Python
24 lines
696 B
Python
import pytest
|
|
from ai_client import set_agent_tools, _build_anthropic_tools
|
|
|
|
def test_agent_tools_wiring():
|
|
# Only enable read_file and run_powershell
|
|
agent_tools = {
|
|
"run_powershell": True,
|
|
"read_file": True,
|
|
"list_directory": False,
|
|
"search_files": False,
|
|
"get_file_summary": False,
|
|
"web_search": False,
|
|
"fetch_url": False
|
|
}
|
|
set_agent_tools(agent_tools)
|
|
|
|
anth_tools = _build_anthropic_tools()
|
|
tool_names = [t["name"] for t in anth_tools]
|
|
|
|
assert "read_file" in tool_names
|
|
assert "run_powershell" in tool_names
|
|
assert "list_directory" not in tool_names
|
|
assert "web_search" not in tool_names
|