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

Enable more of the IO library in Lua sandbox

Browse Source 
Also changes the Lua warning color to red.
This commit is contained in:
Gnome Ann 2021-12-13 11:22:58 -05:00
parent 28e86563b8
commit 11f9866dbe
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
View File

@ -1018,7 +1018,7 @@ def lua_warn(msg):
if(vars.lua_logname != vars.lua_koboldbridge.logging_name): if(vars.lua_logname != vars.lua_koboldbridge.logging_name):
vars.lua_logname = vars.lua_koboldbridge.logging_name vars.lua_logname = vars.lua_koboldbridge.logging_name
print(colors.BLUE + lua_log_format_name(vars.lua_logname) + ":" + colors.END, file=sys.stderr) print(colors.BLUE + lua_log_format_name(vars.lua_logname) + ":" + colors.END, file=sys.stderr)
print(colors.YELLOW + msg.replace("\033", "") + colors.END) print(colors.RED + msg.replace("\033", "") + colors.END)
#==================================================================# #==================================================================#
# Decode tokens into a string using current tokenizer # Decode tokens into a string using current tokenizer

5
bridge.lua
View File

@ -1509,6 +1509,11 @@ return function(_python, _bridged)
ult = math.ult, ult = math.ult,
}, },
io = { io = {
stdin = io.stdin,
stdout = io.stdout,
stderr = io.stderr,
input = io.input,
output = io.output,
read = io.read, read = io.read,
write = io.write, write = io.write,
flush = io.flush, flush = io.flush,