From 28d802609311542cd072bcefd684b65d47c97d8d Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Tue, 9 May 2023 01:19:36 +0900
Subject: [PATCH] [fix] fix textgen again

---
 src/ts/process/request.ts | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

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