zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

perf: 优化流式输出，加快数据展示 #217

Browse Source 
* perf: 优化流式输出，加快数据展示 #217
This commit is contained in:
shaohuzhang1 2024-04-23 16:35:47 +08:00 committed by GitHub
parent 7be349c937
commit 73dfcb6fb0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ui/src/components/ai-chat/index.vue
View File

@ -364,15 +364,15 @@ const getWrite = (chat: any, reader: any, stream: boolean) => {
let str = decoder.decode(value, { stream: true }) let str = decoder.decode(value, { stream: true })
// 这里解释一下 start 因为数据流返回流并不是按照后端chunk返回 我们希望得到的chunk是data:{xxx}\n\n 但是它获取到的可能是 data:{ -> xxx}\n\n 总而言之就是 fetch不能保证每个chunk都说以data:开始 \n\n结束 // 这里解释一下 start 因为数据流返回流并不是按照后端chunk返回 我们希望得到的chunk是data:{xxx}\n\n 但是它获取到的可能是 data:{ -> xxx}\n\n 总而言之就是 fetch不能保证每个chunk都说以data:开始 \n\n结束
tempResult += str tempResult += str
if (tempResult.endsWith('\n\n')) { const split = tempResult.match(/data:.*}\n\n/g)
str = tempResult if (split) {
tempResult = '' str = split.join('')
tempResult = tempResult.replace(str, '')
} else { } else {
return reader.read().then(write_stream) return reader.read().then(write_stream)
} }
// 这里解释一下 end // 这里解释一下 end
if (str && str.startsWith('data:')) { if (str && str.startsWith('data:')) {
const split = str.match(/data:.*}\n\n/g)
if (split) { if (split) {
for (const index in split) { for (const index in split) {
const chunk = JSON?.parse(split[index].replace('data:', '')) const chunk = JSON?.parse(split[index].replace('data:', ''))