zach
/
RA.Aid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix tests.

This commit is contained in:
AI Christianson 2024-12-27 19:59:05 -05:00
parent bd2fd07b1b
commit 9944ec9ea4
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/ra_aid/test_tool_configs.py
View File

@ -54,19 +54,19 @@ def test_get_implementation_tools():
def test_get_web_research_tools(): def test_get_web_research_tools():
# Test with expert enabled # Test with expert enabled
tools = get_web_research_tools(expert_enabled=True) tools = get_web_research_tools(expert_enabled=True)
assert len(tools) == 4 assert len(tools) == 5
assert all(callable(tool) for tool in tools) assert all(callable(tool) for tool in tools)
# Get tool names and verify exact matches # Get tool names and verify exact matches
tool_names = [tool.name for tool in tools] tool_names = [tool.name for tool in tools]
expected_names = ['emit_expert_context', 'ask_expert', 'web_search_tavily', 'emit_research_notes'] expected_names = ['emit_expert_context', 'ask_expert', 'web_search_tavily', 'emit_research_notes', 'task_completed']
assert sorted(tool_names) == sorted(expected_names) assert sorted(tool_names) == sorted(expected_names)
# Test without expert enabled # Test without expert enabled
tools_no_expert = get_web_research_tools(expert_enabled=False) tools_no_expert = get_web_research_tools(expert_enabled=False)
assert len(tools_no_expert) == 2 assert len(tools_no_expert) == 3
assert all(callable(tool) for tool in tools_no_expert) assert all(callable(tool) for tool in tools_no_expert)
# Verify exact tool names when expert is disabled # Verify exact tool names when expert is disabled
tool_names_no_expert = [tool.name for tool in tools_no_expert] 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']) assert sorted(tool_names_no_expert) == sorted(['web_search_tavily', 'emit_research_notes', 'task_completed'])