From 426cfef882e61fb47e5390867083bd585ea7956a Mon Sep 17 00:00:00 2001 From: Reinier van der Leer Date: Thu, 15 Jun 2023 18:19:03 +0200 Subject: [PATCH] Unpin OpenAI model versions in model info mapping --- autogpt/llm/providers/openai.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/autogpt/llm/providers/openai.py b/autogpt/llm/providers/openai.py index b4254cd1..eede1a7d 100644 --- a/autogpt/llm/providers/openai.py +++ b/autogpt/llm/providers/openai.py @@ -48,10 +48,16 @@ OPEN_AI_CHAT_MODELS = { ] } # Set aliases for rolling model IDs -OPEN_AI_CHAT_MODELS["gpt-3.5-turbo"] = OPEN_AI_CHAT_MODELS["gpt-3.5-turbo-0301"] -OPEN_AI_CHAT_MODELS["gpt-3.5-turbo-16k"] = OPEN_AI_CHAT_MODELS["gpt-3.5-turbo-16k-0613"] -OPEN_AI_CHAT_MODELS["gpt-4"] = OPEN_AI_CHAT_MODELS["gpt-4-0314"] -OPEN_AI_CHAT_MODELS["gpt-4-32k"] = OPEN_AI_CHAT_MODELS["gpt-4-32k-0314"] +chat_model_mapping = { + "gpt-3.5-turbo": "gpt-3.5-turbo-0301", + "gpt-3.5-turbo-16k": "gpt-3.5-turbo-16k-0613", + "gpt-4": "gpt-4-0314", + "gpt-4-32k": "gpt-4-32k-0314", +} +for alias, target in chat_model_mapping.items(): + alias_info = ChatModelInfo(**OPEN_AI_CHAT_MODELS[target].__dict__) + alias_info.name = alias + OPEN_AI_CHAT_MODELS[alias] = alias_info OPEN_AI_TEXT_MODELS = { info.name: info