customeragent-api / list_gemini_models.py
anasraza526's picture
Clean deploy to Hugging Face
ac90985
import os
import google.generativeai as genai
from dotenv import load_dotenv
dotenv_path = os.path.join(os.path.dirname(__file__), 'server', '.env')
load_dotenv(dotenv_path)
api_key = os.getenv("GEMINI_API_KEY")
if not api_key:
print("❌ API Key not found")
else:
genai.configure(api_key=api_key)
print("📡 Listing available models:")
try:
for m in genai.list_models():
if 'generateContent' in m.supported_generation_methods:
print(f" - {m.name}: {m.display_name}")
except Exception as e:
print(f"❌ Error listing models: {e}")