ai-workspace-services/accounts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix streaming abort timeout handling (#527)

Browse Source
This commit is contained in:
shenlan 2025-10-15 14:54:46 +08:00 committed by GitHub
parent 1dfb347a30
commit 04ca8b9763
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
ui/homepage/components/AskAIDialog.tsx
View File

@ -57,7 +57,12 @@ export function AskAIDialog({
) {
abortRef.current?.abort()
const controller = new AbortController()
const timer = setTimeout(() => controller.abort(), timeout)
let timer: NodeJS.Timeout | null = null
const scheduleAbort = () => {
if (timer) clearTimeout(timer)
timer = setTimeout(() => controller.abort(), timeout)
}
scheduleAbort()
abortRef.current = controller
try {
@ -92,6 +97,7 @@ export function AskAIDialog({
const { value, done } = await reader.read()
if (done) break
scheduleAbort()
buffer += decoder.decode(value, { stream: true })
const parts = buffer.split('\n\n')
buffer = parts.pop() || ''
@ -121,7 +127,7 @@ export function AskAIDialog({
update(answer, retrieved)
return { answer, retrieved }
} finally {
clearTimeout(timer)
if (timer) clearTimeout(timer)
if (abortRef.current === controller) abortRef.current = null
}
}