AI-Coach / src /config.py
anhlehong
feat(config): add gemini model rotation list and default settings
709cede
raw
history blame contribute delete
893 Bytes
import os
from dotenv import load_dotenv
load_dotenv()
ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY", "")
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "")
DEFAULT_MODEL = os.getenv("DEFAULT_MODEL", "models/gemini-flash-latest")
LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO")
# List of active models for rotation - Priotizing stable production models
GEMINI_MODELS = [
"models/gemini-flash-latest",
"models/gemini-2.0-flash",
"models/gemini-pro-latest",
"models/gemini-flash-lite-latest",
"models/gemini-2.5-flash",
"models/gemma-3-1b-it",
"models/gemma-3-4b-it",
"models/gemma-3-12b-it",
"models/gemma-3-27b-it",
"models/gemma-3n-e4b-it",
"models/gemma-3n-e2b-it",
"models/gemma-4-26b-a4b-it",
"models/gemma-4-31b-it",
"models/gemini-2.5-flash-lite",
"models/gemini-3-flash-preview",
"models/gemini-3.1-flash-lite-preview",
]