diff options
| author | maszhongming <mingz5@illinois.edu> | 2025-09-16 15:15:29 -0500 |
|---|---|---|
| committer | maszhongming <mingz5@illinois.edu> | 2025-09-16 15:15:29 -0500 |
| commit | 73c194f304f827b55081b15524479f82a1b7d94c (patch) | |
| tree | 5e8660e421915420892c5eca18f1ad680f80a861 /kg_rag/config_loader.py | |
Initial commit
Diffstat (limited to 'kg_rag/config_loader.py')
| -rw-r--r-- | kg_rag/config_loader.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/kg_rag/config_loader.py b/kg_rag/config_loader.py new file mode 100644 index 0000000..f3f5bd4 --- /dev/null +++ b/kg_rag/config_loader.py @@ -0,0 +1,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' +] |