config.py

"""Configuration for Veda Programming Assistant"""

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")

VOCAB_SIZE = 8000
MAX_LENGTH = 512
D_MODEL = 256
NUM_HEADS = 8
NUM_LAYERS = 4
FF_DIM = 512
BATCH_SIZE = 4

DEFAULT_TEMPERATURE = 0.7
DEFAULT_MAX_TOKENS = 200