21 lines
507 B
Python
21 lines
507 B
Python
import os
|
|
import google.generativeai as genai
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
API_KEY = os.environ.get("GEMINI_API_KEY")
|
|
|
|
if not API_KEY:
|
|
print("Error: GEMINI_API_KEY not found in .env")
|
|
exit(1)
|
|
|
|
genai.configure(api_key=API_KEY)
|
|
|
|
print(f"Checking available models for your API key...")
|
|
try:
|
|
for m in genai.list_models():
|
|
if 'generateContent' in m.supported_generation_methods:
|
|
print(f"- {m.name}")
|
|
except Exception as e:
|
|
print(f"Error listing models: {e}")
|