dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-28 16:25:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

works with both tsl and tsl-pack

Browse source
This commit is contained in:
Paul Gauthier 2025-02-07 16:18:09 -08:00
parent e3d5eaf388
commit 2a56d892d7
1 changed files with 23 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

40
aider/repomap.py
View file

@ -285,26 +285,32 @@ class RepoMap:
captures = query.captures(tree.root_node)
saw = set()
for tag, nodes in captures.items():
for node in nodes:
if tag.startswith("name.definition."):
kind = "def"
elif tag.startswith("name.reference."):
kind = "ref"
else:
continue
if USING_TSL_PACK:
all_nodes = []
for tag, nodes in captures.items():
all_nodes += [(node, tag) for node in nodes]
else:
all_nodes = list(captures)
saw.add(kind)
for node, tag in all_nodes:
if tag.startswith("name.definition."):
kind = "def"
elif tag.startswith("name.reference."):
kind = "ref"
else:
continue
result = Tag(
rel_fname=rel_fname,
fname=fname,
name=node.text.decode("utf-8"),
kind=kind,
line=node.start_point[0],
)
saw.add(kind)
yield result
result = Tag(
rel_fname=rel_fname,
fname=fname,
name=node.text.decode("utf-8"),
kind=kind,
line=node.start_point[0],
)
yield result
if "ref" in saw:
return