From 73c194f304f827b55081b15524479f82a1b7d94c Mon Sep 17 00:00:00 2001 From: maszhongming Date: Tue, 16 Sep 2025 15:15:29 -0500 Subject: Initial commit --- kg_rag/config_loader.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 kg_rag/config_loader.py (limited to 'kg_rag/config_loader.py') 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' +] -- cgit v1.2.3