update

app_util.py

@@ -22,13 +22,13 @@ from models.cpp_qwen2 import bot
 def generate_user_message(chatbot, history):
     if history and history[-1]["role"] == "user":
         gr.Warning('You should generate assistant-response.')
-        yield None, chatbot, history
+        yield chatbot, history
     else:
         chatbot.append(None)
         streamer = bot.generate(history, stream=True)
         for user_content, user_tokens in streamer:
             chatbot[-1] = (user_content, None)
-            yield user_content, chatbot, history
+            yield chatbot, history
         user_tokens = bot.strip_stoptokens(user_tokens)
         history.append({"role": "user", "content": user_content, "tokens": user_tokens})
         yield chatbot, history
@@ -43,12 +43,12 @@ def generate_assistant_message(chatbot, history):
     user_content = history[-1]["content"]
     if history[-1]["role"] != "user":
         gr.Warning('You should generate or type user-input first.')
-        yield None, chatbot, history
+        yield chatbot, history
     else:
         streamer = bot.generate(history, stream=True)
         for assistant_content, assistant_tokens in streamer:
             chatbot[-1] = (user_content, assistant_content)
-            yield assistant_content, chatbot, history
+            yield chatbot, history
 
         assistant_tokens = bot.strip_stoptokens(assistant_tokens)
         history.append({"role": "assistant", "content": assistant_content, "tokens": assistant_tokens})
@@ -101,7 +101,7 @@ def undo_generate(chatbot, history):
     else:
         pass
     logger.info(f"after undo, {json.dumps(chatbot, ensure_ascii=False)}, {json.dumps(history, ensure_ascii=False)}")
-    return "", chatbot, history
+    return chatbot, history
 
 
 def reset_user_input():