22 lines
703 B
Diff
22 lines
703 B
Diff
|
diff --git a/aider/repomap.py b/aider/repomap.py
|
||
|
|
index 23eee239..0a40f2e6 100644
|
||
|
|
--- a/aider/repomap.py
|
||
|
|
+++ b/aider/repomap.py
|
||
|
|
@@ -16,6 +16,7 @@ from grep_ast import TreeContext, filename_to_lang
|
||
|
|
from pygments.lexers import guess_lexer_for_filename
|
||
|
|
from pygments.token import Token
|
||
|
|
from tqdm import tqdm
|
||
|
|
+from tree_sitter import QueryCursor
|
||
|
|
|
||
|
|
from aider.dump import dump
|
||
|
|
from aider.special import filter_important_files
|
||
|
|
@@ -286,7 +287,7 @@ class RepoMap:
|
||
|
|
|
||
|
|
# Run the tags queries
|
||
|
|
query = language.query(query_scm)
|
||
|
|
- captures = query.captures(tree.root_node)
|
||
|
|
+ captures = QueryCursor(query).captures(tree.root_node)
|
||
|
|
|
||
|
|
saw = set()
|
||
|
|
if USING_TSL_PACK:
|