15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-30 17:52:48 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: 修改流处理

This commit is contained in:
shaohuzhang1 2023-12-15 17:43:49 +08:00
parent 2c9ef25754
commit d3409d9845
1 changed files with 18 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -263,7 +263,7 @@ function chatMessage() {
vote_status: '-1'
})
inputValue.value = ''
applicationApi.postChatMessage(chartOpenId.value, problem_text).then(async (response) => {
applicationApi.postChatMessage(chartOpenId.value, problem_text).then((response) => {
const row = chatList.value.find((item) => item.id === id)
if (row) {
@ -271,27 +271,33 @@ function chatMessage() {
ChatManagement.write(id)
const reader = response.body.getReader()
/*eslint no-constant-condition: ["error", { "checkLoops": false }]*/
while (true) {
const { done, value } = await reader.read()
if (done) {
ChatManagement.close(id)
break
}
const write = ({ done, value }: { done: boolean; value: any }) => {
try {
if (done) {
ChatManagement.close(id)
return
}
const decoder = new TextDecoder('utf-8')
const str = decoder.decode(value, { stream: true })
if (str && str.startsWith('data:')) {
row.record_id = JSON?.parse(str.replace('data:', '')).id
const content = JSON?.parse(str.replace('data:', ''))?.content
if (content) {
ChatManagement.append(id, content)
const split = str.match(/data:.*}\n\n/g)
if (split) {
split.forEach((item_str) => {
row.record_id = JSON?.parse(item_str.replace('data:', '')).id
const content = JSON?.parse(item_str.replace('data:', ''))?.content
if (content) {
ChatManagement.append(id, content)
}
})
}
}
} catch (e) {
console.log(e)
// console
}
return reader.read().then(write)
}
reader.read().then(write)
}
})
}