Add retryNumber as a parameter of delay function (#337)

faros-ai · Nov 18, 2024 · d784146 · d784146
1 parent 221089d
commit d784146
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/axios.ts b/src/axios.ts
@@ -5,7 +5,10 @@ import {Logger} from 'pino';
 
 import {wrapApiError} from './errors';
 
-type DelayFunction = (error: AxiosError<unknown, any>) => number;
+type DelayFunction = (
+  error: AxiosError<unknown, any>,
+  retryNumber?: number
+) => number;
 
 export const DEFAULT_RETRIES = 3;
 export const DEFAULT_RETRY_DELAY = 1000;
@@ -65,7 +68,7 @@ export function makeAxiosInstanceWithRetry(
       if (typeof delay === 'number') {
         return retryNumber * delay;
       }
-      return delay(error);
+      return delay(error, retryNumber);
 
     },
     shouldResetTimeout: true,