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