rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-02-20 05:30:57 +01:00
Code Issues Projects Releases Wiki Activity

Allow Lua to stop generation from input modifier

Browse Source
This commit is contained in:
Gnome Ann 2021-12-13 19:32:01 -05:00
parent e9ed8602b2
commit 6e6e0b2b4d
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
aiserver.py
View File

@ -1865,7 +1865,7 @@ def actionsubmit(data, actionmode=0, force_submit=False):
assert False assert False
# Start the game # Start the game
vars.gamestarted = True vars.gamestarted = True
if(not vars.noai): if(not vars.noai and vars.lua_koboldbridge.generating):
# Save this first action as the prompt # Save this first action as the prompt
vars.prompt = data vars.prompt = data
# Clear the startup text from game screen # Clear the startup text from game screen
@ -1905,7 +1905,7 @@ def actionsubmit(data, actionmode=0, force_submit=False):
vars.actions.append(data) vars.actions.append(data)
update_story_chunk('last') update_story_chunk('last')
if(not vars.noai): if(not vars.noai and vars.lua_koboldbridge.generating):
# Off to the tokenizer! # Off to the tokenizer!
calcsubmit(data) calcsubmit(data)
emit('from_server', {'cmd': 'scrolldown', 'data': ''}, broadcast=True) emit('from_server', {'cmd': 'scrolldown', 'data': ''}, broadcast=True)

2
bridge.lua
View File

@ -1607,7 +1607,7 @@ return function(_python, _bridged)
function koboldbridge.execute_inmod() function koboldbridge.execute_inmod()
local r local r
koboldbridge.regeneration_required = false koboldbridge.regeneration_required = false
koboldbridge.generating = false koboldbridge.generating = true
koboldbridge.userstate = "inmod" koboldbridge.userstate = "inmod"
koboldbridge.generated_cols = 0 koboldbridge.generated_cols = 0
koboldbridge.generated = {} koboldbridge.generated = {}