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

feat: add lorebook regex matching

Browse Source
This commit is contained in:
kwaroran
2024-06-03 17:28:00 +09:00
parent a39fbee4ff
commit 9583142ee7
1 changed files with 14 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
src/ts/process/lorebook.ts
View File

@@ -236,21 +236,22 @@ export async function loadLoreBookV3Prompt(){
return msg.data
}).join('||')
if(arg.regex){
const regexString = arg.keys[0]
if(!regexString.startsWith('/')){
return false
}
const regexFlag = regexString.split('/').pop()
if(regexFlag){
arg.keys[0] = regexString.replace('/'+regexFlag,'')
try {
const regex = new RegExp(arg.keys[0],regexFlag)
return regex.test(mText)
} catch (error) {
for(const regexString of arg.keys){
if(!regexString.startsWith('/')){
return false
}
const regexFlag = regexString.split('/').pop()
if(regexFlag){
arg.keys[0] = regexString.replace('/'+regexFlag,'')
try {
const regex = new RegExp(arg.keys[0],regexFlag)
return regex.test(mText)
} catch (error) {
return false
}
}
return false
}
return false
}
mText = mText.toLocaleLowerCase()
@@ -433,18 +434,6 @@ export async function loadLoreBookV3Prompt(){
if(!activated || forceState !== 'none' || fullLore[i].alwaysActive){
//if the lore is not activated or force activated, skip the search
}
else if(fullLore[i].useRegex){
const match = searchMatch(currentChat, {
keys: [fullLore[i].key],
searchDepth: scanDepth,
regex: true,
fullWordMatching: fullWordMatching
})
if(!match){
activated = false
}
}
else{
searchQueries.push({
keys: fullLore[i].key.split(','),
@@ -455,7 +444,7 @@ export async function loadLoreBookV3Prompt(){
const result = searchMatch(currentChat, {
keys: query.keys,
searchDepth: scanDepth,
regex: false,
regex: fullLore[i].useRegex,
fullWordMatching: fullWordMatching
})
if(query.negative){