115 lines
3.7 KiB
Python
115 lines
3.7 KiB
Python
from ra_aid.tool_configs import (
|
|
MODIFICATION_TOOLS,
|
|
get_implementation_tools,
|
|
get_planning_tools,
|
|
get_read_only_tools,
|
|
get_research_tools,
|
|
get_web_research_tools,
|
|
set_modification_tools,
|
|
)
|
|
|
|
|
|
def test_get_read_only_tools():
|
|
# Test without human interaction, without aider
|
|
tools = get_read_only_tools(human_interaction=False, use_aider=False)
|
|
assert len(tools) > 0
|
|
assert all(callable(tool) for tool in tools)
|
|
|
|
# Check emit_related_files is not included when use_aider is False
|
|
tool_names = [tool.name for tool in tools]
|
|
assert "emit_related_files" not in tool_names
|
|
|
|
# Test with use_aider=True
|
|
tools_with_aider = get_read_only_tools(human_interaction=False, use_aider=True)
|
|
tool_names_with_aider = [tool.name for tool in tools_with_aider]
|
|
assert "emit_related_files" in tool_names_with_aider
|
|
|
|
# Test with human interaction
|
|
tools_with_human = get_read_only_tools(human_interaction=True, use_aider=False)
|
|
assert len(tools_with_human) == len(tools) + 1
|
|
|
|
|
|
def test_get_research_tools():
|
|
# Test basic research tools
|
|
tools = get_research_tools()
|
|
assert len(tools) > 0
|
|
assert all(callable(tool) for tool in tools)
|
|
|
|
# Test without expert
|
|
tools_no_expert = get_research_tools(expert_enabled=False)
|
|
assert len(tools_no_expert) < len(tools)
|
|
|
|
# Test research-only mode
|
|
tools_research_only = get_research_tools(research_only=True)
|
|
assert len(tools_research_only) < len(tools)
|
|
|
|
|
|
def test_get_planning_tools():
|
|
# Test with expert enabled
|
|
tools = get_planning_tools(expert_enabled=True)
|
|
assert len(tools) > 0
|
|
assert all(callable(tool) for tool in tools)
|
|
|
|
# Test without expert
|
|
tools_no_expert = get_planning_tools(expert_enabled=False)
|
|
assert len(tools_no_expert) < len(tools)
|
|
|
|
|
|
def test_get_implementation_tools():
|
|
# Test with expert enabled
|
|
tools = get_implementation_tools(expert_enabled=True)
|
|
assert len(tools) > 0
|
|
assert all(callable(tool) for tool in tools)
|
|
|
|
# Test without expert
|
|
tools_no_expert = get_implementation_tools(expert_enabled=False)
|
|
assert len(tools_no_expert) < len(tools)
|
|
|
|
|
|
def test_get_web_research_tools():
|
|
# Test with expert enabled
|
|
tools = get_web_research_tools(expert_enabled=True)
|
|
assert len(tools) == 5
|
|
assert all(callable(tool) for tool in tools)
|
|
|
|
# Get tool names and verify exact matches
|
|
tool_names = [tool.name for tool in tools]
|
|
expected_names = [
|
|
"emit_expert_context",
|
|
"ask_expert",
|
|
"web_search_tavily",
|
|
"emit_research_notes",
|
|
"task_completed",
|
|
]
|
|
assert sorted(tool_names) == sorted(expected_names)
|
|
|
|
# Test without expert enabled
|
|
tools_no_expert = get_web_research_tools(expert_enabled=False)
|
|
assert len(tools_no_expert) == 3
|
|
assert all(callable(tool) for tool in tools_no_expert)
|
|
|
|
# Verify exact tool names when expert is disabled
|
|
tool_names_no_expert = [tool.name for tool in tools_no_expert]
|
|
assert sorted(tool_names_no_expert) == sorted(
|
|
["web_search_tavily", "emit_research_notes", "task_completed"]
|
|
)
|
|
|
|
|
|
def test_set_modification_tools():
|
|
# Test with use_aider=False (default setting)
|
|
set_modification_tools(use_aider=False)
|
|
tool_names = [tool.name for tool in MODIFICATION_TOOLS]
|
|
assert "file_str_replace" in tool_names
|
|
assert "put_complete_file_contents" in tool_names
|
|
assert "run_programming_task" not in tool_names
|
|
|
|
# Test with use_aider=True
|
|
set_modification_tools(use_aider=True)
|
|
tool_names = [tool.name for tool in MODIFICATION_TOOLS]
|
|
assert "file_str_replace" not in tool_names
|
|
assert "put_complete_file_contents" not in tool_names
|
|
assert "run_programming_task" in tool_names
|
|
|
|
# Reset to default for other tests
|
|
set_modification_tools(use_aider=False)
|