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

[feat] trigger runner2

Browse Source
This commit is contained in:
kwaroran
2023-07-28 05:53:12 +09:00
parent 0e9f3cbce0
commit 7b768d3e73
7 changed files with 186 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/lang/en.ts
View File

@@ -359,10 +359,11 @@ export const languageEnglish = {
changeFolderColor: "Change Folder Color",
fullWordMatching: "Full Word Matching",
botSettingAtStart: "Bot Menu when Launch",
triggerStart: "On start of a chat",
triggerStart: "On chat Send",
triggerInput: "On user's output",
triggerOutput: "On character's output",
triggerManual: "Manual Trigger Only",
triggerCondVar: "If Variable is",
triggerManual: "Manual trigger only",
triggerCondVar: "If Variable",
triggerCondExists: "If Text Exists on Chat",
triggerScript: "Trigger Script",
triggerMatchRegex: "Match with Regex",
@@ -376,7 +377,7 @@ export const languageEnglish = {
greaterEqual: "Greater or Equal to",
lessEqual: "Less or Equal to",
triggerEffSysPrompt: 'Add System Prompt',
triggerEffSetVar: 'Set Variable',
triggerEffSetVar: 'Modify Variable',
triggerEffImperson: 'Send Chat',
varableName: "Variable Name",
role: "Role",
@@ -386,5 +387,14 @@ export const languageEnglish = {
historyend: "End of History",
always: "Always",
noEffect: "No Effect",
invaildTriggerEffect: "This effect doesn't works for this trigger type.",
operator: "Operator",
TriggerSetToVar: "Set to Variable",
TriggerAddToVar: "Add to Variable",
TriggerSubToVar: "Subtract from Variable",
TriggerMulToVar: "Multiply to Variable",
TriggerDivToVar: "Divide from Variable",
isNull: "is not set",
ifChatIndex: "If chat index",
ifRandom: "If random",
}

6
src/lib/ChatScreens/DefaultChatScreen.svelte
View File

@@ -19,6 +19,7 @@
import Help from '../Others/Help.svelte';
import AssetInput from './AssetInput.svelte';
import { downloadFile } from 'src/ts/storage/globalApi';
import { runTrigger } from 'src/ts/process/triggers';
let messageInput:string = ''
let messageInputTranslate:string = ''
@@ -61,6 +62,11 @@
else{
const char = $DataBase.characters[selectedChar]
if(char.type === 'character'){
let triggerResult = runTrigger(char,'input', {chat: char.chats[char.chatPage]})
if(triggerResult){
cha = triggerResult.chat.message
}
cha.push({
role: 'user',
data: processScript(char,messageInput,'editinput')

2
src/lib/SideBars/CharConfig.svelte
View File

@@ -469,7 +469,7 @@
let script = currentChar.data.triggerscript
script.push({
comment: "",
type: "output",
type: "start",
conditions: [],
effect: []
})

44
src/lib/SideBars/Scripts/TriggerData.svelte
View File

@@ -50,8 +50,9 @@
<TextInput size="sm" bind:value={value.comment} />
<span class="text-neutral-200 mt-4">{language.type}</span>
<SelectInput bind:value={value.type}>
<OptionInput value="output">{language.triggerOutput}</OptionInput>
<OptionInput value="start">{language.triggerStart}</OptionInput>
<OptionInput value="output">{language.triggerOutput}</OptionInput>
<OptionInput value="input">{language.triggerInput}</OptionInput>
<OptionInput value="manual">{language.triggerManual}</OptionInput>
</SelectInput>
@@ -100,9 +101,18 @@
operator: '='
}
}
if(cond.type === 'chatindex'){
cond = {
type: 'chatindex',
value: '',
operator: '='
}
}
}}>
<OptionInput value="exists">{language.triggerCondExists}</OptionInput>
<OptionInput value="var">{language.triggerCondVar}</OptionInput>
<OptionInput value="chatindex">{language.ifChatIndex}</OptionInput>
</SelectInput>
{#if cond.type === 'exists'}
@@ -117,9 +127,11 @@
<span class="text-gray-400 text-sm">{language.searchDepth}</span>
<NumberInput size="sm" bind:value={cond.depth} />
{/if}
{#if cond.type === 'var' || cond.type === 'chatindex'}
{#if cond.type === 'var'}
<span class="text-gray-400 text-sm">{language.varableName}</span>
<TextInput size="sm" bind:value={cond.var} />
{/if}
<span class="text-gray-400 text-sm">{language.value}</span>
<SelectInput bind:value={cond.operator} size="sm">
<OptionInput value="=">{language.equal}</OptionInput>
@@ -128,19 +140,33 @@
<OptionInput value="<">{language.less}</OptionInput>
<OptionInput value=">=">{language.greaterEqual}</OptionInput>
<OptionInput value="<=">{language.lessEqual}</OptionInput>
<OptionInput value="null">{language.isNull}</OptionInput>
</SelectInput>
{#if cond.operator !== 'null'}
<TextInput size="sm" bind:value={cond.value} />
{/if}
{/if}
{/each}
</div>
<span class="text-neutral-200 mt-4">Effects
<button aria-labelledby="Add Effects" class="float-right text-gray-400 hover:text-green-500" on:click={() => {
if(value.type === 'start'){
value.effect.push({
type: 'systemprompt',
value: '',
location: 'historyend'
})
}
else{
value.effect.push({
type: 'setvar',
var: '',
value: '',
operator: '='
})
}
value.effect = value.effect
}}><PlusIcon size={18} /></button>
@@ -174,7 +200,8 @@
effect = {
type: 'setvar',
var: '',
value: ''
value: '',
operator: '='
}
}
if(effect.type === 'impersonate'){
@@ -185,11 +212,16 @@
}
}
}}>
{#if effect.type === 'systemprompt' || value.type === 'start'}
<OptionInput value="systemprompt">{language.triggerEffSysPrompt}</OptionInput>
{/if}
<OptionInput value="setvar">{language.triggerEffSetVar}</OptionInput>
<OptionInput value="impersonate">{language.triggerEffImperson}</OptionInput>
</SelectInput>
{#if effect.type === 'systemprompt'}
{#if value.type !== 'start'}
<span class="text-red-400 text-sm">{language.invaildTriggerEffect}</span>
{/if}
<span class="text-gray-400 text-sm">{language.location}</span>
<SelectInput bind:value={effect.location} size="sm">
<OptionInput value="start">{language.promptstart}</OptionInput>
@@ -202,6 +234,14 @@
{#if effect.type === 'setvar'}
<span class="text-gray-400 text-sm">{language.varableName}</span>
<TextInput size="sm" bind:value={effect.var} />
<span class="text-gray-400 text-sm">{language.operator}</span>
<SelectInput bind:value={effect.operator} size="sm">
<OptionInput value="=">{language.TriggerSetToVar}</OptionInput>
<OptionInput value="+=">{language.TriggerAddToVar}</OptionInput>
<OptionInput value="-=">{language.TriggerSubToVar}</OptionInput>
<OptionInput value="*=">{language.TriggerMulToVar}</OptionInput>
<OptionInput value="/=">{language.TriggerDivToVar}</OptionInput>
</SelectInput>
<span class="text-gray-400 text-sm">{language.value}</span>
<TextInput size="sm" bind:value={effect.value} />
{/if}

10
src/ts/parser.ts
View File

@@ -305,7 +305,7 @@ function wppParser(data:string){
const rgx = /(?:{{|<)(.+?)(?:}}|>)/gm
type matcherArg = {chatID:number,db:Database,chara:character|string,rmVar:boolean}
type matcherArg = {chatID:number,db:Database,chara:character|string,rmVar:boolean,var?:{[key:string]:string}}
const matcher = (p1:string,matcherArg:matcherArg) => {
if(p1.length > 10000){
return ''
@@ -448,7 +448,7 @@ const matcher = (p1:string,matcherArg:matcherArg) => {
const v = arra[1]
switch(arra[0]){
case 'getvar':{
const d =getVarChat(chatID)
const d = matcherArg.var ?? getVarChat(chatID)
return d[v] ?? "[Null]"
}
case 'calc':{
@@ -568,7 +568,8 @@ export function risuChatParser(da:string, arg:{
chatID?:number
db?:Database
chara?:string|character|groupChat
rmVar?:boolean
rmVar?:boolean,
var?:{[key:string]:string}
} = {}):string{
const chatID = arg.chatID ?? -1
const db = arg.db ?? get(DataBase)
@@ -597,7 +598,8 @@ export function risuChatParser(da:string, arg:{
chatID: chatID,
chara: chara,
rmVar: arg.rmVar ?? false,
db: db
db: db,
var: arg.var ?? null
}
while(pointer < da.length){
switch(da[pointer]){

45
src/ts/process/index.ts
View File

@@ -15,6 +15,7 @@ import { supaMemory } from "./memory/supaMemory";
import { v4 } from "uuid";
import { cloneDeep } from "lodash";
import { groupOrder } from "./group";
import { runTrigger, type additonalSysPrompt } from "./triggers";
export interface OpenAIChat{
role: 'system'|'user'|'assistant'|'function'
@@ -185,7 +186,6 @@ export async function sendChat(chatProcessIndex = -1,arg:{chatAdditonalTokens?:n
chatObjects.push({ role, content });
}
console.log(chatObjects)
return chatObjects;
}
@@ -288,7 +288,14 @@ export async function sendChat(chatProcessIndex = -1,arg:{chatAdditonalTokens?:n
currentTokens += await tokenizer.tokenizeChat(chat)
}
const ms = currentChat.message
let ms = currentChat.message
const triggerResult = runTrigger(currentChar, 'start', {chat: currentChat})
if(triggerResult){
currentChat = triggerResult.chat
ms = currentChat.message
}
for(const msg of ms){
let formedChat = processScript(nowChatroom,risuChatParser(msg.data, {chara: currentChar, rmVar: true}), 'editprocess')
let name = ''
@@ -368,6 +375,29 @@ export async function sendChat(chatProcessIndex = -1,arg:{chatAdditonalTokens?:n
unformated.chats = chats
if(triggerResult){
if(triggerResult.additonalSysPrompt.promptend){
unformated.postEverything.push({
role: 'system',
content: triggerResult.additonalSysPrompt.promptend
})
}
if(triggerResult.additonalSysPrompt.historyend){
unformated.lastChat.push({
role: 'system',
content: triggerResult.additonalSysPrompt.historyend
})
}
if(triggerResult.additonalSysPrompt.start){
unformated.lastChat.unshift({
role: 'system',
content: triggerResult.additonalSysPrompt.start
})
}
}
//make into one
let formated:OpenAIChat[] = []
@@ -462,6 +492,11 @@ export async function sendChat(chatProcessIndex = -1,arg:{chatAdditonalTokens?:n
}
}
const triggerResult = runTrigger(currentChar, 'output', {chat:currentChat})
if(triggerResult){
db.characters[selectedChar].chats[selectedChat] = triggerResult.chat
setDatabase(db)
}
await sayTTS(currentChar, result)
}
else{
@@ -477,6 +512,10 @@ export async function sendChat(chatProcessIndex = -1,arg:{chatAdditonalTokens?:n
saying: currentChar.chaId
})
db.characters[selectedChar].reloadKeys += 1
const triggerResult = runTrigger(currentChar, 'output', {chat:currentChat})
if(triggerResult){
db.characters[selectedChar].chats[selectedChat] = triggerResult.chat
}
await sayTTS(currentChar, result)
setDatabase(db)
}
@@ -646,7 +685,6 @@ export async function sendChat(chatProcessIndex = -1,arg:{chatAdditonalTokens?:n
const msgs = db.characters[selectedChar].chats[selectedChat].message
let msgStr = ''
for(let i = (msgs.length - 1);i>=0;i--){
console.log(i,msgs.length,msgs[i])
if(msgs[i].role === 'char'){
msgStr = `character: ${msgs[i].data.replace(/\n/, ' ')} \n` + msgStr
}
@@ -663,5 +701,6 @@ export async function sendChat(chatProcessIndex = -1,arg:{chatAdditonalTokens?:n
setDatabase(db)
}
}
return true
}

83
src/ts/process/triggers.ts
View File

@@ -1,5 +1,6 @@
import { getVarChat } from "../parser";
import type { character } from "../storage/database";
import { cloneDeep } from "lodash";
import { getVarChat, risuChatParser } from "../parser";
import type { Chat, character } from "../storage/database";
export interface triggerscript{
comment: string;
@@ -8,7 +9,7 @@ export interface triggerscript{
effect:triggerEffect[]
}
export type triggerCondition = triggerConditionsVar|triggerConditionsExists
export type triggerCondition = triggerConditionsVar|triggerConditionsExists|triggerConditionsChatIndex
export type triggerEffect = triggerEffectSetvar|triggerEffectSystemPrompt|triggerEffectImpersonate
@@ -16,7 +17,13 @@ export type triggerConditionsVar = {
type:'var'
var:string
value:string
operator:'='|'!='|'>'|'<'|'>='|'<='
operator:'='|'!='|'>'|'<'|'>='|'<='|'null'
}
export type triggerConditionsChatIndex = {
type:'chatindex'
value:string
operator:'='|'!='|'>'|'<'|'>='|'<='|'null'
}
export type triggerConditionsExists ={
@@ -45,8 +52,17 @@ export interface triggerEffectImpersonate{
value:string
}type triggerMode = 'start'|'manual'|'output'|'input'
export function runTrigger(char:character,mode:triggerMode){
let additonalSysPrompt = {
export type additonalSysPrompt = {
start:string,
historyend: string,
promptend: string
}
export function runTrigger(char:character,mode:triggerMode, arg:{
chat?: Chat
} = {}){
char = cloneDeep(char)
let additonalSysPrompt:additonalSysPrompt = {
start:'',
historyend: '',
promptend: ''
@@ -54,9 +70,9 @@ export function runTrigger(char:character,mode:triggerMode){
let varValues = getVarChat(-1, char)
let varValuesChanged = false
const triggers = char.triggerscript
const chat = char.chats[char.chatPage]
const chat = arg.chat ?? char.chats[char.chatPage]
if(!triggers){
return {additonalSysPrompt, char}
return null
}
for(const trigger of triggers){
@@ -66,61 +82,63 @@ export function runTrigger(char:character,mode:triggerMode){
let pass = true
for(const condition of trigger.conditions){
if(condition.type === 'var'){
const varValue = varValues[condition.var]
if(condition.type === 'var' || condition.type === 'chatindex'){
const varValue = (condition.type === 'var') ? (varValues[condition.var] ?? '[Null]') : (chat.message.length)
if(varValue === undefined || varValue === null){
pass = false
break
}
else{
if(condition.operator === '='){
switch(condition.operator){
case '=':
if(varValue !== condition.value){
pass = false
}
break
}
}
else if(condition.operator === '!='){
case '!=':
if(varValue === condition.value){
pass = false
}
break
}
}
else if(condition.operator === '>'){
case '>':
if(Number(varValue) > Number(condition.value)){
pass = false
}
break
}
}
else if(condition.operator === '<'){
case '<':
if(Number(varValue) < Number(condition.value)){
pass = false
}
break
}
}
else if(condition.operator === '>='){
case '>=':
if(Number(varValue) >= Number(condition.value)){
pass = false
}
break
}
}
else if(condition.operator === '<='){
case '<=':
if(Number(varValue) <= Number(condition.value)){
pass = false
break
}
break
case 'null':
if(varValue !== '[Null]'){
pass = false
}
break
}
}
}
else if(condition.type === 'exists'){
const val = risuChatParser(condition.value,{chara:char, var:varValues})
let da = chat.message.slice(0-condition.depth).map((v)=>v.data).join(' ')
if(condition.type2 === 'strict'){
pass = da.split(' ').includes(condition.value)
pass = da.split(' ').includes(val)
}
else if(condition.type2 === 'loose'){
pass = da.toLowerCase().includes(condition.value.toLowerCase())
pass = da.toLowerCase().includes(val.toLowerCase())
}
else if(condition.type2 === 'regex'){
pass = new RegExp(condition.value).test(da)
pass = new RegExp(val).test(da)
}
}
if(!pass){
@@ -169,8 +187,9 @@ export function runTrigger(char:character,mode:triggerMode){
chat.message[chat.message.length-1].data = chat.message.at(-1).data.replaceAll(/{{(setvar|getvar)::.+?}}/gis,'') + Object.keys(varValues).map((v)=>`{{setvar::${v}::${varValues[v]}}}`).join('')
}
if(arg.chat !== undefined && arg.chat !== null){
char.chats[char.chatPage] = chat
return {additonalSysPrompt, char}
}
return {additonalSysPrompt, chat}
}