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

added aider: prefix to gpt generated edits

Browse source
This commit is contained in:
Paul Gauthier 2023-05-09 14:06:26 -07:00
parent a7490cdb24
commit 0fdd5d101b
2 changed files with 5 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
aider/coder.py
View file

@ -126,7 +126,6 @@ class Coder:
return max(Path(fname).stat().st_mtime for fname in self.fnames) return max(Path(fname).stat().st_mtime for fname in self.fnames)
def get_files_messages(self): def get_files_messages(self):
files_content = prompts.files_content_prefix files_content = prompts.files_content_prefix
files_content += self.get_files_content() files_content += self.get_files_content()
@ -138,8 +137,7 @@ class Coder:
repo_content = prompts.repo_content_prefix repo_content = prompts.repo_content_prefix
repo_content += files_listing repo_content += files_listing
all_content = repo_content + '\n\n' + files_content all_content = repo_content + "\n\n" + files_content
files_messages = [ files_messages = [
dict(role="user", content=all_content), dict(role="user", content=all_content),
@ -201,7 +199,7 @@ class Coder:
messages += self.get_files_messages() messages += self.get_files_messages()
messages += self.cur_messages messages += self.cur_messages
#self.show_messages(messages, "all") # self.show_messages(messages, "all")
content, interrupted = self.send(messages) content, interrupted = self.send(messages)
if interrupted: if interrupted:
@ -228,7 +226,7 @@ class Coder:
if not edited: if not edited:
return True return True
res = self.commit(history=self.cur_messages) res = self.commit(history=self.cur_messages, prefix="aider: ")
if res: if res:
commit_hash, commit_message = res commit_hash, commit_message = res

3
aider/main.py
View file

@ -4,6 +4,7 @@ import argparse
from dotenv import load_dotenv from dotenv import load_dotenv
from aider.coder import Coder from aider.coder import Coder
def main(): def main():
load_dotenv() load_dotenv()
env_prefix = "AIDER_" env_prefix = "AIDER_"
@ -64,7 +65,7 @@ def main():
pretty = args.pretty pretty = args.pretty
coder = Coder(args.model, fnames, pretty, args.history_file, args.show_diffs) coder = Coder(args.model, fnames, pretty, args.history_file, args.show_diffs)
coder.commit(ask=not args.commit_dirty, prefix="WIP: ") coder.commit(ask=not args.commit_dirty, prefix="wip: ")
if args.apply: if args.apply:
with open(args.apply, "r") as f: with open(args.apply, "r") as f: