fix: when requests stop, input box sometimes isn't reset properly (#462)

This commit is contained in:
josc146
2023-07-22 14:55:26 +08:00
parent ad4f93e5de
commit 3bef0c0c98
5 changed files with 11 additions and 5 deletions
+1
View File
@@ -59,6 +59,7 @@ export async function generateAnswersWithAzureOpenaiApi(port, question, session)
},
async onStart() {},
async onEnd() {
port.postMessage({ done: true })
port.onMessage.removeListener(messageListener)
port.onDisconnect.removeListener(disconnectListener)
},
+2 -1
View File
@@ -123,7 +123,7 @@ export async function generateAnswersWithChatgptWebApi(port, question, session,
}),
onMessage(message) {
console.debug('sse message', message)
if (message === '[DONE]') {
if (message.trim() === '[DONE]') {
pushRecord(session, question, answer)
console.debug('conversation history', { content: session.conversationRecords })
port.postMessage({ answer: null, done: true, session: session })
@@ -148,6 +148,7 @@ export async function generateAnswersWithChatgptWebApi(port, question, session,
// sendModerations(accessToken, question, session.conversationId, session.messageId)
},
async onEnd() {
port.postMessage({ done: true })
port.onMessage.removeListener(messageListener)
port.onDisconnect.removeListener(disconnectListener)
},
+2 -1
View File
@@ -47,7 +47,7 @@ export async function generateAnswersWithCustomApi(port, question, session, apiK
}),
onMessage(message) {
console.debug('sse message', message)
if (message === '[DONE]') {
if (message.trim() === '[DONE]') {
pushRecord(session, question, answer)
console.debug('conversation history', { content: session.conversationRecords })
port.postMessage({ answer: null, done: true, session: session })
@@ -72,6 +72,7 @@ export async function generateAnswersWithCustomApi(port, question, session, apiK
},
async onStart() {},
async onEnd() {
port.postMessage({ done: true })
port.onMessage.removeListener(messageListener)
port.onDisconnect.removeListener(disconnectListener)
},
+4 -2
View File
@@ -55,7 +55,7 @@ export async function generateAnswersWithGptCompletionApi(
}),
onMessage(message) {
console.debug('sse message', message)
if (message === '[DONE]') {
if (message.trim() === '[DONE]') {
pushRecord(session, question, answer)
console.debug('conversation history', { content: session.conversationRecords })
port.postMessage({ answer: null, done: true, session: session })
@@ -73,6 +73,7 @@ export async function generateAnswersWithGptCompletionApi(
},
async onStart() {},
async onEnd() {
port.postMessage({ done: true })
port.onMessage.removeListener(messageListener)
port.onDisconnect.removeListener(disconnectListener)
},
@@ -122,7 +123,7 @@ export async function generateAnswersWithChatgptApi(port, question, session, api
}),
onMessage(message) {
console.debug('sse message', message)
if (message === '[DONE]') {
if (message.trim() === '[DONE]') {
pushRecord(session, question, answer)
console.debug('conversation history', { content: session.conversationRecords })
port.postMessage({ answer: null, done: true, session: session })
@@ -144,6 +145,7 @@ export async function generateAnswersWithChatgptApi(port, question, session, api
},
async onStart() {},
async onEnd() {
port.postMessage({ done: true })
port.onMessage.removeListener(messageListener)
port.onDisconnect.removeListener(disconnectListener)
},
+2 -1
View File
@@ -36,7 +36,7 @@ export async function generateAnswersWithWaylaidwandererApi(port, question, sess
}),
onMessage(message) {
console.debug('sse message', message)
if (message === '[DONE]') {
if (message.trim() === '[DONE]') {
pushRecord(session, question, answer)
console.debug('conversation history', { content: session.conversationRecords })
port.postMessage({ answer: null, done: true, session: session })
@@ -64,6 +64,7 @@ export async function generateAnswersWithWaylaidwandererApi(port, question, sess
},
async onStart() {},
async onEnd() {
port.postMessage({ done: true })
port.onMessage.removeListener(messageListener)
port.onDisconnect.removeListener(disconnectListener)
},