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

Add gettokens API

Browse Source
This commit is contained in:
kwaroran
2025-03-27 19:21:31 +09:00
parent 58f1cb4fcc
commit 9e00c9b17a
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -12,6 +12,7 @@ import { requestChatData } from "./request";
import { v4 } from "uuid"; import { v4 } from "uuid";
import { getModuleTriggers } from "./modules"; import { getModuleTriggers } from "./modules";
import { Mutex } from "../mutex"; import { Mutex } from "../mutex";
import { tokenize } from "../tokenizer";
let luaFactory:LuaFactory let luaFactory:LuaFactory
let LuaSafeIds = new Set<string>() let LuaSafeIds = new Set<string>()
@@ -154,6 +155,12 @@ export async function runLua(code:string, arg:{
let roleData:'user'|'char' = role === 'user' ? 'user' : 'char' let roleData:'user'|'char' = role === 'user' ? 'user' : 'char'
luaEngineState.chat.message.splice(index, 0, {role: roleData, data: value}) luaEngineState.chat.message.splice(index, 0, {role: roleData, data: value})
}) })
luaEngine.global.set('getTokens', async (id:string, value:string) => {
if(!LuaSafeIds.has(id)){
return
}
return await tokenize(value)
})
luaEngine.global.set('getChatLength', (id:string) => { luaEngine.global.set('getChatLength', (id:string) => {
if(!LuaSafeIds.has(id)){ if(!LuaSafeIds.has(id)){
return return