xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-08-16 07:20:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(modelmgr.py): 完善可选模型列表

Browse Source
This commit is contained in:
Rock Chin
2023-03-02 17:57:39 +08:00
parent 13e5d76a44
commit f7830b5e9d
1 changed files with 18 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/openai/modelmgr.py
+18 -4
View File
@@ -1,12 +1,26 @@
# 提供与模型交互的抽象接口 # 提供与模型交互的抽象接口
import openai, logging import openai, logging
CHAT_COMPLETION_MODELS = { COMPLETION_MODELS = {
'gpt-3.5-turbo' 'text-davinci-003',
'text-davinci-002',
'code-davinci-002',
'code-cushman-001',
'text-curie-001',
'text-babbage-001',
'text-ada-001',
} }
COMPLETION_MODELS = { CHAT_COMPLETION_MODELS = {
'text-davinci-003' 'gpt-3.5-turbo',
'gpt-3.5-turbo-0301',
'text-davinci-003',
'text-davinci-002',
'code-davinci-002',
'code-cushman-001',
'text-curie-001',
'text-babbage-001',
'text-ada-001',
} }
EDIT_MODELS = { EDIT_MODELS = {