Apply suggestions from code review

This commit is contained in:
Anton Osika
2023-06-22 10:44:34 +02:00
committed by GitHub
parent ed1a7da296
commit 8062d901d1

View File

@@ -205,11 +205,11 @@ def use_feedback(ai: AI, dbs: DBs):
def fix_code(ai: AI, dbs: DBs):
code_ouput = json.loads(dbs.logs[gen_code.__name__])[-1]["content"]
code_output = json.loads(dbs.logs[gen_code.__name__])[-1]["content"]
messages = [
ai.fsystem(setup_sys_prompt(dbs)),
ai.fuser(f"Instructions: {dbs.input['main_prompt']}"),
ai.fuser(code_ouput),
ai.fuser(code_output),
ai.fsystem(dbs.preprompts["fix_code"]),
]
messages = ai.next(messages, "Please fix any errors in the code above.")