github/sim
1
1
Fork 0
mirror of https://github.com/simstudioai/sim.git synced 2026-04-28 03:00:29 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add client retry logic

Browse Source
This commit is contained in:
Siddharth Ganesan
2026-04-04 17:13:38 -07:00
parent 0b3f3ed728
commit 0a41b8b287
1 changed files with 26 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
apps/sim/app/workspace/[workspaceId]/home/hooks/use-chat.ts
View File

@@ -108,7 +108,9 @@ const DEPLOY_TOOL_NAMES: Set<string> = new Set([
])
const RECONNECT_TAIL_ERROR =
'Live reconnect failed before the stream finished. The latest response may be incomplete.'
const RECOVERY_RETRY_DELAYS_MS = [250, 500, 1000, 2000] as const
const MAX_RECONNECT_ATTEMPTS = 10
const RECONNECT_BASE_DELAY_MS = 1000
const RECONNECT_MAX_DELAY_MS = 30_000
const logger = createLogger('useChat')
@@ -1278,11 +1280,29 @@ export function useChat(
setIsReconnecting(true)
try {
for (let attempt = 0; attempt <= RECOVERY_RETRY_DELAYS_MS.length; attempt++) {
for (let attempt = 0; attempt <= MAX_RECONNECT_ATTEMPTS; attempt++) {
if (abortController.signal.aborted || isStale()) {
return { attached: false, hadStreamError: false, aborted: true }
}
if (attempt > 0) {
const delayMs = Math.min(
RECONNECT_BASE_DELAY_MS * 2 ** (attempt - 1),
RECONNECT_MAX_DELAY_MS
)
logger.warn('Reconnect attempt', {
streamId,
attempt,
maxAttempts: MAX_RECONNECT_ATTEMPTS,
delayMs,
})
setIsReconnecting(true)
await waitForRetry(delayMs)
if (abortController.signal.aborted || isStale()) {
return { attached: false, hadStreamError: false, aborted: true }
}
}
if (!streamId && chatId) {
streamId = (await getActiveStreamIdForChat(chatId)) ?? undefined
}
@@ -1340,12 +1360,12 @@ export function useChat(
}
}
}
if (attempt < RECOVERY_RETRY_DELAYS_MS.length) {
await waitForRetry(RECOVERY_RETRY_DELAYS_MS[attempt] ?? 1000)
}
}
logger.error('All reconnect attempts exhausted', {
streamId,
maxAttempts: MAX_RECONNECT_ATTEMPTS,
})
setError(lastError)
return { attached: false, hadStreamError: true, aborted: false }
} finally {