From 78d9d014f25f8f6b5792a747e7beacb2c2f70812 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Fri, 14 Jul 2023 20:00:04 -0700 Subject: [PATCH] fix bug blocking addition of subdir filename mentions --- aider/coders/base_coder.py | 3 +++ tests/test_coder.py | 4 ++-- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py index 716d9f01f..46d325f31 100755 --- a/aider/coders/base_coder.py +++ b/aider/coders/base_coder.py @@ -586,6 +586,9 @@ class Coder: mentioned_rel_fnames = set() fname_to_rel_fnames = {} for rel_fname in addable_rel_fnames: + if rel_fname in words: + mentioned_rel_fnames.add(str(rel_fname)) + fname = os.path.basename(rel_fname) if fname not in fname_to_rel_fnames: fname_to_rel_fnames[fname] = [] diff --git a/tests/test_coder.py b/tests/test_coder.py index 0708ee780..01c90e4a3 100644 --- a/tests/test_coder.py +++ b/tests/test_coder.py @@ -108,7 +108,7 @@ class TestCoder(unittest.TestCase): # Call the check_for_file_mentions method coder.check_for_file_mentions("Please check file1.txt!") - self.assertEqual(coder.abs_fnames, set()) + self.assertEqual(coder.abs_fnames, set([str(Path("file1.txt").resolve())])) def test_check_for_subdir_mention(self): # Mock the IO object @@ -124,7 +124,7 @@ class TestCoder(unittest.TestCase): # Call the check_for_file_mentions method coder.check_for_file_mentions("Please check `other/file1.txt`") - self.assertEqual(coder.abs_fnames, set([Path("other/file1.txt").resolve()])) + self.assertEqual(coder.abs_fnames, set([str(Path("other/file1.txt").resolve())])) def test_get_commit_message(self): # Mock the IO object