jan-contract / debug_models.py
Amodit's picture
Cleanup and Fixes
66c7ada
raw
history blame contribute delete
509 Bytes
import os
import google.generativeai as genai
from dotenv import load_dotenv
load_dotenv()
api_key = os.getenv("GOOGLE_API_KEY")
if not api_key:
print("Error: GOOGLE_API_KEY not found in environment.")
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 listing models: {e}")