rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-02-17 12:10:49 +01:00
Code Issues Projects Releases Wiki Activity

Don't error in update_story_chunk if chunk index doesn't exist

Browse Source
This commit is contained in:
Gnome Ann 2022-01-21 21:19:32 -05:00
parent 9cddaa8041
commit bf2b02d366
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
aiserver.py
View File

@ -3237,6 +3237,8 @@ def update_story_chunk(idx: Union[int, str]):
else:
# Actions are 0 based, but in chunks 0 is the prompt.
# So the chunk index is one more than the corresponding action index.
if(idx - 1 not in vars.actions):
return
text = vars.actions[idx - 1]
item = html.escape(text)
@ -3365,7 +3367,9 @@ def inlineedit(chunk, data):
else:
if(chunk-1 in vars.actions):
vars.actions[chunk-1] = data
else:
print(f"WARNING: Attempted to edit non-existent chunk {chunk}")
setgamesaved(False)
update_story_chunk(chunk)
emit('from_server', {'cmd': 'texteffect', 'data': chunk}, broadcast=True)
@ -3386,6 +3390,8 @@ def inlinedelete(chunk):
else:
if(chunk-1 in vars.actions):
del vars.actions[chunk-1]
else:
print(f"WARNING: Attempted to delete non-existent chunk {chunk}")
setgamesaved(False)
remove_story_chunk(chunk)
emit('from_server', {'cmd': 'editmode', 'data': 'false'}, broadcast=True)