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

Compatability update for latest AIDCAT export format

Browse Source
This commit is contained in:
KoboldAI Dev
2021-05-16 17:45:21 -04:00
parent ce2e4e1f9e
commit 95cb94e979
1 changed files with 23 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
aiserver.py
View File

@ -1235,7 +1235,10 @@ def importRequest():
ob["descr"] = story["description"] ob["descr"] = story["description"]
else: else:
ob["descr"] = "(No Description)" ob["descr"] = "(No Description)"
ob["acts"] = len(story["actions"]) if("actions" in story):
ob["acts"] = len(story["actions"])
elif("actionWindow" in story):
ob["acts"] = len(story["actionWindow"])
emit('from_server', {'cmd': 'addimportline', 'data': ob}) emit('from_server', {'cmd': 'addimportline', 'data': ob})
num += 1 num += 1
@ -1252,8 +1255,17 @@ def importgame():
# Copy game contents to vars # Copy game contents to vars
vars.gamestarted = True vars.gamestarted = True
if(len(ref["actions"]) > 0):
vars.prompt = ref["actions"][0]["text"] if("actions" in ref):
if(len(ref["actions"]) > 0):
vars.prompt = ref["actions"][0]["text"]
else:
vars.prompt = ""
elif("actionWindow" in ref):
if(len(ref["actionWindow"]) > 0):
vars.prompt = ref["actionWindow"][0]["text"]
else:
vars.prompt = ""
else: else:
vars.prompt = "" vars.prompt = ""
vars.memory = ref["memory"] vars.memory = ref["memory"]
@ -1262,9 +1274,14 @@ def importgame():
vars.worldinfo = [] vars.worldinfo = []
# Get all actions except for prompt # Get all actions except for prompt
if(len(ref["actions"]) > 1): if("actions" in ref):
for act in ref["actions"][1:]: if(len(ref["actions"]) > 1):
vars.actions.append(act["text"]) for act in ref["actions"][1:]:
vars.actions.append(act["text"])
elif("actionWindow" in ref):
if(len(ref["actionWindow"]) > 1):
for act in ref["actionWindow"][1:]:
vars.actions.append(act["text"])
# Get just the important parts of world info # Get just the important parts of world info
if(ref["worldInfo"] != None): if(ref["worldInfo"] != None):