rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Allow require() in userscripts to import built-in modules

Browse Source
This commit is contained in:
Gnome Ann
2021-12-29 13:47:03 -05:00
parent 38bad263e1
commit 35539a8785
1 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
bridge.lua
View File

@@ -1511,6 +1511,18 @@ return function(_python, _bridged)
koboldbridge.logging_name = nil koboldbridge.logging_name = nil
koboldbridge.filename = nil koboldbridge.filename = nil
local sandbox_require_builtins = {
coroutine = true,
package = true,
string = true,
utf8 = true,
table = true,
math = true,
io = true,
os = true,
debug = true,
}
local old_load = load local old_load = load
local function _safe_load(_g) local function _safe_load(_g)
return function(chunk, chunkname, mode, env) return function(chunk, chunkname, mode, env)
@@ -1541,6 +1553,11 @@ return function(_python, _bridged)
if modname == "bridge" then if modname == "bridge" then
return function() return env.kobold, env.koboldcore end return function() return env.kobold, env.koboldcore end
end end
for k, v in pairs(sandbox_require_builtins) do
if modname == k then
return env[k]
end
end
if type(modname) == "number" then if type(modname) == "number" then
modname = tostring(modname) modname = tostring(modname)
elseif type(modname) ~= "string" then elseif type(modname) ~= "string" then