dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-29 08:44:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

cleaned up Model init

Browse source
This commit is contained in:
Paul Gauthier 2023-06-15 06:26:16 -07:00
parent bd91cb7fb7
commit 8fb4ab2be3
2 changed files with 6 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

14
aider/models.py
View file

@ -13,6 +13,11 @@ class Model:
self.max_context_tokens = tokens * 1024
if self.is_gpt4() or self.is_gpt35():
return
raise ValueError(f"Unsupported model: {name}")
def is_gpt4(self):
return self.name.startswith("gpt-4")
@ -26,12 +31,3 @@ class Model:
GPT4 = Model("gpt-4", 8)
GPT35 = Model("gpt-3.5-turbo")
GPT35_16k = Model("gpt-3.5-turbo-16k")
def get_model(name):
model = Model(name)
if model.is_gpt4() or model.is_gpt35():
return model
raise ValueError(f"Unsupported model: {name}")