From 5098be6172e1874ce73839544ebb36ef8f5cb4bc Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Mon, 6 Nov 2023 20:18:40 -0800 Subject: [PATCH] Add context windows and pricing for 1106 models --- aider/models/openai.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/aider/models/openai.py b/aider/models/openai.py index ca6d36994..75f2808c9 100644 --- a/aider/models/openai.py +++ b/aider/models/openai.py @@ -7,6 +7,8 @@ from .model import Model known_tokens = { "gpt-3.5-turbo": 4, "gpt-4": 8, + "gpt-4-1106-preview": 128, + "gpt-3.5-turbo-1106": 16, } @@ -41,6 +43,9 @@ class OpenAIModel(Model): elif tokens == 32: self.prompt_price = 0.06 self.completion_price = 0.12 + elif tokens == 128: + self.prompt_price = 0.01 + self.completion_price = 0.03 return @@ -48,7 +53,10 @@ class OpenAIModel(Model): self.edit_format = "whole" self.always_available = True - if tokens == 4: + if self.name == "gpt-3.5-turbo-1106": + self.prompt_price = 0.001 + self.completion_price = 0.002 + elif tokens == 4: self.prompt_price = 0.0015 self.completion_price = 0.002 elif tokens == 16: