config.py

import os

BASE_DIR = os.path.dirname(os.path.abspath(__file__))
DATA_DIR = os.path.join(BASE_DIR, "data")
MODEL_DIR = os.path.join(BASE_DIR, "veda_model")

os.makedirs(DATA_DIR, exist_ok=True)
os.makedirs(MODEL_DIR, exist_ok=True)

DATABASE_PATH = os.path.join(DATA_DIR, "conversations.db")

# Student model settings
VOCAB_SIZE = 8000
MAX_LENGTH = 512
D_MODEL = 256
NUM_HEADS = 8
NUM_LAYERS = 4
FF_DIM = 512
BATCH_SIZE = 4

# OpenRouter Teacher
OPENROUTER_API_KEY = os.environ.get("OPENROUTER_API_KEY", "").strip()
OPENROUTER_BASE_URL = "https://openrouter.ai/api/v1/chat/completions"

# ✅ Teacher model (confirmed working)
TEACHER_MODEL = "mistralai/mistral-7b-instruct:free"

# Optional fallback if that model ever fails
TEACHER_FALLBACK_MODELS = [
    "openrouter/auto"
]

TEACHER_TEMPERATURE = 0.7
TEACHER_MAX_TOKENS = 400