#!/bin/bash
# update_models_config.sh - Fetches latest models and quotas
export PATH="$HOME/.local/bin:$PATH"

DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
CONFIG_FILE="$DIR/../models_config.json"

echo "{" > "$CONFIG_FILE"
echo "  \"last_updated\": \"$(date -u +%Y-%m-%dT%H:%M:%SZ)\"," >> "$CONFIG_FILE"

# Logic to update config
echo "  \"providers\": {" >> "$CONFIG_FILE"

# Codex Models
echo "    \"codex\": {" >> "$CONFIG_FILE"
if command -v codex &> /dev/null; then
    echo "      \"models\": $(codex models --json 2>/dev/null || echo '[]')" >> "$CONFIG_FILE"
else
    echo "      \"models\": []" >> "$CONFIG_FILE"
fi
echo "    }," >> "$CONFIG_FILE"

# Antigravity Models
echo "    \"antigravity\": {" >> "$CONFIG_FILE"
if command -v agy &> /dev/null; then
    echo "      \"models\": $(agy models --json 2>/dev/null || echo '[]')" >> "$CONFIG_FILE"
else
    echo "      \"models\": []" >> "$CONFIG_FILE"
fi
echo "    }" >> "$CONFIG_FILE"

echo "  }" >> "$CONFIG_FILE"
echo "}" >> "$CONFIG_FILE"

echo "Models config updated at $CONFIG_FILE"