update

app_util.py

@@ -34,17 +34,16 @@ def generate_query(chatbot, history):
         yield query, chatbot, history
 
 
-def generate_response(query, chatbot, history):
+def generate_response(chatbot, history, user_input=None):
     """
     auto-mode：query is None
     manual-mode：query 是用户输入
-    :param query:
     :param chatbot:
     :param history:
     :return:
     """
-    if query and history[-1]["role"] != "user":
-        history.append({"role": "user", "content": query})
+    if user_input and history[-1]["role"] != "user":
+        history.append({"role": "user", "content": user_input})
     query = history[-1]["content"]
 
     if history[-1]["role"] != "user":
@@ -64,7 +63,7 @@ def generate_response(query, chatbot, history):
 
 def generate(chatbot, history):
     if history[-1]["role"] == "assistant":
-        return generate_query(chatbot, history)
+        return generate_response(chatbot, history)
     elif history[-1]["role"] in ["user", "system"]:
         return generate_query(chatbot, history)
     else: