forge/backend/scripts/list_gemini_models.py

26 lines
822 B
Python

import asyncio
import os
import httpx
import json
from dotenv import load_dotenv
load_dotenv()
API_KEY = os.getenv("GOOGLE_API_KEY")
async def list_models():
url = f"https://generativelanguage.googleapis.com/v1beta/models?key={API_KEY}"
async with httpx.AsyncClient() as client:
response = await client.get(url)
if response.status_code == 200:
data = response.json()
models = data.get("models", [])
print(f"Found {len(models)} models:")
for m in models:
name = m.get("name")
display = m.get("displayName", "No Display Name")
print(f" - {name} ({display})")
else:
print(f"Error: {response.status_code} - {response.text}")
if __name__ == "__main__":
asyncio.run(list_models())