修复模型切换角色改变引起的BUG

pkg/openai/modelmgr.py

@@ -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接口实现"""
     Chat_role = ['system', 'user', 'assistant']
     def __init__(self, model_name, user_name):
         request_fun = openai.ChatCompletion.create
@@ -69,7 +72,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)
@@ -91,6 +96,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)