chore

aixMonster · Mar 24, 2024 · a3b7d2c · a3b7d2c
1 parent e3006b0
commit a3b7d2c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/services/apis/custom-api.mjs b/src/services/apis/custom-api.mjs
@@ -67,6 +67,7 @@ export async function generateAnswersWithCustomApi(port, question, session, apiK
         pushRecord(session, question, answer)
         console.debug('conversation history', { content: session.conversationRecords })
         port.postMessage({ answer: null, done: true, session: session })
+        return
       }
 
       if (data.response) answer = data.response

diff --git a/src/services/apis/openai-api.mjs b/src/services/apis/openai-api.mjs
@@ -75,6 +75,7 @@ export async function generateAnswersWithGptCompletionApi(
         pushRecord(session, question, answer)
         console.debug('conversation history', { content: session.conversationRecords })
         port.postMessage({ answer: null, done: true, session: session })
+        return
       }
 
       answer += data.choices[0].text
@@ -170,6 +171,7 @@ export async function generateAnswersWithChatgptApiCompat(
         pushRecord(session, question, answer)
         console.debug('conversation history', { content: session.conversationRecords })
         port.postMessage({ answer: null, done: true, session: session })
+        return
       }
 
       const delta = data.choices[0]?.delta?.content