File size: 1,173 Bytes
6e02dfb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | import os
from dotenv import load_dotenv
import requests
load_dotenv()
api_key = os.getenv("ABACUS_API_KEY")
def audit_abacus_models():
url = "https://routellm.abacus.ai/v1/models"
headers = {"Authorization": f"Bearer {api_key}"}
print("\n--- CONTACTING ABACUS HQ ---")
try:
response = requests.get(url, headers=headers)
if response.status_code != 200:
print(f"Error: {response.status_code} - {response.text}")
return
data = response.json()
model_list = data.get('data', [])
print(f"Found {len(model_list)} models. Filtering for GPT/Claude/Gemini/Sonnet...\n")
# Sort them so they are easier to read
ids = sorted([m.get('id') for m in model_list])
for model_id in ids:
# Flexible filter
if any(x in model_id.lower() for x in ['gpt', 'claude', 'gemini', 'sonnet']):
print(f"ID: {model_id}")
print("\n[Action] Copy the exact IDs above into your brain.py registry.")
except Exception as e:
print(f"Critical Failure: {e}")
if __name__ == "__main__":
audit_abacus_models() |