File size: 2,046 Bytes
568021d 5650eb2 568021d df36287 568021d 38df0bb 5650eb2 38df0bb 5650eb2 38df0bb 568021d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
import pRetry, { AbortError } from 'p-retry';
import pTimeout from 'p-timeout';
// Global reference to reset qwen client
let qwenClientResetter: (() => Promise<void>) | null = null;
export function setQwenClientResetter(resetFn: () => Promise<void>) {
qwenClientResetter = resetFn;
}
export async function withQwenTimeout<T>(
fn: () => Promise<T>,
{ totalTimeout = 70_000, retries = 2 } = {}
): Promise<T> {
return pRetry(
async () => {
try {
return await pTimeout(fn(), { milliseconds: totalTimeout });
} catch (error) {
// Check if this is a session/connection error that requires client reset
const errorMessage = (error as any)?.message || error?.toString() || '';
const isSessionError = errorMessage.includes('Session not found') ||
errorMessage.includes('network error') ||
errorMessage.includes('ERR_HTTP2_PROTOCOL_ERROR') ||
errorMessage.includes('Connection errored out');
if (isSessionError && qwenClientResetter) {
console.log('🔄 Qwen client session error detected, resetting client...');
try {
await qwenClientResetter();
console.log('✅ Qwen client reset successfully');
} catch (resetError) {
console.error('❌ Failed to reset qwen client:', resetError);
}
}
// Convert non-Error objects to proper Error objects for p-retry
if (error && typeof error === 'object' && !(error instanceof Error)) {
const errorMessage = (error as any).message || error.toString() || 'Network connection error';
throw new Error(errorMessage);
}
throw error;
}
},
{
retries,
onFailedAttempt: (error) => {
console.error(`qwen3 attempt #${error.attemptNumber} failed: ${error.message}`);
if (error.retriesLeft === 0) {
console.error('qwen3 max retries reached, giving up');
}
}
}
);
} |