import os
import google.generativeai as genai
try:
    from backend.config_loader import get_config
except ImportError:
    import sys
    sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
    from backend.config_loader import get_config

config = get_config()
API_KEY = config.get("ai", {}).get("gemini_api_key")

if not API_KEY:
    print("Error: gemini_api_key not found in config/backend.yaml, config/secrets.yaml or GEMINI_API_KEY environment variable")
    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}")