diff --git a/coder.py b/coder.py index 60a12db4e..21d04df03 100755 --- a/coder.py +++ b/coder.py @@ -109,8 +109,8 @@ class Coder: messages += [ dict(role = 'user', content = files_content), dict(role = 'assistant', content = "Ok."), - dict(role = 'user', content = inp + prompts.user_suffix), - dict(role = 'system', content = 'ONLY RETURN CODE USING THE ORIGINAL/UPDATED FORMAT!'), + dict(role = 'user', content = inp), + dict(role = 'system', content = 'REMEMBER, ONLY RETURN CODE USING THE ORIGINAL/UPDATED FORMAT!'), ] content = self.send(messages) @@ -134,9 +134,16 @@ class Coder: print() traceback.print_exc() - def send(self, messages, show_progress = 0): + def show_messages(self, messages): for msg in messages: - dump(msg) + print() + print('-' * 50) + role = msg['role'].upper() + content = msg['content'] + print(f'{role}: {content.strip()}') + + def send(self, messages, show_progress = 0): + self.show_messages(messages) completion = openai.ChatCompletion.create( model="gpt-3.5-turbo", diff --git a/prompts.py b/prompts.py index 2242fba5e..5472d707d 100644 --- a/prompts.py +++ b/prompts.py @@ -98,9 +98,7 @@ files_content_prefix_edited = 'I made your suggested changes, here are the updat files_content_prefix_plain = 'Here are the files I need you to edit:\n\n' -files_content_suffix = '' - -user_suffix = ''' +files_content_suffix = ''' BASE ANY EDITS ON THE CURRENT CONTENTS OF THE FILES AS SHOWN IN THIS MESSAGE. '''