diff --git a/.~lock.model-roleplay-evaluation.ods# b/.~lock.model-roleplay-evaluation.ods# new file mode 100644 index 0000000..d07e8e0 --- /dev/null +++ b/.~lock.model-roleplay-evaluation.ods# @@ -0,0 +1 @@ +,hendrik,deep-thought,16.04.2023 14:39,file:///home/hendrik/.config/libreoffice/4; \ No newline at end of file diff --git a/matrix_pygmalion_bot/ai/model_helpers.py b/matrix_pygmalion_bot/ai/model_helpers.py index b7fc55c..3810355 100644 --- a/matrix_pygmalion_bot/ai/model_helpers.py +++ b/matrix_pygmalion_bot/ai/model_helpers.py @@ -45,7 +45,7 @@ async def get_full_prompt(simple_prompt: str, bot, chat_history, model_name: str elif model_name.startswith("alpaca"): ai_name = bot.name # ToDo user_name = bot.user_name # ToDo - elif mode_name.startswith("koboldai"): + elif model_name.startswith("koboldai"): ai_name = bot.name user_name = "You" else: @@ -69,20 +69,20 @@ async def get_full_prompt(simple_prompt: str, bot, chat_history, model_name: str prompt += f"NSFW content allowed.\n" # ToDo prompt += f"\n" prompt += "### Input:\n" - elif mode_name.startswith("koboldai"): + elif model_name.startswith("koboldai"): prompt = "" else: prompt = "" # Character description if model_name.startswith("pygmalion"): - prompt = bot.name + "'s Persona: " + bot.get_persona() + "\n" + prompt += bot.name + "'s Persona: " + bot.get_persona() + "\n" elif model_name.startswith("vicuna"): prompt += bot.name + "'s Persona: " + bot.get_persona() + "\n" # ToDo elif model_name.startswith("alpaca"): prompt += bot.name + "'s Persona: " + bot.get_persona() + "\n" # ToDo - elif mode_name.startswith("koboldai"): - prompt = f"[Character: {bot.get_persona()}]\n" + elif model_name.startswith("koboldai"): + prompt += f"[Character: {bot.get_persona()}]\n" else: prompt += bot.name + "'s Persona: " + bot.get_persona() + "\n" # ToDo @@ -93,8 +93,8 @@ async def get_full_prompt(simple_prompt: str, bot, chat_history, model_name: str prompt += "Scenario: " + bot.get_scenario() + "\n\n" # ToDo elif model_name.startswith("alpaca"): prompt += "Scenario: " + bot.get_scenario() + "\n\n" # ToDo - elif mode_name.startswith("koboldai"): - prompt = f"[Start Scene: {bot.get_scenario()}]\n\n" + elif model_name.startswith("koboldai"): + prompt += f"[Start Scene: {bot.get_scenario()}]\n\n" else: prompt += "Scenario: " + bot.get_scenario() + "\n\n" # ToDo @@ -105,7 +105,7 @@ async def get_full_prompt(simple_prompt: str, bot, chat_history, model_name: str prompt += "### Response:\n" # ToDo elif model_name.startswith("alpaca"): prompt += "### Response:\n" - elif mode_name.startswith("koboldai"): + elif model_name.startswith("koboldai"): pass else: pass @@ -125,7 +125,7 @@ async def get_full_prompt(simple_prompt: str, bot, chat_history, model_name: str pass # ToDo elif model_name.startswith("alpaca"): pass # ToDo - elif mode_name.startswith("koboldai"): + elif model_name.startswith("koboldai"): pass else: pass # ToDo diff --git a/model-roleplay-evaluation.ods b/model-roleplay-evaluation.ods new file mode 100644 index 0000000..27ff500 Binary files /dev/null and b/model-roleplay-evaluation.ods differ