rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

Fix random story generator when No Prompt Gen is enabled

This commit is contained in:
Gnome Ann 2021-12-26 19:40:20 -05:00
parent 6aa2f50045
commit 1ff563ebda
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
aiserver.py
View File

@ -2055,7 +2055,7 @@ def settingschanged():
#==================================================================#
# Take input text from SocketIO and decide what to do with it
#==================================================================#
def actionsubmit(data, actionmode=0, force_submit=False):
def actionsubmit(data, actionmode=0, force_submit=False, force_prompt_gen=False):
# Ignore new submissions if the AI is currently busy
if(vars.aibusy):
return
@ -2092,7 +2092,7 @@ def actionsubmit(data, actionmode=0, force_submit=False):
assert False
# Start the game
vars.gamestarted = True
if(not vars.noai and vars.lua_koboldbridge.generating and not vars.nopromptgen):
if(not vars.noai and vars.lua_koboldbridge.generating and (not vars.nopromptgen or force_prompt_gen)):
# Save this first action as the prompt
vars.prompt = data
# Clear the startup text from game screen
@ -4152,7 +4152,7 @@ def randomGameRequest(topic):
newGameRequest()
vars.memory = "You generate the following " + topic + " story concept :"
vars.lua_koboldbridge.feedback = None
actionsubmit("", force_submit=True)
actionsubmit("", force_submit=True, force_prompt_gen=True)
vars.memory = ""
#==================================================================#