瀏覽代碼

Fix bug in "remove last" button

oobabooga 3 年之前
父節點
當前提交
c5cc3a3075
共有 1 個文件被更改,包括 3 次插入2 次删除
  1. 3 2
      server.py

+ 3 - 2
server.py

@@ -334,10 +334,11 @@ if args.chat or args.cai_chat:
 
     def remove_last_message(name1, name2):
         history.pop()
+        _history = remove_example_dialogue_from_history(history)
         if args.cai_chat:
-            return generate_chat_html(history, name1, name2, character)
+            return generate_chat_html(_history, name1, name2, character)
         else:
-            return history
+            return _history
 
     def clear():
         global history