refactor: extract key-value pairs from a yml using a function

src/utils.py

@@ -0,0 +1,24 @@
+import yaml
+
+# Load in model configuration and check the required keys are present
+model_config_dir = "config/model_config.yml"
+config_keys = ["system_message", "model_id", "template"]
+
+def load_config_values(
+        model_config_dir=model_config_dir, 
+        config_keys=config_keys
+    ):
+    """
+    Function to extract user-input keys from config.yml
+    Returns `system_message`, `model_id`, `template`.
+    TO DO: fix how the function returns values so can be flexible (based on the order of the input `config_keys` list)
+
+    """
+    with open(model_config_dir, "r") as file:
+        model_config = yaml.safe_load(file)
+
+    for var in model_config.keys():
+        if var not in config_keys:
+            raise ValueError(f"`{var}` key missing from `{model_config_dir}`")
+        
+    return model_config["system_message"], model_config["model_id"], model_config["template"]