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

Fix autoContinueMinTokens

Browse Source
This commit is contained in:
kwaroran
2024-04-18 22:06:29 +09:00
parent 44f3d3980c
commit f47f2e95a5
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -55,7 +55,7 @@ export const doingChat = writable(false)
export const chatProcessStage = writable(0) export const chatProcessStage = writable(0)
export const abortChat = writable(false) export const abortChat = writable(false)
export async function sendChat(chatProcessIndex = -1,arg:{chatAdditonalTokens?:number,signal?:AbortSignal,continue?:boolean} = {}):Promise<boolean> { export async function sendChat(chatProcessIndex = -1,arg:{chatAdditonalTokens?:number,signal?:AbortSignal,continue?:boolean,usedContinueTokens?:number} = {}):Promise<boolean> {
chatProcessStage.set(0) chatProcessStage.set(0)
const abortSignal = arg.signal ?? (new AbortController()).signal const abortSignal = arg.signal ?? (new AbortController()).signal
@@ -1132,12 +1132,14 @@ export async function sendChat(chatProcessIndex = -1,arg:{chatAdditonalTokens?:n
} }
} }
if(db.autoContinueMinTokens > 0 && (await tokenize(result)) < db.autoContinueMinTokens){ const resultTokens = await tokenize(result) + (arg.usedContinueTokens || 0)
if(db.autoContinueMinTokens > 0 && resultTokens < db.autoContinueMinTokens){
doingChat.set(false) doingChat.set(false)
return await sendChat(chatProcessIndex, { return await sendChat(chatProcessIndex, {
chatAdditonalTokens: arg.chatAdditonalTokens, chatAdditonalTokens: arg.chatAdditonalTokens,
continue: true, continue: true,
signal: abortSignal signal: abortSignal,
usedContinueTokens: resultTokens
}) })
} }