import time
import random
import requests

def call_with_retry(url, headers, payload, max_retries=4):
    delay = 1.0  # Base delay in seconds
    for attempt in range(max_retries):
        response = requests.post(url, headers=headers, json=payload)
        
        if response.status_code == 200:
            return response.json()
        
        if response.status_code in [429, 503]:
            # Jittered exponential backoff: delay * (2^attempt) + random jitter
            sleep_time = delay * (2 ** attempt) + random.uniform(0.1, 0.5)
            print(f"Rate limited or busy. Retrying in {sleep_time:.2f}s...")
            time.sleep(sleep_time)
            continue
            
        # Do not retry on client-side errors like 400 or 401
        response.raise_for_status()

    raise Exception("Max retry attempts reached")

async function fetchWithRetry<T>(
  url: string,
  options: RequestInit,
  maxRetries = 4
): Promise<T> {
  let delay = 1000;

  for (let attempt = 0; attempt < maxRetries; attempt++) {
    const res = await fetch(url, options);

    if (res.ok) {
      return (await res.json()) as T;
    }

    if (res.status === 429 || res.status === 503) {
      const jitter = Math.random() * 500;
      const sleepTime = delay * Math.pow(2, attempt) + jitter;
      console.warn(`Encountered ${res.status}. Retrying in ${Math.round(sleepTime)}ms...`);
      await new Promise((resolve) => setTimeout(resolve, sleepTime));
      continue;
    }

    throw new Error(`API Request failed with status ${res.status}`);
  }

  throw new Error('Max retries exceeded');
}