bugfix: 修复了重试和新建对话时可能会出错的问题

modules/models/base_model.py

@@ -520,7 +520,7 @@ class BaseLLMModel:
         reply_language="中文",
     ):
         logging.debug("重试中……")
-        if len(self.history) > 0:
+        if len(self.history) > 1:
             inputs = self.history[-2]["content"]
             del self.history[-2:]
             if len(self.all_token_counts) > 0:
@@ -530,6 +530,9 @@ class BaseLLMModel:
             if '<div class="user-message">' in inputs:
                 inputs = inputs.split('<div class="user-message">')[1]
                 inputs = inputs.split("</div>")[0]
+        elif len(self.history) == 1:
+            inputs = self.history[-1]["content"]
+            del self.history[-1]
         else:
             yield chatbot, f"{STANDARD_ERROR_MSG}上下文是空的"
             return

modules/models/midjourney.py

@@ -221,7 +221,7 @@ class Midjourney_Client(XMChat):
     def reset(self):
         self.image_bytes = None
         self.image_path = None
-        return [], "已重置"
+        return super().reset()
 
     def get_answer_at_once(self):
         content = self.history[-1]['content']
@@ -367,7 +367,7 @@ UPSCALE - 确认后放大图片，第一个数值为需要放大的图片（1~4
     请使用SD进行UPSCALE
 VARIATION - 图片变体，第一个数值为需要放大的图片（1~4），第二参数为任务ID
     /mj VARIATION::1::123456789
-    
+
 【绘图参数】
 所有命令默认会带上参数--v 5.2
 其他参数参照 https://docs.midjourney.com/docs/parameter-list

modules/models/models.py

@@ -474,7 +474,7 @@ class XMChat(BaseLLMModel):
     def reset(self):
         self.session_id = str(uuid.uuid4())
         self.last_conv_id = None
-        return [], "已重置"
+        return super().reset()
 
     def image_to_base64(self, image_path):
         # 打开并加载图片