fix(look-at): add timeout to sync model retry to prevent process hang

Closes #2087
This commit is contained in:
YeonGyu-Kim
2026-02-26 20:59:34 +09:00
parent c505989ad4
commit 1e060e9028
3 changed files with 149 additions and 15 deletions
+49
View File
@@ -0,0 +1,49 @@
export interface PromptTimeoutArgs {
signal?: AbortSignal
}
export interface PromptRetryOptions {
timeoutMs?: number
}
export const PROMPT_TIMEOUT_MS = 120000
export function createPromptTimeoutContext(args: PromptTimeoutArgs, timeoutMs: number): {
signal: AbortSignal
wasTimedOut: () => boolean
cleanup: () => void
} {
const timeoutController = new AbortController()
let timeoutID: ReturnType<typeof setTimeout> | null = null
let timedOut = false
const abortOnUpstreamSignal = (): void => {
timeoutController.abort(args.signal?.reason)
}
if (args.signal) {
if (args.signal.aborted) {
timeoutController.abort(args.signal.reason)
} else {
args.signal.addEventListener("abort", abortOnUpstreamSignal, { once: true })
}
}
timeoutID = setTimeout(() => {
timedOut = true
timeoutController.abort(new Error(`prompt timed out after ${timeoutMs}ms`))
}, timeoutMs)
return {
signal: timeoutController.signal,
wasTimedOut: () => timedOut,
cleanup: () => {
if (timeoutID !== null) {
clearTimeout(timeoutID)
}
if (args.signal) {
args.signal.removeEventListener("abort", abortOnUpstreamSignal)
}
},
}
}