rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

retry bug fix

Browse Source
This commit is contained in:
ebolam
2022-08-29 08:30:36 -04:00
parent d6a91fae44
commit c10e534275
2 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
koboldai_settings.py
View File

@@ -812,6 +812,7 @@ class KoboldStoryRegister(object):
def append(self, text):
self.clear_unused_options()
print("setting action_count {} -> {}".format(self.action_count, self.action_count+1))
self.action_count+=1
if self.action_count in self.actions:
if self.actions[self.action_count]["Selected Text"] != text:
@@ -947,6 +948,7 @@ class KoboldStoryRegister(object):
del self.actions[action_step]['Options'][option_number]
#If this is the current spot in the story, advance
if action_step-1 == self.action_count:
print("setting action_count {} -> {}".format(self.action_count, self.action_count+1))
self.action_count+=1
self.socketio.emit("var_changed", {"classname": "actions", "name": "Action Count", "old_value": None, "value":self.action_count}, broadcast=True, room="UI_2")
process_variable_changes(self.socketio, "story", 'actions', {"id": action_step, 'action': self.actions[action_step]}, None)
@@ -961,6 +963,7 @@ class KoboldStoryRegister(object):
self.actions[action_id]["Options"].append({"text": self.actions[action_id]["Selected Text"], "Pinned": False, "Previous Selection": True, "Edited": False})
self.actions[action_id]["Selected Text"] = ""
self.actions[action_id]['Selected Text Length'] = 0
print("setting action_count {} -> {}".format(self.action_count, self.action_count-1))
self.action_count -= 1
process_variable_changes(self.socketio, "story", 'actions', {"id": action_id, 'action': self.actions[action_id]}, None)
self.set_game_saved()
@@ -1051,7 +1054,6 @@ class KoboldStoryRegister(object):
process_variable_changes(self.socketio, "story", 'actions', {"id": self.action_count+1, 'action': self.actions[self.action_count+1]}, None)
def set_probabilites(self, probabilities, action_id=None):
print(probabilities)
if action_id is None:
action_id = self.action_count
if action_id in self.actions: