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

[fix] prompt template

Browse Source
This commit is contained in:
kwaroran
2023-08-04 17:41:05 +09:00
parent e37ec7e47c
commit 3227b33cb1
1 changed files with 37 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -318,8 +318,24 @@ export async function sendChat(chatProcessIndex = -1,arg:{chatAdditonalTokens?:n
break break
} }
case 'chat':{ case 'chat':{
const start = card.rangeStart let start = card.rangeStart
const end = (card.rangeEnd === 'end') ? unformated.chats.length : card.rangeEnd let end = (card.rangeEnd === 'end') ? unformated.chats.length : card.rangeEnd
if(start < 0){
start = unformated.chats.length + start
if(start < 0){
start = 0
}
}
if(end < 0){
end = unformated.chats.length + end
if(end < 0){
end = 0
}
}
if(start >= end){
break
}
const chats = unformated.chats.slice(start, end) const chats = unformated.chats.slice(start, end)
await tokenizeChatArray(chats) await tokenizeChatArray(chats)
break break
@@ -561,8 +577,25 @@ export async function sendChat(chatProcessIndex = -1,arg:{chatAdditonalTokens?:n
break break
} }
case 'chat':{ case 'chat':{
const start = card.rangeStart let start = card.rangeStart
const end = (card.rangeEnd === 'end') ? unformated.chats.length : card.rangeEnd let end = (card.rangeEnd === 'end') ? unformated.chats.length : card.rangeEnd
if(start < 0){
start = unformated.chats.length + start
if(start < 0){
start = 0
}
}
if(end < 0){
end = unformated.chats.length + end
if(end < 0){
end = 0
}
}
if(start >= end){
break
}
const chats = unformated.chats.slice(start, end) const chats = unformated.chats.slice(start, end)
pushPrompts(chats) pushPrompts(chats)
break break