withRetry.ts

const withRetry = (
  func: (...args: any[]) => Promise<any>,
  maxAttempts: number = 3,
  timeout: number = 1_000
) => {
  const sleep = (ms: number) => {
    return new Promise((resolve) => {
      return setTimeout(resolve, ms);
    });
  };

  const withRetryCapability = async (
    attempt: number,
    ...args: any
  ): Promise<any> => {
    try {
      return await func(...args);
    } catch (error) {
      if (error.code && error.code === 'rate_limit' && attempt <= maxAttempts) {
        await sleep(timeout * attempt);

        return await withRetryCapability(attempt + 1, ...args);
      }
      throw error;
    }
  };

  return (...args: any) => {
    return withRetryCapability(1, ...args);
  };
};