| 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) | |