mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-29 16:54:59 +00:00
feat: Add model settings loading and registration to benchmark script
This commit is contained in:
parent
aa18b63c16
commit
0336a982ff
1 changed files with 16 additions and 0 deletions
|
@ -314,6 +314,22 @@ def main(
|
||||||
|
|
||||||
test_dnames = sorted(str(d.relative_to(original_dname)) for d in exercise_dirs)
|
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:
|
if keywords:
|
||||||
keywords = keywords.split(",")
|
keywords = keywords.split(",")
|
||||||
test_dnames = [dn for dn in test_dnames for keyword in keywords if keyword in dn]
|
test_dnames = [dn for dn in test_dnames for keyword in keywords if keyword in dn]
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue