feat: return a list to allow for unpacking

src/utils.py

@@ -10,15 +10,15 @@ def load_config_values(
     ):
     """
     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)
-
+    Returns values as a list with length = length(config_keys)
     """
     with open(model_config_dir, "r") as file:
         model_config = yaml.safe_load(file)
 
+    values_list = []
     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"]
+        values_list.append(model_config[var])
+
+    return values_list