mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-16 19:57:04 +08:00
35 lines
548 B
Python
35 lines
548 B
Python
|
# 提供与模型交互的抽象接口
|
||
|
|
|
||
|
|
COMPLETION_MODELS = {
|
||
|
|
'text-davinci-003'
|
||
|
|
}
|
||
|
|
|
||
|
|
EDIT_MODELS = {
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
IMAGE_MODELS = {
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
# ModelManager
|
||
|
|
# 由session包含
|
||
|
|
class ModelMgr(object):
|
||
|
|
|
||
|
|
using_completion_model = ""
|
||
|
|
using_edit_model = ""
|
||
|
|
using_image_model = ""
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def get_using_completion_model(self):
|
||
|
|
return self.using_completion_model
|
||
|
|
|
||
|
|
def get_using_edit_model(self):
|
||
|
|
return self.using_edit_model
|
||
|
|
|
||
|
|
def get_using_image_model(self):
|
||
|
|
return self.using_image_model
|