diff --git a/aiserver.py b/aiserver.py index 66303695..0e27bc08 100644 --- a/aiserver.py +++ b/aiserver.py @@ -4254,7 +4254,7 @@ def kml(txt): # Send start message and tell Javascript to set UI state #==================================================================# def setStartState(): - if(koboldai_vars.welcome): + if koboldai_vars.welcome and isinstance(koboldai_vars.welcome, str): txt = kml(koboldai_vars.welcome) + "
" else: txt = "Welcome to KoboldAI! You are running "+getmodelname()+".
" @@ -6687,7 +6687,7 @@ def load_story_v1(js): data = js["actions_metadata"][key]["Alternative Text"] for i in range(len(js["actions_metadata"][key]["Alternative Text"])): data[i]["text"] = data[i].pop("Text") - koboldai_vars.actions.set_options(data, key) + koboldai_vars.actions.set_options(data, int(key)) # Try not to break older save files if("authorsnote" in js): diff --git a/koboldai_settings.py b/koboldai_settings.py index 54fa75cf..038765e5 100644 --- a/koboldai_settings.py +++ b/koboldai_settings.py @@ -156,7 +156,8 @@ class koboldai_vars(object): #we're going to split our actions by sentence for better context. We'll add in which actions the sentence covers. Prompt will be added at a -1 ID actions = {i: self.actions[i] for i in range(len(self.actions))} actions[-1] = self.prompt - action_text = self.prompt + str(self.actions) + action_text = str(self.actions) + action_text = "{}{}".format(self.prompt, action_text) ###########action_text_split = [sentence, actions used in sentence, token length, included in AI context]################ action_text_split = [[x+" ", [], 0 if self.tokenizer is None else len(self.tokenizer.encode(x+" ")), False] for x in re.split("(?<=[.!?])\s+", action_text)] #The last action shouldn't have the extra space from the sentence splitting, so let's remove it