From 6a733f8e76219600c82302d04b7edf9146ba9beb Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Sun, 22 Sep 2024 16:55:58 -0700
Subject: [PATCH] refactor: Rename print_model_settings_as_yaml to
 get_model_settings_as_yaml

---
 aider/models.py | 29 +++++++++++++++--------------
 1 file changed, 15 insertions(+), 14 deletions(-)

diff --git a/aider/models.py b/aider/models.py
index 0cd2446d4..744515336 100644
--- a/aider/models.py
+++ b/aider/models.py
@@ -931,13 +931,27 @@ def print_matching_models(io, search):
         io.tool_output(f'No models match "{search}".')
 
 
+def get_model_settings_as_yaml():
+    import yaml
+
+    model_settings_list = []
+    for ms in MODEL_SETTINGS:
+        model_settings_dict = {
+            field.name: getattr(ms, field.name) for field in fields(ModelSettings)
+        }
+        model_settings_list.append(model_settings_dict)
+
+    return yaml.dump(model_settings_list, default_flow_style=False)
+
+
 def main():
     if len(sys.argv) < 2:
         print("Usage: python models.py <model_name> or python models.py --yaml")
         sys.exit(1)
 
     if sys.argv[1] == "--yaml":
-        print_model_settings_as_yaml()
+        yaml_string = get_model_settings_as_yaml()
+        print(yaml_string)
     else:
         model_name = sys.argv[1]
         matching_models = fuzzy_match_models(model_name)
@@ -950,18 +964,5 @@ def main():
             print(f"No matching models found for '{model_name}'.")
 
 
-def print_model_settings_as_yaml():
-    import yaml
-
-    model_settings_list = []
-    for ms in MODEL_SETTINGS:
-        model_settings_dict = {
-            field.name: getattr(ms, field.name) for field in fields(ModelSettings)
-        }
-        model_settings_list.append(model_settings_dict)
-
-    print(yaml.dump(model_settings_list, default_flow_style=False))
-
-
 if __name__ == "__main__":
     main()