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

[fix] fix textgen again

Browse Source
This commit is contained in:
kwaroran
2023-05-09 01:19:36 +09:00
parent ba286099eb
commit 28d8026093
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -111,7 +111,8 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
let DURL = db.textgenWebUIURL let DURL = db.textgenWebUIURL
let bodyTemplate:any let bodyTemplate:any
const proompt = stringlizeChat(formated, currentChar.name) const proompt = stringlizeChat(formated, currentChar.name)
if(DURL.includes('api')){ const isNewAPI = DURL.includes('api')
if(isNewAPI){
bodyTemplate = { bodyTemplate = {
'max_new_tokens': 80, 'max_new_tokens': 80,
'do_sample': true, 'do_sample': true,
@@ -174,6 +175,12 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
console.log(res.data) console.log(res.data)
if(res.ok){ if(res.ok){
try { try {
if(isNewAPI){
return {
type: 'success',
result: dat.results[0].text.substring(proompt.length)
}
}
return { return {
type: 'success', type: 'success',
result: dat.data[0].substring(proompt.length) result: dat.data[0].substring(proompt.length)