fix: return elements in the same order as input config_keys

src/utils.py

@@ -5,7 +5,7 @@ model_config_dir = "config/model_config.yml"
 config_keys = ["system_message", "model_id", "template"]
 
 def load_config_values(
-        model_config_dir=model_config_dir, 
+        model_config_dir=model_config_dir,
         config_keys=config_keys
     ):
     """
@@ -16,9 +16,9 @@ def load_config_values(
         model_config = yaml.safe_load(file)
 
     values_list = []
-    for var in model_config.keys():
-        if var not in config_keys:
+    for var in config_keys:
+        if var not in model_config.keys():
             raise ValueError(f"`{var}` key missing from `{model_config_dir}`")
         values_list.append(model_config[var])
 
-    return values_list
+    return values_list