kg_rag/config_loader.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

import yaml
import os

with open('config.yaml', 'r') as f:
    config_data = yaml.safe_load(f)
        
with open('system_prompts.yaml', 'r') as f:
    system_prompts = yaml.safe_load(f)
    
#if 'GPT_CONFIG_FILE' in config_data:
#    config_data['GPT_CONFIG_FILE'] = config_data['GPT_CONFIG_FILE'].replace('$HOME', os.environ['HOME'])

    
__all__ = [
    'config_data',
    'system_prompts'
]