From 68bf3cc7f0a4064cd8d9ce65bcbfac23c31dc52b Mon Sep 17 00:00:00 2001 From: ebolam Date: Mon, 26 Sep 2022 12:55:19 -0400 Subject: [PATCH] WI fix --- aiserver.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/aiserver.py b/aiserver.py index 2d4073d7..692acd96 100644 --- a/aiserver.py +++ b/aiserver.py @@ -7034,8 +7034,18 @@ def load_story_v1(js): if("worldinfo" in js): num = 0 for wi in js["worldinfo"]: + if wi.get("folder", "root") == 'root': + folder = "root" + else: + if 'wifolders_d' in js: + if wi['folder'] in js['wifolders_d']: + folder = js['wifolders_d'][wi['folder']]['name'] + else: + folder = "root" + else: + folder = "root" koboldai_vars.worldinfo_v2.add_item([x.strip() for x in wi["key"].split(",")][0], wi["key"], wi.get("keysecondary", ""), - "root" if wi.get("folder", "root") == 'root' else js['wifolders_d'][wi['folder']]['name'], wi.get("constant", False), + folder, wi.get("constant", False), wi["content"], wi.get("comment", "")) # Save path for save button