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): 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 = [ messages = [
ai.fsystem(setup_sys_prompt(dbs)), ai.fsystem(setup_sys_prompt(dbs)),
ai.fuser(f"Instructions: {dbs.input['main_prompt']}"), ai.fuser(f"Instructions: {dbs.input['main_prompt']}"),
ai.fuser(code_ouput), ai.fuser(code_output),
ai.fsystem(dbs.preprompts["fix_code"]), ai.fsystem(dbs.preprompts["fix_code"]),
] ]
messages = ai.next(messages, "Please fix any errors in the code above.") messages = ai.next(messages, "Please fix any errors in the code above.")