dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-31 17:55:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

feat: add error handling for TreeContext with fallback to raw comments

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-12-06 12:15:46 -08:00
parent a96d87814b
commit fd301c519d
1 changed files with 21 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

25
aider/watch.py
View file

@ -195,10 +195,27 @@ Be sure to remove all these "AI" comments from the code!
# Convert comment line numbers to line indices (0-based)
lois = [ln - 1 for ln, _ in zip(line_nums, comments) if ln > 0]
# Handle ValueError from TreeContext and just include the actual comments instead. ai!
context = TreeContext(
rel_fname,
code,
try:
context = TreeContext(
rel_fname,
code,
color=False,
line_number=False,
child_context=False,
last_line=False,
margin=0,
mark_lois=True,
loi_pad=3,
show_top_of_file_parent_scope=False,
)
context.lines_of_interest = set()
context.add_lines_of_interest(lois)
context.add_context()
res += context.format()
except ValueError:
# Fall back to just showing the comments if TreeContext fails
for ln, comment in zip(line_nums, comments):
res += f" Line {ln}: {comment}\n"
color=False,
line_number=False,
child_context=False,