mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-31 01:35:00 +00:00
refac
This commit is contained in:
parent
4d3fc3de7e
commit
43ba638ac4
2 changed files with 4 additions and 4 deletions
|
@ -2,6 +2,8 @@ import re
|
|||
|
||||
|
||||
class Model:
|
||||
always_available = False
|
||||
|
||||
def __init__(self, name, tokens=None):
|
||||
self.name = name
|
||||
if tokens is None:
|
||||
|
@ -19,6 +21,7 @@ class Model:
|
|||
|
||||
if self.is_gpt35():
|
||||
self.edit_format = "whole"
|
||||
self.always_available = True
|
||||
return
|
||||
|
||||
raise ValueError(f"Unsupported model: {name}")
|
||||
|
@ -29,9 +32,6 @@ class Model:
|
|||
def is_gpt35(self):
|
||||
return self.name.startswith("gpt-3.5-turbo")
|
||||
|
||||
def is_always_available(self):
|
||||
return self.is_gpt35()
|
||||
|
||||
|
||||
GPT4 = Model("gpt-4", 8)
|
||||
GPT35 = Model("gpt-3.5-turbo")
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue