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

Merge branch 'UI2' of https://github.com/ebolam/KoboldAI into UI2

Browse Source
This commit is contained in:
ebolam
2022-08-17 13:15:12 -04:00
parent e798fc9584 7e722763e7
commit bbd3e60bce
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
koboldai_settings.py
View File

@@ -1027,12 +1027,16 @@ class KoboldStoryRegister(object):
process_variable_changes(self.socketio, "actions", "Selected Text", {"id": self.action_count+1, "text": self.actions[self.action_count+1]['Selected Text']}, None)
process_variable_changes(self.socketio, "actions", 'Selected Text Length', {"id": self.action_count+1, 'length': self.actions[self.action_count+1]['Selected Text Length']}, {"id": self.action_count, 'length': 0})
def set_probabilites(self, probabilities, action_id=self.action_count):
def set_probabilites(self, probabilities, action_id=None):
if action_id is None:
action_id = self.action_count
if action_id in self.actions:
self.actions[action_id]['Probabilities'] = probabilities
process_variable_changes(self.socketio, "actions", 'Probabilities', {"id": action_id, 'Probabilities': self.actions[action_id]['Probabilities']}, None)
def set_option_probabilities(self, probabilities, option_number, action_id=self.action_count+1):
def set_option_probabilities(self, probabilities, option_number, action_id=None):
if action_id is None:
action_id = self.action_count+1
if action_id in self.actions:
old_options = self.actions[action_id]["options"]
if option_number < len(self.actions[action_id]["options"]):