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

Refactor requestChatDataMain function to handle non-multiGen

Browse Source
This commit is contained in:
kwaroran
2024-01-16 17:54:42 +09:00
parent 1b1c6ad2fd
commit 1397be4f03
1 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/ts/process/request.ts
View File

@@ -575,12 +575,20 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
const choices = JSON.parse(rawChunk).choices const choices = JSON.parse(rawChunk).choices
for(const choice of choices){ for(const choice of choices){
const chunk = choice.delta.content const chunk = choice.delta.content
const ind = choice.index.toString()
if(chunk){ if(chunk){
if(!readed[ind]){ if(multiGen){
readed[ind] = "" const ind = choice.index.toString()
if(!readed[ind]){
readed[ind] = ""
}
readed[ind] += chunk
}
else{
if(!readed["0"]){
readed["0"] = ""
}
readed["0"] += chunk
} }
readed[ind] += chunk
} }
} }
} catch (error) {} } catch (error) {}