From b228ba3dc5363deabd741458b025cb28f3ae2d45 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Fri, 30 Aug 2024 15:06:52 -0700 Subject: [PATCH] feat: add special file filtering to RepoMap ranking --- aider/repomap.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/aider/repomap.py b/aider/repomap.py index 882b55c1f..f8d98084f 100644 --- a/aider/repomap.py +++ b/aider/repomap.py @@ -17,6 +17,7 @@ from pygments.token import Token from tqdm import tqdm from aider.dump import dump +from aider.special import filter_important_files from aider.utils import Spinner # tree_sitter is throwing a FutureWarning @@ -501,6 +502,12 @@ class RepoMap: progress=spin.step, ) + other_rel_fnames = sorted(set(self.get_rel_fname(fname) for fname in other_fnames)) + special_fnames = filter_important_files(other_rel_fnames) + special_fnames = [(fn,) for fn in special_fnames] + + ranked_tags = special_fnames + ranked_tags + spin.step() num_tags = len(ranked_tags)