File size: 632 Bytes
c7256ee | 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 | 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() |