From d720ca64490a567969747c2df54dba269df4399c Mon Sep 17 00:00:00 2001 From: DavidKorczynski Date: Tue, 22 Oct 2024 15:19:23 +0100 Subject: [PATCH] models: add gpt4-turbo (#672) Signed-off-by: David Korczynski --- llm_toolkit/models.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/llm_toolkit/models.py b/llm_toolkit/models.py index 2c7ae8819..069294682 100644 --- a/llm_toolkit/models.py +++ b/llm_toolkit/models.py @@ -290,23 +290,29 @@ def query_llm(self, prompt: prompts.Prompt, response_dir: str) -> None: class GPT4(GPT): - """OpenAI's GPTi-4 model.""" + """OpenAI's GPT-4 model.""" name = 'gpt-4' class GPT4o(GPT): - """OpenAI's GPTi-4o model.""" + """OpenAI's GPT-4o model.""" name = 'gpt-4o' class GPT4oMini(GPT): - """OpenAI's GPTi-4o-mini model.""" + """OpenAI's GPT-4o-mini model.""" name = 'gpt-4o-mini' +class GPT4Turbo(GPT): + """OpenAI's GPT-4 Turbo model.""" + + name = 'gpt-4-turbo' + + class AzureGPT(GPT): """Azure's GPT model."""