dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-03 19:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

adding todo notes

Browse source
This commit is contained in:
JV 2023-08-15 04:34:40 +12:00 • committed by Joshua Vial
parent 22f3349862
commit a39829c5f8
2 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
aider/models/model.py
View file

@ -1,7 +1,6 @@
import importlib
saved_openai = None saved_openai = None
class Model: class Model:
name = None name = None
edit_format = None edit_format = None

2
aider/models/openrouter.py
View file

@ -18,6 +18,7 @@ class OpenRouterModel(Model):
# TODO: figure out proper encodings for non openai models # TODO: figure out proper encodings for non openai models
self.tokenizer = tiktoken.get_encoding("cl100k_base") 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 available_models = openai.Model.list().data
found = next((details for details in available_models if details.get('id') == name), None) 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}') 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): def edit_format_for_model(name):
if any(str in name for str in ['gpt-4', 'claude-2']): if any(str in name for str in ['gpt-4', 'claude-2']):
return "diff" return "diff"