xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-02 03:55:55 +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

22
pkg/openai/modelmgr.py
View File

@@ -1,12 +1,26 @@
# 提供与模型交互的抽象接口
import openai, logging
CHAT_COMPLETION_MODELS = {
'gpt-3.5-turbo'
COMPLETION_MODELS = {
'text-davinci-003',
'text-davinci-002',
'code-davinci-002',
'code-cushman-001',
'text-curie-001',
'text-babbage-001',
'text-ada-001',
}
COMPLETION_MODELS = {
'text-davinci-003'
CHAT_COMPLETION_MODELS = {
'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 = {