huangquan-wuchang-repe-npc / list_models.py
ifmanzhang's picture
Add files using upload-large-folder tool
038aa4b verified
Raw
History Blame Contribute Delete
391 Bytes
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)