dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-31 17:55:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

all models to be registered with litellm

Browse source
This commit is contained in:
Krazer 2024-06-11 13:07:19 -05:00
parent d05f8e77b4
commit d91707f73e
3 changed files with 39 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

12
aider/models.py
View file

@ -426,6 +426,18 @@ class Model:
return res
def register_models(model_def_fnames):
for model_def_fname in model_def_fnames:
if not os.path.exists(model_def_fname):
continue
print(f"Registering model definition from {model_def_fname}")
try:
with open(model_def_fname, "r") as model_def_file:
model_def = json.load(model_def_file)
except json.JSONDecodeError as e:
print(f"Error opening/decoding model definition: {e}")
litellm.register_model(model_def)
def validate_variables(vars):
missing = []