diff --git a/ra_aid/prompts.py b/ra_aid/prompts.py index 0bcca9d..a51dac2 100644 --- a/ra_aid/prompts.py +++ b/ra_aid/prompts.py @@ -458,7 +458,7 @@ Exit Criteria: Context Cleanup: - Use delete_key_facts to remove any key facts that no longer apply. - Use delete_key_snippets to remove any key snippets that no longer apply. - - Use delete_research_notes to remove any research notes that no longer apply. + - Use deregister_related_files to remove any related files that no longer apply. Remember: - Always begin by calling ask_human. @@ -467,4 +467,4 @@ Remember: - Do communicate results/responses from tools that you call as it pertains to the users request. - If the user interrupts/cancels an operation, you may want to ask why. - For deep debugging, logic analysis, or correctness checks, rely on the expert (if expert is available) for guidance. -""" \ No newline at end of file +""" diff --git a/ra_aid/tool_configs.py b/ra_aid/tool_configs.py index 3e92e41..cac8f6f 100644 --- a/ra_aid/tool_configs.py +++ b/ra_aid/tool_configs.py @@ -3,7 +3,7 @@ from ra_aid.tools import ( ask_expert, ask_human, run_shell_command, run_programming_task, emit_research_notes, emit_plan, emit_related_files, emit_task, emit_expert_context, emit_key_facts, delete_key_facts, - emit_key_snippets, delete_key_snippets, delete_related_files, delete_tasks, read_file_tool, + emit_key_snippets, delete_key_snippets, deregister_related_files, delete_tasks, read_file_tool, fuzzy_find_project_files, ripgrep_search, list_directory_tree, swap_task_order, monorepo_detected, existing_project_detected, ui_detected, task_completed, plan_implementation_completed @@ -20,7 +20,7 @@ def get_read_only_tools(human_interaction: bool = False) -> list: delete_key_facts, emit_key_snippets, delete_key_snippets, - delete_related_files, + deregister_related_files, list_directory_tree, read_file_tool, fuzzy_find_project_files, @@ -118,7 +118,7 @@ def get_chat_tools(expert_enabled: bool = True) -> list: request_research_and_implementation, delete_key_facts, delete_key_snippets, - delete_related_files + deregister_related_files ] return tools diff --git a/ra_aid/tools/__init__.py b/ra_aid/tools/__init__.py index 0c1e5a0..832dee0 100644 --- a/ra_aid/tools/__init__.py +++ b/ra_aid/tools/__init__.py @@ -13,14 +13,14 @@ from .memory import ( delete_tasks, emit_research_notes, emit_plan, emit_task, get_memory_value, emit_key_facts, request_implementation, delete_key_facts, emit_key_snippets, delete_key_snippets, emit_related_files, swap_task_order, task_completed, - plan_implementation_completed, delete_related_files + plan_implementation_completed, deregister_related_files ) __all__ = [ 'ask_expert', 'delete_key_facts', 'delete_key_snippets', - 'delete_related_files', + 'deregister_related_files', 'emit_expert_context', 'emit_key_facts', 'emit_key_snippets', diff --git a/ra_aid/tools/memory.py b/ra_aid/tools/memory.py index bc4afd6..8a6e7f0 100644 --- a/ra_aid/tools/memory.py +++ b/ra_aid/tools/memory.py @@ -384,8 +384,8 @@ def emit_related_files(files: List[str]) -> str: return '\n'.join(results) -@tool("delete_related_files") -def delete_related_files(file_ids: List[int]) -> str: +@tool("deregister_related_files") +def deregister_related_files(file_ids: List[int]) -> str: """Delete multiple related files from global memory by their IDs. Silently skips any IDs that don't exist. diff --git a/tests/ra_aid/tools/test_memory.py b/tests/ra_aid/tools/test_memory.py index e69d5ed..2af2d99 100644 --- a/tests/ra_aid/tools/test_memory.py +++ b/tests/ra_aid/tools/test_memory.py @@ -8,7 +8,7 @@ from ra_aid.tools.memory import ( delete_key_snippets, emit_related_files, get_related_files, - delete_related_files, + deregister_related_files, emit_task, delete_tasks, swap_task_order @@ -269,19 +269,19 @@ def test_related_files_id_tracking(reset_memory): assert _global_memory['related_files'][0] == "file1.py" assert _global_memory['related_files'][1] == "file2.py" -def test_delete_related_files(reset_memory): +def test_deregister_related_files(reset_memory): """Test deleting related files""" # Add test files emit_related_files.invoke({"files": ["file1.py", "file2.py", "file3.py"]}) # Delete middle file - result = delete_related_files.invoke({"file_ids": [1]}) + result = deregister_related_files.invoke({"file_ids": [1]}) assert result == "File references removed." assert 1 not in _global_memory['related_files'] assert len(_global_memory['related_files']) == 2 # Delete multiple files including non-existent ID - result = delete_related_files.invoke({"file_ids": [0, 2, 999]}) + result = deregister_related_files.invoke({"file_ids": [0, 2, 999]}) assert result == "File references removed." assert len(_global_memory['related_files']) == 0