import os
import json
import requests
from dotenv import load_dotenv

load_dotenv()
key = os.getenv("NVIDIA_API_KEY_RERANK")

headers = {"Authorization": f"Bearer {key}"}
try:
    r = requests.get("https://integrate.api.nvidia.com/v1/models", headers=headers)
    models = r.json()
    with open("nvidia_models.json", "w", encoding="utf-8") as f:
        json.dump(models, f, indent=2)
except Exception as e:
    print(e)