NLP-RAG / config_loader.py
Qar-Raz's picture
hf-space: deploy branch without frontend/data/results
c7256ee
raw
history blame contribute delete
632 Bytes
import yaml
from pathlib import Path
class RAGConfig:
def __init__(self, config_path="config.yaml"):
with open(config_path, 'r') as f:
self.data = yaml.safe_load(f)
@property
def project(self): return self.data['project']
@property
def processing(self): return self.data['processing']
@property
def db(self): return self.data['vector_db']
@property
def retrieval(self): return self.data['retrieval']
@property
def gen(self): return self.data['generation']
@property
def model_list(self): return self.data['models']
cfg = RAGConfig()