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

refac

Browse source
This commit is contained in:
Paul Gauthier 2024-10-26 07:31:28 -07:00
parent 41f126cad7
commit 5c4b75909e
1 changed files with 25 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

31
aider/io.py
View file

@ -377,13 +377,11 @@ class InputOutput:
):
if changed:
self.changed_files = changed
# ai move all this ...
self.changed_files = process_file_changes(changed)
self.interrupt_input()
break
except Exception as e:
self.tool_error(f"File watcher error: {e}")
raise e
# Start the watcher thread
watcher = threading.Thread(target=watch_files, daemon=True)
@ -447,11 +445,10 @@ class InputOutput:
# Check if we were interrupted by a file change
if self.changed_files:
# ai ...down to here!
changed = " ".join(self.changed_files)
res = process_file_changes(self.changed_files)
self.changed_files = None
return res
return f"/add {changed}" # Return an edit command for the changed file
except EOFError:
return ""
except Exception as err:
@ -770,3 +767,25 @@ def get_rel_fname(fname, root):
return os.path.relpath(fname, root)
except ValueError:
return fname
def process_file_changes(changed):
"""Process file changes and handle special ! comments"""
has_bangs = any(
"!" in comment for comments in changed.values() if comments for comment in comments
)
if not has_bangs:
return "/add " + " ".join(changed.keys())
res = "\n".join(comment for comments in changed.values() if comments for comment in comments)
res = """The "ai" comments below can be found in the code above.
They contain your instructions.
Make the requested changes.
Also remove all these comments from the code.
""" + res
dump(res)
return res