dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-01 18:25:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

worked

Browse source
This commit is contained in:
Paul Gauthier 2023-04-03 21:41:46 -07:00
parent 6fc98423c6
commit 1f4706e0e7
1 changed files with 27 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

36
coder.py
View file

@ -31,8 +31,16 @@ class Chat:
def request(self, prompt): def request(self, prompt):
self.request_prompt = prompt self.request_prompt = prompt
def quoted_file(self, fname):
prompt = '\n'
prompt += fname.name
prompt += '\n```\n'
prompt += fname.read_text()
prompt += '\n```\n'
return prompt
def plan(self): def plan(self):
prompt = ''' self.plan_prompt = '''
Briefly describe all the code changes needed to complete the user request. Briefly describe all the code changes needed to complete the user request.
Think carefully about the code and the request! Think carefully about the code and the request!
Just describe ALL the changes needed to complete the request. Just describe ALL the changes needed to complete the request.
@ -44,14 +52,11 @@ Just briefly describe the changes.
Request: Request:
''' '''
prompt = self.plan_prompt
prompt += self.request_prompt + '\n###\n' prompt += self.request_prompt + '\n###\n'
for fname in self.fnames: for fname in self.fnames:
prompt += '\n' prompt += self.quoted_file(fname)
prompt += fname.name
prompt += '\n```\n'
prompt += fname.read_text()
prompt += '\n```\n'
### ###
#print(self.system_prompt) #print(self.system_prompt)
@ -93,8 +98,14 @@ Request:
self.update_file(fname) self.update_file(fname)
def update_file(self, fname): def update_file(self, fname):
messages = list(self.messages) prompt = self.plan_prompt
messages.append( prompt += self.request_prompt + '\n###\n'
prompt += self.quoted_file(fname)
messages = [
dict(role = 'system', content = self.system_prompt),
dict(role = 'user', content = prompt),
dict(role = 'assistant', content = self.plan),
dict(role = 'user', dict(role = 'user',
content = f''' content = f'''
Make the requested changes to {fname.name} and output the changed code. Make the requested changes to {fname.name} and output the changed code.
@ -102,8 +113,15 @@ MAKE NO OTHER CHANGES!
JUST OUTPUT CODE. JUST OUTPUT CODE.
NO EXPLANATIONS. NO EXPLANATIONS.
''' '''
)) )
]
dump(messages)
new_content = chat.send(messages) new_content = chat.send(messages)
if new_content.startswith('```\n'):
new_content = new_content[4:]
if new_content.endswith('```\n'):
new_content = new_content[:-4]
fname.write_text(new_content) fname.write_text(new_content)
chat = Chat() chat = Chat()