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

Enable renaming deleting wi root folder by creating a new one

Browse Source
This commit is contained in:
jojorne
2023-03-14 04:04:01 -03:00
parent db7b53f52d
commit 8d5a581d5d
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
koboldai_settings.py
View File

@@ -2260,16 +2260,16 @@ class KoboldWorldInfo(object):
self._socketio.emit("world_info_folder", {x: self.world_info_folder[x] for x in self.world_info_folder}, broadcast=True, room="UI_2") self._socketio.emit("world_info_folder", {x: self.world_info_folder[x] for x in self.world_info_folder}, broadcast=True, room="UI_2")
def delete_folder(self, folder): def delete_folder(self, folder):
if folder == "root":
raise Exception("removing the root folder is not supported")
keys = [key for key in self.world_info] keys = [key for key in self.world_info]
for key in keys: for key in keys:
if self.world_info[key]['folder'] == folder: if self.world_info[key]['folder'] == folder:
self.delete(key) self.delete(key)
if folder in self.world_info_folder: if folder in self.world_info_folder:
del self.world_info_folder[folder] del self.world_info_folder[folder]
self.sync_world_info_to_old_format()
if self._socketio is not None: if self._socketio is not None:
self._socketio.emit("delete_world_info_folder", folder, broadcast=True, room="UI_2") self._socketio.emit("delete_world_info_folder", folder, broadcast=True, room="UI_2")
self._socketio.emit("world_info_folder", {x: self.world_info_folder[x] for x in self.world_info_folder}, broadcast=True, room="UI_2")
logger.debug("Calcing AI Text from WI Folder Delete") logger.debug("Calcing AI Text from WI Folder Delete")
ignore = self._koboldai_vars.calc_ai_text() ignore = self._koboldai_vars.calc_ai_text()
@@ -2446,9 +2446,6 @@ class KoboldWorldInfo(object):
self._socketio.emit("world_info_entry", self.world_info[uid], broadcast=True, room="UI_2") self._socketio.emit("world_info_entry", self.world_info[uid], broadcast=True, room="UI_2")
def delete(self, uid): def delete(self, uid):
if self.world_info[uid]['folder'] == "root":
raise Exception("removing the root folder is not supported")
del self.world_info[uid] del self.world_info[uid]
try: try:
@@ -2470,8 +2467,6 @@ class KoboldWorldInfo(object):
ignore = self._koboldai_vars.calc_ai_text() ignore = self._koboldai_vars.calc_ai_text()
def rename_folder(self, old_folder, folder): def rename_folder(self, old_folder, folder):
if old_folder == "root":
raise Exception("renaming the root folder is not supported")
self.story_settings.gamesaved = False self.story_settings.gamesaved = False
if folder in self.world_info_folder: if folder in self.world_info_folder:
i=0 i=0
@@ -2498,7 +2493,8 @@ class KoboldWorldInfo(object):
self.story_settings.gamesaved = False self.story_settings.gamesaved = False
self.sync_world_info_to_old_format() self.sync_world_info_to_old_format()
if self._socketio is not None: if self._socketio is not None:
self._socketio.emit("world_info_folder", {x: self.world_info_folder[x] for x in self.world_info_folder}, broadcast=True, room="UI_2") self._socketio.emit("delete_world_info_folder", old_folder, broadcast=True, room="UI_2")
self.send_to_ui()
def reorder(self, uid, before): def reorder(self, uid, before):
self.add_item_to_folder(uid, self.world_info[before]['folder'], before=before) self.add_item_to_folder(uid, self.world_info[before]['folder'], before=before)
@@ -2576,6 +2572,11 @@ class KoboldWorldInfo(object):
def sync_world_info_to_old_format(self): def sync_world_info_to_old_format(self):
#Since the old UI uses world info entries for folders, we need to make some up #Since the old UI uses world info entries for folders, we need to make some up
if "root" not in self.world_info_folder:
olf_world_info_folder = self.world_info_folder
self.world_info_folder = OrderedDict()
self.world_info_folder["root"] = []
self.world_info_folder.update(olf_world_info_folder)
folder_entries = {} folder_entries = {}
i=-1 i=-1
for folder in self.world_info_folder: for folder in self.world_info_folder: