26 lines
788 B
Python
26 lines
788 B
Python
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}")
|