xiongzhu 2 лет назад
Родитель
Сommit
1f5a75bd46
1 измененных файлов с 2 добавлено и 1 удалено
  1. 2 1
      src/chatapi/chatgpt-api.ts

+ 2 - 1
src/chatapi/chatgpt-api.ts

@@ -208,7 +208,8 @@ export class ChatGPTAPI {
                         body: JSON.stringify(body),
                         signal: abortSignal,
                         onMessage: (data: string) => {
-                            if (data === '[DONE]') {
+                            const json = JSON.parse(data)
+                            if (data === '[DONE]' || json.choices[0]?.finish_reason === 'stop') {
                                 result.text = result.text.trim()
                                 return resolve(result)
                             }