From 07c5c7f47710334d24e764be3894e06bceef42f6 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Wed, 7 Jun 2023 14:57:57 -0700 Subject: [PATCH] Update debugging main in repomap --- aider/repomap.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/aider/repomap.py b/aider/repomap.py index ef7e72975..79bb37004 100644 --- a/aider/repomap.py +++ b/aider/repomap.py @@ -384,16 +384,18 @@ def get_random_color(): if __name__ == "__main__": fnames = sys.argv[1:] - fnames = [] + chat_fnames = [] + other_fnames = [] for dname in sys.argv[1:]: - fnames += find_py_files(dname) + if ".venv" in dname: + other_fnames += find_py_files(dname) + else: + chat_fnames += find_py_files(dname) - fnames = sorted(fnames) - - root = os.path.commonpath(fnames) + root = os.path.commonpath(chat_fnames) rm = RepoMap(root=root) - repo_map = rm.get_ranked_tags_map(fnames) + repo_map = rm.get_ranked_tags_map(chat_fnames, other_fnames) dump(len(repo_map)) print(repo_map)