diff --git a/benchmark/benchmark.py b/benchmark/benchmark.py index ca26d0ce9..f8267761a 100755 --- a/benchmark/benchmark.py +++ b/benchmark/benchmark.py @@ -314,6 +314,22 @@ def main( test_dnames = sorted(str(d.relative_to(original_dname)) for d in exercise_dirs) + resource_metadata = importlib_resources.files("aider.resources").joinpath("model-metadata.json") + model_metadata_files_loaded = models.register_litellm_models([resource_metadata]) + dump(model_metadata_files_loaded) + + if read_model_settings: + try: + files_loaded = models.register_models([read_model_settings]) + if verbose: + if files_loaded: + print(f"Loaded model settings from: {files_loaded[0]}") + else: + print(f"No model settings loaded from: {read_model_settings}") + except Exception as e: + print(f"Error loading model settings: {e}") + return 1 + if keywords: keywords = keywords.split(",") test_dnames = [dn for dn in test_dnames for keyword in keywords if keyword in dn]