Rate Limits

import time
from collections import deque

class RateLimiter:
    """
    Limit API calls to max_calls per time_window_seconds.
    Blocks the agent if the limit is reached.
    """
    def __init__(self, max_calls: int, time_window_seconds: float):
        self.max_calls = max_calls
        self.window = time_window_seconds
        self.timestamps = deque()

    def wait_if_needed(self):
        """Call before each API request."""
        now = time.time()

        # Remove timestamps outside the current window
        while self.timestamps and self.timestamps[0] < now - self.window:
            self.timestamps.popleft()

        if len(self.timestamps) >= self.max_calls:
            # Too many calls — wait until oldest is outside window
            sleep_time = self.window - (now - self.timestamps[0])
            print(f"Rate limit reached. Waiting {sleep_time:.1f}s...")
            time.sleep(sleep_time)

        self.timestamps.append(time.time())

# Allow 10 calls per minute
limiter = RateLimiter(max_calls=10, time_window_seconds=60)

def safe_api_call(url: str) -> str:
    limiter.wait_if_needed()
    # make your API call here
    return f"Result from {url}"

# The agent uses this function and never hits rate limits
for i in range(3):
    result = safe_api_call(f"https://api.example.com/item/{i}")
    print(result)