Austin/QChatGPT
1
0
Fork 0
mirror of https://github.com/RockChinQ/QChatGPT.git synced 2024-11-16 19:57:04 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

doc: 添加注释；完善格式

Browse Source
This commit is contained in:
Rock Chin 2023-03-02 16:41:03 +08:00
parent fd25d61b56
commit 623f094e5b
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pkg/openai/modelmgr.py
View File

@ -17,6 +17,7 @@ IMAGE_MODELS = {
}
class Model():
can_chat = False
@ -46,7 +47,9 @@ class Model():
def get_response(self):
return self.ret
class ChatCompletionModel(Model):
"""ChatCompletion接口实现"""
def __init__(self, model_name, user_name):
request_fun = openai.ChatCompletion.create
self.can_chat = True
@ -60,7 +63,9 @@ class ChatCompletionModel(Model):
def get_content(self):
return self.message
class CompletionModel(Model):
"""Completion接口实现"""
def __init__(self, model_name, user_name):
request_fun = openai.Completion.create
super().__init__(model_name, user_name, request_fun)
@ -83,6 +88,7 @@ class CompletionModel(Model):
def get_text(self):
return self.message
def OpenaiModel(model_name:str, user_name='user'):
if model_name in CHAT_COMPLETION_MODELS:
model = ChatCompletionModel(model_name, user_name)