From 60163fdf67b116ef560c2217595d5d0dd71fbe15 Mon Sep 17 00:00:00 2001 From: Johannes Reichhardt Date: Tue, 31 Mar 2026 20:24:11 -0500 Subject: [PATCH] update list models --- list_models.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 list_models.py diff --git a/list_models.py b/list_models.py new file mode 100644 index 0000000..7f85748 --- /dev/null +++ b/list_models.py @@ -0,0 +1,15 @@ +import os +import google.generativeai as genai +from dotenv import load_dotenv + +load_dotenv() +api_key = os.environ.get("AI_API_KEY") or os.environ.get("GOOGLE_API_KEY") +genai.configure(api_key=api_key) + +try: + print("Listing available models...") + for m in genai.list_models(): + if 'generateContent' in m.supported_generation_methods: + print(f"Model: {m.name}") +except Exception as e: + print(f"Error: {e}")