mincomk/risuai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Allow more data to be readable from Lua

Browse Source
This commit is contained in:
enzi221
2025-05-18 22:03:14 +09:00
parent 6e5cf5eb48
commit 8fdd8d3beb
1 changed files with 17 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
src/ts/process/lua.ts
View File

@@ -67,22 +67,16 @@ export async function runLua(code:string, arg:{
luaEngineState.code = code luaEngineState.code = code
luaEngineState.engine = await luaFactory.createEngine({injectObjects: true}) luaEngineState.engine = await luaFactory.createEngine({injectObjects: true})
const luaEngine = luaEngineState.engine const luaEngine = luaEngineState.engine
luaEngine.global.set('getChatVar', (id:string,key:string) => {
return luaEngineState.getVar(key)
})
luaEngine.global.set('setChatVar', (id:string,key:string, value:string) => { luaEngine.global.set('setChatVar', (id:string,key:string, value:string) => {
if(!LuaSafeIds.has(id) && !LuaEditDisplayIds.has(id)){ if(!LuaSafeIds.has(id) && !LuaEditDisplayIds.has(id)){
return return
} }
luaEngineState.setVar(key, value) luaEngineState.setVar(key, value)
}) })
luaEngine.global.set('getChatVar', (id:string,key:string) => {
if(!LuaSafeIds.has(id) && !LuaEditDisplayIds.has(id)){
return
}
return luaEngineState.getVar(key)
})
luaEngine.global.set('getGlobalVar', (id:string, key:string) => { luaEngine.global.set('getGlobalVar', (id:string, key:string) => {
if(!LuaSafeIds.has(id) && !LuaEditDisplayIds.has(id)){
return
}
return getGlobalChatVar(key) return getGlobalChatVar(key)
}) })
luaEngine.global.set('stopChat', (id:string) => { luaEngine.global.set('stopChat', (id:string) => {
@@ -166,11 +160,9 @@ export async function runLua(code:string, arg:{
return await tokenize(value) return await tokenize(value)
}) })
luaEngine.global.set('getChatLength', (id:string) => { luaEngine.global.set('getChatLength', (id:string) => {
if(!LuaSafeIds.has(id)){
return
}
return luaEngineState.chat.message.length return luaEngineState.chat.message.length
}) })
luaEngine.global.set('getFullChatMain', (id:string) => { luaEngine.global.set('getFullChatMain', (id:string) => {
const data = JSON.stringify(luaEngineState.chat.message.map((v) => { const data = JSON.stringify(luaEngineState.chat.message.map((v) => {
return { return {
@@ -183,10 +175,11 @@ export async function runLua(code:string, arg:{
}) })
luaEngine.global.set('setFullChatMain', (id:string, value:string) => { luaEngine.global.set('setFullChatMain', (id:string, value:string) => {
const realValue = JSON.parse(value)
if(!LuaSafeIds.has(id)){ if(!LuaSafeIds.has(id)){
return return
} }
const realValue = JSON.parse(value)
luaEngineState.chat.message = realValue.map((v) => { luaEngineState.chat.message = realValue.map((v) => {
return { return {
role: v.role, role: v.role,
@@ -396,9 +389,6 @@ export async function runLua(code:string, arg:{
}) })
luaEngine.global.set('getName', async (id:string) => { luaEngine.global.set('getName', async (id:string) => {
if(!LuaSafeIds.has(id)){
return
}
const db = getDatabase() const db = getDatabase()
const selectedChar = get(selectedCharID) const selectedChar = get(selectedCharID)
const char = db.characters[selectedChar] const char = db.characters[selectedChar]
@@ -436,6 +426,13 @@ export async function runLua(code:string, arg:{
setDatabase(db) setDatabase(db)
}) })
luaEngine.global.set('getCharacterFirstMessage', async (id:string) => {
const db = getDatabase()
const selectedChar = get(selectedCharID)
const char = db.characters[selectedChar]
return char.firstMessage
})
luaEngine.global.set('setCharacterFirstMessage', async (id:string, data:string) => { luaEngine.global.set('setCharacterFirstMessage', async (id:string, data:string) => {
if(!LuaSafeIds.has(id)){ if(!LuaSafeIds.has(id)){
return return
@@ -452,29 +449,11 @@ export async function runLua(code:string, arg:{
return true return true
}) })
luaEngine.global.set('getCharacterFirstMessage', async (id:string) => {
if(!LuaSafeIds.has(id)){
return
}
const db = getDatabase()
const selectedChar = get(selectedCharID)
const char = db.characters[selectedChar]
return char.firstMessage
})
luaEngine.global.set('getPersonaName', (id:string) => { luaEngine.global.set('getPersonaName', (id:string) => {
if(!LuaSafeIds.has(id)){
return
}
return getUserName() return getUserName()
}) })
luaEngine.global.set('getPersonaDescription', (id:string) => { luaEngine.global.set('getPersonaDescription', (id:string) => {
if(!LuaSafeIds.has(id)){
return
}
const db = getDatabase() const db = getDatabase()
const selectedChar = get(selectedCharID) const selectedChar = get(selectedCharID)
const char = db.characters[selectedChar] const char = db.characters[selectedChar]
@@ -482,6 +461,10 @@ export async function runLua(code:string, arg:{
return risuChatParser(getPersonaPrompt(), { chara: char }) return risuChatParser(getPersonaPrompt(), { chara: char })
}) })
luaEngine.global.set('getAuthorsNote', (id:string) => {
return luaEngineState.chat?.note ?? ''
})
luaEngine.global.set('getBackgroundEmbedding', async (id:string) => { luaEngine.global.set('getBackgroundEmbedding', async (id:string) => {
if(!LuaSafeIds.has(id)){ if(!LuaSafeIds.has(id)){
return return
@@ -508,10 +491,6 @@ export async function runLua(code:string, arg:{
// Lore books // Lore books
luaEngine.global.set('getLoreBooksMain', (id:string, search: string) => { luaEngine.global.set('getLoreBooksMain', (id:string, search: string) => {
if(!LuaSafeIds.has(id)){
return
}
const db = getDatabase() const db = getDatabase()
const selectedChar = db.characters[get(selectedCharID)] const selectedChar = db.characters[get(selectedCharID)]
if (selectedChar.type !== 'character') { if (selectedChar.type !== 'character') {