#!/bin/bash
# check_quotas.sh - Modular script to check quotas
export PATH="$HOME/.local/bin:$PATH"

check_codex_quota() {
    if command -v codex &> /dev/null; then
        echo "[Codex Quota]"
        codex quota || echo "Failed to fetch Codex quota"
    fi
}

check_agy_quota() {
    if command -v agy &> /dev/null; then
        echo "[Antigravity Quota]"
        agy quota || echo "Failed to fetch Antigravity quota"
    fi
}

echo "--- Current Quotas ---"
check_codex_quota
check_agy_quota
echo "----------------------"