move to gpt-3.5-turbo-1106

This commit is contained in:
Paul Gauthier 2023-12-05 10:16:33 -08:00
parent 6ebc142377
commit fb07b784f6
8 changed files with 27 additions and 23 deletions

View file

@ -4,6 +4,7 @@ from .openrouter import OpenRouterModel
GPT4 = Model.create("gpt-4")
GPT35 = Model.create("gpt-3.5-turbo")
GPT35_1106 = Model.create("gpt-3.5-turbo-1106")
GPT35_16k = Model.create("gpt-3.5-turbo-16k")
__all__ = [

View file

@ -33,11 +33,11 @@ class Model:
@staticmethod
def weak_model():
return Model.create("gpt-3.5-turbo")
return Model.create("gpt-3.5-turbo-1106")
@staticmethod
def commit_message_models():
return [Model.create("gpt-3.5-turbo"), Model.create("gpt-3.5-turbo-16k")]
return [Model.weak_model()]
def token_count(self, messages):
if not self.tokenizer:

View file

@ -55,6 +55,7 @@ class OpenAIModel(Model):
if self.is_gpt35():
self.edit_format = "whole"
self.always_available = True
self.send_undo_reply = False
if self.name == "gpt-3.5-turbo-1106":
self.prompt_price = 0.001

View file

@ -7,12 +7,8 @@ cached_model_details = None
class OpenRouterModel(Model):
def __init__(self, client, name):
if name == "gpt-4":
name = "openai/gpt-4"
elif name == "gpt-3.5-turbo":
name = "openai/gpt-3.5-turbo"
elif name == "gpt-3.5-turbo-16k":
name = "openai/gpt-3.5-turbo-16k"
if name.startswith("gpt-4") or name.startswith("gpt-3.5-turbo"):
name = "openai/" + name
self.name = name
self.edit_format = edit_format_for_model(name)