dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-01 10:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

refactor: Rename print_model_settings_as_yaml to get_model_settings_as_yaml

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-09-22 16:55:58 -07:00
parent 60082d0d16
commit 6a733f8e76
1 changed files with 15 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

29
aider/models.py
View file

@ -931,13 +931,27 @@ def print_matching_models(io, search):
io.tool_output(f'No models match "{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(): def main():
if len(sys.argv) < 2: if len(sys.argv) < 2:
print("Usage: python models.py <model_name> or python models.py --yaml") print("Usage: python models.py <model_name> or python models.py --yaml")
sys.exit(1) sys.exit(1)
if sys.argv[1] == "--yaml": if sys.argv[1] == "--yaml":
print_model_settings_as_yaml() yaml_string = get_model_settings_as_yaml()
print(yaml_string)
else: else:
model_name = sys.argv[1] model_name = sys.argv[1]
matching_models = fuzzy_match_models(model_name) matching_models = fuzzy_match_models(model_name)
@ -950,18 +964,5 @@ def main():
print(f"No matching models found for '{model_name}'.") 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__": if __name__ == "__main__":
main() main()