diff --git a/config-template.py b/config-template.py index 401526ee..64e67ae3 100644 --- a/config-template.py +++ b/config-template.py @@ -204,11 +204,13 @@ auto_reset = True # 现已支持的模型有: # # 'gpt-4' -# 'gpt-4-0314' +# 'gpt-4-0613' # 'gpt-4-32k' -# 'gpt-4-32k-0314' +# 'gpt-4-32k-0613' # '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' # 'text-davinci-003' # 'text-davinci-002' # 'code-davinci-002' 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 = {