Skip to content

Commit d49280c

Browse files
committed
improve delta process (#657)
1 parent dba68e7 commit d49280c

File tree

2 files changed

+22
-11
lines changed

2 files changed

+22
-11
lines changed

src/services/apis/custom-api.mjs

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -62,12 +62,18 @@ export async function generateAnswersWithCustomApi(port, question, session, apiK
6262
}
6363

6464
if (data.response) answer = data.response
65-
else
66-
answer +=
67-
data.choices[0]?.delta?.content ||
68-
data.choices[0]?.message?.content ||
69-
data.choices[0]?.text ||
70-
''
65+
else {
66+
const delta = data.choices[0]?.delta?.content
67+
const content = data.choices[0]?.message?.content
68+
const text = data.choices[0]?.text
69+
if (delta !== undefined) {
70+
answer += delta
71+
} else if (content) {
72+
answer = content
73+
} else if (text) {
74+
answer += text
75+
}
76+
}
7177
port.postMessage({ answer: answer, done: false, session: null })
7278
},
7379
async onStart() {},

src/services/apis/openai-api.mjs

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -154,11 +154,16 @@ export async function generateAnswersWithChatgptApiCompat(
154154
console.debug('json error', error)
155155
return
156156
}
157-
answer +=
158-
data.choices[0]?.delta?.content ||
159-
data.choices[0]?.message?.content ||
160-
data.choices[0]?.text ||
161-
''
157+
const delta = data.choices[0]?.delta?.content
158+
const content = data.choices[0]?.message?.content
159+
const text = data.choices[0]?.text
160+
if (delta !== undefined) {
161+
answer += delta
162+
} else if (content) {
163+
answer = content
164+
} else if (text) {
165+
answer += text
166+
}
162167
port.postMessage({ answer: answer, done: false, session: null })
163168
},
164169
async onStart() {},

0 commit comments

Comments
 (0)