NLP_RAG / config.py
droushb's picture
Initial commit for RAG Question Answering system
dc75be1
raw
history blame
271 Bytes
import os
from dotenv import load_dotenv
load_dotenv()
CONFIG = {
"DATASET": "aalksii/ml-arxiv-papers",
"MAX_NUM_OF_RECORDS": 1000,
"TEXTAREA_HEIGHT": 200,
"CHUNK_SIZE": 200,
"OPENAI_ENGINE": "gpt-4o-mini",
"MAX_TOKENS": 500,
"TOP_DOCS": 3
}