Rate Limits

import time
import requests

def query_with_retry(kb_id, query, api_key, max_retries=3):
    url = f"https://api.ardie.ai/kb/{kb_id}/query"
    headers = {
        "Authorization": f"Bearer {api_key}",
        "Content-Type": "application/json"
    }
    
    for attempt in range(max_retries):
        response = requests.post(url, headers=headers, json={"query": query})
        
        if response.status_code == 200:
            return response.json()
        
        if response.status_code == 429:
            error = response.json()
            if error.get("error") == "query_cap_exceeded":
                raise Exception("Query cap exceeded. Upgrade your plan.")
            
            # Rate limited - wait and retry
            retry_after = error.get("retry_after", 2 ** attempt)
            time.sleep(retry_after)
            continue
        
        response.raise_for_status()
    
    raise Exception("Max retries exceeded")

async function queryWithRetry(kbId, query, apiKey, maxRetries = 3) {
  const url = `https://api.ardie.ai/kb/${kbId}/query`;
  
  for (let attempt = 0; attempt < maxRetries; attempt++) {
    const response = await fetch(url, {
      method: 'POST',
      headers: {
        'Authorization': `Bearer ${apiKey}`,
        'Content-Type': 'application/json'
      },
      body: JSON.stringify({ query })
    });
    
    if (response.ok) {
      return response.json();
    }
    
    if (response.status === 429) {
      const error = await response.json();
      if (error.error === 'query_cap_exceeded') {
        throw new Error('Query cap exceeded. Upgrade your plan.');
      }
      
      // Rate limited - wait and retry
      const retryAfter = error.retry_after || Math.pow(2, attempt);
      await new Promise(r => setTimeout(r, retryAfter * 1000));
      continue;
    }
    
    throw new Error(`Request failed: ${response.status}`);
  }
  
  throw new Error('Max retries exceeded');
}