From 9c76745bcc32e19fc49472b0522951a9a7d7978d Mon Sep 17 00:00:00 2001 From: AI Christianson Date: Tue, 25 Feb 2025 18:34:28 -0500 Subject: [PATCH] improve binary file filtering --- ra_aid/tools/memory.py | 5 +++-- tests/ra_aid/tools/test_memory.py | 4 +++- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/ra_aid/tools/memory.py b/ra_aid/tools/memory.py index 43a6058..278b435 100644 --- a/ra_aid/tools/memory.py +++ b/ra_aid/tools/memory.py @@ -39,7 +39,7 @@ _global_memory: Dict[ bool, str, List[str], - List[WorkLogEntry] + List[WorkLogEntry], ], ] = { "research_notes": [], @@ -462,7 +462,8 @@ def emit_related_files(files: List[str]) -> str: # Return summary message if binary_files: - return f"Files noted. {len(binary_files)} binary files were skipped." + binary_files_list = ", ".join(f"'{file}'" for file in binary_files) + return f"Files noted. Binary files skipped: {binary_files_list}" else: return "Files noted." diff --git a/tests/ra_aid/tools/test_memory.py b/tests/ra_aid/tools/test_memory.py index af782c7..551a39b 100644 --- a/tests/ra_aid/tools/test_memory.py +++ b/tests/ra_aid/tools/test_memory.py @@ -750,7 +750,9 @@ def test_emit_related_files_binary_filtering(reset_memory, tmp_path, monkeypatch # Verify the result message mentions skipped binary files assert "Files noted." in result - assert "2 binary files were skipped" in result + assert "Binary files skipped:" in result + assert f"'{binary_file1}'" in result + assert f"'{binary_file2}'" in result # Verify only text files were added to related_files assert len(_global_memory["related_files"]) == 2