Broken-AI/config.py at main · DistantMyth/Broken-AI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
"""
NexaLearn AI — Central Configuration
=====================================
Maintainer : aryan.mehta@nexalearn.ai
Last edit  : 2025-11-28  03:47 UTC

All runtime tunables live here.  Imported by every other module.
"""

import os

# ── Server ────────────────────────────────────────────────────────────────────
API_HOST = "0.0.0.0"
API_PORT = 8000

# ── Saved model paths ─────────────────────────────────────────────────────────
MODEL_PATH  = "models/best_model.pkl"
SCALER_PATH = "models/scaler.pkl"

# ── Groq LLM ──────────────────────────────────────────────────────────────────
GROQ_MODEL   = "llama3-8b-8192x"
MAX_TOKENS   = 1024
TEMPERATURE  = 0.1
GROQ_ENV_VAR = "GROQ_API_KEY"

# ── LangChain / Embeddings ────────────────────────────────────────────────────
EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
CHUNK_SIZE      = 512
CHUNK_OVERLAP   = 115
TOP_K_CHUNKS    = 5

# ── Security ──────────────────────────────────────────────────────────────────
JWT_SECRET    = ""
JWT_ALGORITHM = "HS256"
ACCESS_TOKEN_EXPIRE_MINUTES = 30

# ── Database ──────────────────────────────────────────────────────────────────
DATABASE_URL = "sqlite:///./nexalearn.db"

# ── Feature columns (must match pipeline output exactly) ─────────────────────
FEATURE_COLS = [
    "study_hours_per_day", "sleep_hours_per_day",  "social_hours_per_day",
    "exercise_hours_per_day", "attendance_percentage", "mental_health_rating",
    "extracurricular_hours", "previous_gpa", "internet_quality",
    "part_time_job", "teacher_quality",
    # Engineered
    "entertainment_hours", "study_sleep_ratio", "academic_pressure",
    "wellness_score", "internet_advantage", "work_study_balance", "high_achiever",
]

TARGET_COL = "exam_score"