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

refactor models to add gpt-4-32k-0613

Browse source
This commit is contained in:
Paul Gauthier 2023-06-14 17:09:53 -07:00
parent ce8069196c
commit 67ee800c55
1 changed files with 12 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

37
aider/models.py
View file

@ -1,36 +1,23 @@
class Model_GPT4_32k:
name = "gpt-4-32k"
max_context_tokens = 32 * 1024
class Model:
def __init__(self, name, max_context_tokens):
self.name = name
self.max_context_tokens = max_context_tokens * 1024
GPT4_32k = Model_GPT4_32k()
# 4
GPT4_32k = Model("gpt-4-32k", 32)
GPT4_32k_0613 = Model("gpt-4-32k-0613", 32)
GPT4 = Model("gpt-4", 8)
class Model_GPT4:
name = "gpt-4"
max_context_tokens = 8 * 1024
GPT4_models = [GPT4, GPT4_32k, GPT4_32k_0613]
# 3.5
GPT4 = Model_GPT4()
class Model_GPT35:
name = "gpt-3.5-turbo"
max_context_tokens = 4 * 1024
GPT35 = Model_GPT35()
class Model_GPT35_16k:
name = "gpt-3.5-turbo-16k"
max_context_tokens = 16 * 1024
GPT35_16k = Model_GPT35_16k()
GPT35 = Model("gpt-3.5-turbo", 4)
GPT35_16k = Model("gpt-3.5-turbo-16k", 16)
GPT35_models = [GPT35, GPT35_16k]
GPT4_models = [GPT4, GPT4_32k]
def get_model(name):