From ab860981d8c85a9c85d6fb08c488c26b7cbbdbb1 Mon Sep 17 00:00:00 2001 From: Reinier van der Leer Date: Mon, 29 Jan 2024 11:22:32 +0100 Subject: [PATCH] feat(agent/llm): Add support for `gpt-4-0125-preview` * Add `gpt-4-0125-preview` model to OpenAI model list * Add `gpt-4-turbo-preview` alias to OpenAI model list --- .../autogpt/autogpt/core/resource/model_providers/openai.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/autogpts/autogpt/autogpt/core/resource/model_providers/openai.py b/autogpts/autogpt/autogpt/core/resource/model_providers/openai.py index af26ba96..6d1eca5e 100644 --- a/autogpts/autogpt/autogpt/core/resource/model_providers/openai.py +++ b/autogpts/autogpt/autogpt/core/resource/model_providers/openai.py @@ -62,8 +62,10 @@ class OpenAIModelName(str, enum.Enum): GPT4_v2 = "gpt-4-0613" GPT4_v2_32k = "gpt-4-32k-0613" GPT4_v3 = "gpt-4-1106-preview" + GPT4_v4 = "gpt-4-0125-preview" GPT4_ROLLING = "gpt-4" GPT4_ROLLING_32k = "gpt-4-32k" + GPT4_TURBO = "gpt-4-turbo-preview" GPT4_VISION = "gpt-4-vision-preview" GPT4 = GPT4_ROLLING GPT4_32k = GPT4_ROLLING_32k @@ -130,7 +132,7 @@ OPEN_AI_CHAT_MODELS = { has_function_call_api=True, ), ChatModelInfo( - name=OpenAIModelName.GPT4_v3, + name=OpenAIModelName.GPT4_TURBO, service=ModelProviderService.CHAT, provider_name=ModelProviderName.OPENAI, prompt_token_cost=0.01 / 1000, @@ -149,6 +151,7 @@ chat_model_mapping = { OpenAIModelName.GPT4_v1_32k, OpenAIModelName.GPT4_v2_32k, ], + OpenAIModelName.GPT4_TURBO: [OpenAIModelName.GPT4_v3, OpenAIModelName.GPT4_v4], } for base, copies in chat_model_mapping.items(): for copy in copies: