From f6a2ec15d7882e6c3ad85973d9f61a3fca1826b3 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Fri, 7 Feb 2025 15:23:48 -0800 Subject: [PATCH] refactor: Update get_scm_fname to support multiple query subdirectories --- aider/repomap.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/aider/repomap.py b/aider/repomap.py index 2a202253a..b184b6566 100644 --- a/aider/repomap.py +++ b/aider/repomap.py @@ -733,8 +733,19 @@ def get_random_color(): def get_scm_fname(lang): # Load the tags queries + if USING_TSL_PACK: + subdir = "tree-sitter-language-pack" + # check this subdir first, if file exists, return it otherwise check the other. ai! + else: + subdir = "tree-sitter-languages" + + try: - return resources.files(__package__).joinpath("queries", f"tree-sitter-{lang}-tags.scm") + return resources.files(__package__).joinpath( + "queries", + subdir, + f"tree-sitter-{lang}-tags.scm", + ) except KeyError: return