xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-07-22 12:26:08 +00:00
Code Issues 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
Download Patch File Download Diff File Expand all files Collapse all files
pkg/openai/modelmgr.py
+6
View File
@@ -17,6 +17,7 @@ IMAGE_MODELS = {
} }
class Model(): class Model():
can_chat = False can_chat = False
@@ -45,8 +46,10 @@ class Model():
def get_response(self): def get_response(self):
return self.ret return self.ret
class ChatCompletionModel(Model): class ChatCompletionModel(Model):
"""ChatCompletion接口实现"""
def __init__(self, model_name, user_name): def __init__(self, model_name, user_name):
request_fun = openai.ChatCompletion.create request_fun = openai.ChatCompletion.create
self.can_chat = True self.can_chat = True
@@ -59,8 +62,10 @@ class ChatCompletionModel(Model):
def get_content(self): def get_content(self):
return self.message return self.message
class CompletionModel(Model): class CompletionModel(Model):
"""Completion接口实现"""
def __init__(self, model_name, user_name): def __init__(self, model_name, user_name):
request_fun = openai.Completion.create request_fun = openai.Completion.create
super().__init__(model_name, user_name, request_fun) super().__init__(model_name, user_name, request_fun)
@@ -82,6 +87,7 @@ class CompletionModel(Model):
def get_text(self): def get_text(self):
return self.message return self.message
def OpenaiModel(model_name:str, user_name='user'): def OpenaiModel(model_name:str, user_name='user'):
if model_name in CHAT_COMPLETION_MODELS: if model_name in CHAT_COMPLETION_MODELS: