import argparse
import os
from google import genai

parser = argparse.ArgumentParser()
parser.add_argument("--api_key", default=None)
args = parser.parse_args()

key = args.api_key or os.environ.get("GEMINI_API_KEY", "")
if not key:
    print("Usage: python list_models.py --api_key YOUR_KEY")
    exit(1)

client = genai.Client(api_key=key)
for m in client.models.list():
    print(m.name)