mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 11:29:39 +08:00
修复模型切换角色改变引起的BUG
This commit is contained in:
@@ -17,6 +17,7 @@ IMAGE_MODELS = {
|
||||
|
||||
}
|
||||
|
||||
|
||||
class Model():
|
||||
|
||||
can_chat = False
|
||||
@@ -45,8 +46,10 @@ class Model():
|
||||
|
||||
def get_response(self):
|
||||
return self.ret
|
||||
|
||||
|
||||
class ChatCompletionModel(Model):
|
||||
"""ChatCompletion接口实现"""
|
||||
Chat_role = ['system', 'user', 'assistant']
|
||||
def __init__(self, model_name, user_name):
|
||||
request_fun = openai.ChatCompletion.create
|
||||
@@ -68,8 +71,10 @@ 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)
|
||||
@@ -90,6 +95,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:
|
||||
|
||||
Reference in New Issue
Block a user