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

fix tests; fix off-by-one bug in output of repomap

Browse source
This commit is contained in:
Paul Gauthier 2023-10-18 15:42:38 -07:00
parent 10b856e0af
commit c0375d3328
2 changed files with 14 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

13
aider/repomap.py
View file

@ -28,12 +28,17 @@ def to_tree(tags):
cur_fname = None
context = None
output = ""
for tag in tags:
if type(tag) is tuple:
# add a bogus tag at the end so we trip the this_fname != cur_fname...
for tag in tags + [None]:
if tag is None:
this_fname = None
elif type(tag) is tuple:
this_fname = tag[0]
else:
this_fname = tag.rel_fname
# ... here ... to output the final real entry in the list
if this_fname != cur_fname:
if context:
context.add_context()
@ -45,7 +50,7 @@ def to_tree(tags):
elif cur_fname:
output += cur_fname + "\n"
if type(tag) is not tuple:
if type(tag) is Tag:
context = TreeContext(
tag.rel_fname,
Path(tag.fname).read_text(), # TODO: encoding
@ -120,7 +125,7 @@ class RepoMap:
num_tokens = self.token_count(files_listing)
if self.verbose:
self.io.tool_output(f"ctags map: {num_tokens/1024:.1f} k-tokens")
self.io.tool_output(f"ast map: {num_tokens/1024:.1f} k-tokens")
if chat_files:
other = "other "