improve delta process (#657)

This commit is contained in:
josc146
2024-03-23 11:27:50 +08:00
parent dba68e7eee
commit d49280c89a
2 changed files with 22 additions and 11 deletions
+12 -6
View File
@@ -62,12 +62,18 @@ export async function generateAnswersWithCustomApi(port, question, session, apiK
} }
if (data.response) answer = data.response if (data.response) answer = data.response
else else {
answer += const delta = data.choices[0]?.delta?.content
data.choices[0]?.delta?.content || const content = data.choices[0]?.message?.content
data.choices[0]?.message?.content || const text = data.choices[0]?.text
data.choices[0]?.text || if (delta !== undefined) {
'' answer += delta
} else if (content) {
answer = content
} else if (text) {
answer += text
}
}
port.postMessage({ answer: answer, done: false, session: null }) port.postMessage({ answer: answer, done: false, session: null })
}, },
async onStart() {}, async onStart() {},
+10 -5
View File
@@ -154,11 +154,16 @@ export async function generateAnswersWithChatgptApiCompat(
console.debug('json error', error) console.debug('json error', error)
return return
} }
answer += const delta = data.choices[0]?.delta?.content
data.choices[0]?.delta?.content || const content = data.choices[0]?.message?.content
data.choices[0]?.message?.content || const text = data.choices[0]?.text
data.choices[0]?.text || if (delta !== undefined) {
'' answer += delta
} else if (content) {
answer = content
} else if (text) {
answer += text
}
port.postMessage({ answer: answer, done: false, session: null }) port.postMessage({ answer: answer, done: false, session: null })
}, },
async onStart() {}, async onStart() {},