diff --git a/aider/models/model.py b/aider/models/model.py index 4b3397a50..3781d0770 100644 --- a/aider/models/model.py +++ b/aider/models/model.py @@ -1,7 +1,6 @@ -import importlib - saved_openai = None + class Model: name = None edit_format = None diff --git a/aider/models/openrouter.py b/aider/models/openrouter.py index 7250a521a..97802702f 100644 --- a/aider/models/openrouter.py +++ b/aider/models/openrouter.py @@ -18,6 +18,7 @@ class OpenRouterModel(Model): # TODO: figure out proper encodings for non openai models self.tokenizer = tiktoken.get_encoding("cl100k_base") + # TODO cache the model list data to speed up using multiple models available_models = openai.Model.list().data found = next((details for details in available_models if details.get('id') == name), None) @@ -30,6 +31,7 @@ class OpenRouterModel(Model): raise ValueError(f'invalid openrouter model: {name}') +# TODO run benchmarks and figure out which models support which edit-formats def edit_format_for_model(name): if any(str in name for str in ['gpt-4', 'claude-2']): return "diff"