diff --git a/pkg/openai/modelmgr.py b/pkg/openai/modelmgr.py index dd05cb3f..f16105cb 100644 --- a/pkg/openai/modelmgr.py +++ b/pkg/openai/modelmgr.py @@ -20,11 +20,14 @@ COMPLETION_MODELS = { CHAT_COMPLETION_MODELS = { 'gpt-3.5-turbo', - 'gpt-3.5-turbo-0301', + 'gpt-3.5-turbo-16k', + 'gpt-3.5-turbo-0613', + 'gpt-3.5-turbo-16k-0613', + # 'gpt-3.5-turbo-0301', 'gpt-4', - 'gpt-4-0314', + 'gpt-4-0613', 'gpt-4-32k', - 'gpt-4-32k-0314' + 'gpt-4-32k-0613' } EDIT_MODELS = {