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

feat: add /save command to store chat session file list

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-10-29 12:43:51 -07:00
parent 79cdff6163
commit dbefa6b010
1 changed files with 22 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

23
aider/commands.py
View file

@ -1247,7 +1247,6 @@ class Commands:
self.io.tool_output(output)
#ai add cmd_save(fname) that generates a file that cmd_load can read!
#it should use /add and /read to reconstruct the list of added files, one per line
def cmd_load(self, args):
"Load and execute commands from a file"
if not args.strip():
@ -1272,6 +1271,28 @@ class Commands:
self.io.tool_output(f"\nExecuting command: {cmd}")
self.run(cmd)
def cmd_save(self, args):
"Save commands to a file that can reconstruct the current chat session's files"
if not args.strip():
self.io.tool_error("Please provide a filename to save the commands to.")
return
try:
with open(args.strip(), "w", encoding=self.io.encoding) as f:
# Write commands to add editable files
for fname in sorted(self.coder.abs_fnames):
rel_fname = self.coder.get_rel_fname(fname)
f.write(f"/add {rel_fname}\n")
# Write commands to add read-only files
for fname in sorted(self.coder.abs_read_only_fnames):
rel_fname = self.coder.get_rel_fname(fname)
f.write(f"/read-only {rel_fname}\n")
self.io.tool_output(f"Saved commands to {args.strip()}")
except Exception as e:
self.io.tool_error(f"Error saving commands to file: {e}")
def cmd_copy(self, args):
"Copy the last assistant message to the clipboard"
all_messages = self.coder.done_messages + self.coder.cur_messages