dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-30 01:04:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

fix sorting and whitespace in map

Browse source
This commit is contained in:
Paul Gauthier 2023-10-19 08:15:28 -07:00
parent 0291a99257
commit cba883d4cd
1 changed files with 4 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

11
aider/repomap.py
View file

@ -16,7 +16,7 @@ from aider import models
from .dump import dump # noqa: F402 from .dump import dump # noqa: F402
Tag = namedtuple("Tag", "fname rel_fname line name kind".split()) Tag = namedtuple("Tag", "rel_fname fname line name kind".split())
def to_tree(tags): def to_tree(tags):
@ -32,10 +32,7 @@ def to_tree(tags):
# add a bogus tag at the end so we trip the this_fname != cur_fname... # add a bogus tag at the end so we trip the this_fname != cur_fname...
dummy_tag = (None,) dummy_tag = (None,)
for tag in tags + [dummy_tag]: for tag in tags + [dummy_tag]:
if type(tag) is Tag: this_fname = tag[0]
this_fname = tag.rel_fname
else:
this_fname = tag[0]
# ... here ... to output the final real entry in the list # ... here ... to output the final real entry in the list
if this_fname != cur_fname: if this_fname != cur_fname:
@ -46,7 +43,7 @@ def to_tree(tags):
output += context.format() output += context.format()
context = None context = None
elif cur_fname: elif cur_fname:
output += cur_fname + "\n" output += "\n" + cur_fname + "\n"
if type(tag) is Tag: if type(tag) is Tag:
context = TreeContext( context = TreeContext(
@ -190,8 +187,8 @@ class RepoMap:
continue continue
result = Tag( result = Tag(
fname=fname,
rel_fname=rel_fname, rel_fname=rel_fname,
fname=fname,
name=node.text.decode("utf-8"), name=node.text.decode("utf-8"),
kind=kind, kind=kind,
line=node.start_point[0], line=node.start_point[0],