sentimind / test_models.py
anggars's picture
Sync from GitHub Actions: e01f9df8b054a5718601dd513fd3516293f9eac8
0630ed4 verified
import google.generativeai as genai
import os
from dotenv import load_dotenv
load_dotenv()
api_key = os.getenv("GEMINI_API_KEY")
if not api_key:
print("No API Key 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(m.name)
except Exception as e:
print(f"Error: {e}")