From 6e6e0b2b4df523ec661352ba145b3115e228569e Mon Sep 17 00:00:00 2001
From: Gnome Ann <>
Date: Mon, 13 Dec 2021 19:32:01 -0500
Subject: [PATCH] Allow Lua to stop generation from input modifier

---
 aiserver.py | 4 ++--
 bridge.lua  | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/aiserver.py b/aiserver.py
index 05ff92fb..ee0457fa 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -1865,7 +1865,7 @@ def actionsubmit(data, actionmode=0, force_submit=False):
             assert False
         # Start the game
         vars.gamestarted = True
-        if(not vars.noai):
+        if(not vars.noai and vars.lua_koboldbridge.generating):
             # Save this first action as the prompt
             vars.prompt = data
             # Clear the startup text from game screen
@@ -1905,7 +1905,7 @@ def actionsubmit(data, actionmode=0, force_submit=False):
                 vars.actions.append(data)
             update_story_chunk('last')
 
-        if(not vars.noai):
+        if(not vars.noai and vars.lua_koboldbridge.generating):
             # Off to the tokenizer!
             calcsubmit(data)
             emit('from_server', {'cmd': 'scrolldown', 'data': ''}, broadcast=True)
diff --git a/bridge.lua b/bridge.lua
index bcb32203..77178916 100644
--- a/bridge.lua
+++ b/bridge.lua
@@ -1607,7 +1607,7 @@ return function(_python, _bridged)
     function koboldbridge.execute_inmod()
         local r
         koboldbridge.regeneration_required = false
-        koboldbridge.generating = false
+        koboldbridge.generating = true
         koboldbridge.userstate = "inmod"
         koboldbridge.generated_cols = 0
         koboldbridge.generated = {}