rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2024-12-26 00:03:51 +01:00
Code Issues Projects Releases Wiki Activity

Fix a bug that occurs when userscript doesn't have all 3 modifiers

Browse Source
This commit is contained in:
Gnome Ann 2021-12-13 11:50:10 -05:00
parent 34c52a1a23
commit ed9c2a4d52
2 changed files with 6 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
bridge.lua
View File

@ -1584,9 +1584,9 @@ return function(_python, _bridged)
local _userscript = old_loadfile(join_folder_and_filename(bridged.userscript_path, filename), "t", koboldbridge.get_universe(filename))()
koboldbridge.logging_name = nil
local userscript = deepcopy(KoboldUserScriptModule)
rawset(userscript, "_inmod", function() koboldbridge.logging_name = modulenames[i]; _userscript.inmod() end)
rawset(userscript, "_genmod", function() koboldbridge.logging_name = modulenames[i]; _userscript.genmod() end)
rawset(userscript, "_outmod", function() koboldbridge.logging_name = modulenames[i]; _userscript.outmod() end)
rawset(userscript, "_inmod", function() koboldbridge.logging_name = modulenames[i]; if _userscript.inmod ~= nil then _userscript.inmod() end end)
rawset(userscript, "_genmod", function() koboldbridge.logging_name = modulenames[i]; if _userscript.genmod ~= nil then _userscript.genmod() end end)
rawset(userscript, "_outmod", function() koboldbridge.logging_name = modulenames[i]; if _userscript.outmod ~= nil then _userscript.outmod() end end)
rawset(userscript, "_filename", filename)
rawset(userscript, "_modulename", modulenames[i])
rawset(userscript, "_description", descriptions[i])

6
cores/default.lua
View File

@ -12,28 +12,22 @@ local corescript = {}
function corescript.inmod()
for i = #koboldcore.userscripts, 1, -1 do
local userscript = koboldcore.userscripts[i]
if userscript.inmod ~= nil then
userscript.inmod()
end
end
end
-- Run all the generation modifiers from top to bottom
function corescript.genmod()
for i, userscript in ipairs(koboldcore.userscripts) do
if userscript.genmod ~= nil then
userscript.genmod()
end
end
end
-- Run all the generation modifiers from top to bottom
function corescript.outmod()
for i, userscript in ipairs(koboldcore.userscripts) do
if userscript.outmod ~= nil then
userscript.outmod()
end
end
end