diff --git a/main.py b/main.py index 585fa519..eaf93887 100644 --- a/main.py +++ b/main.py @@ -68,9 +68,9 @@ async def main(): # Moved defaults out of FLAGS so that we can dynamically select the default persona based on model chosen_human = FLAGS.human if FLAGS.human is not None else humans.DEFAULT - chosen_persona = FLAGS.persona if FLAGS.persona is not None else (personas.GPT35_DEFAULT if 'gpt-3.5' in flags.MODEL else personas.DEFAULT) + chosen_persona = FLAGS.persona if FLAGS.persona is not None else (personas.GPT35_DEFAULT if 'gpt-3.5' in FLAGS.model else personas.DEFAULT) - memgpt_agent = presets.use_preset(presets.DEFAULT, FLAGS.model, personas.get_persona_text(FLAGS.persona), humans.get_human_text(FLAGS.human), interface, persistence_manager) + memgpt_agent = presets.use_preset(presets.DEFAULT, FLAGS.model, chosen_persona, chosen_human, interface, persistence_manager) print_messages = interface.print_messages await print_messages(memgpt_agent.messages)