From f422575a172a19ad4c6ae27218228aca36fa5582 Mon Sep 17 00:00:00 2001 From: Light <85721607+lightedcode@users.noreply.github.com> Date: Thu, 7 Sep 2023 20:44:21 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dchat=E6=8E=A5=E5=8F=A3?= =?UTF-8?q?=E6=8A=A5=E9=94=99=E6=97=B6=E4=BF=A1=E6=81=AF=E6=98=BE=E7=A4=BA?= =?UTF-8?q?KeyError:=20'answer'=E7=9A=84Bug=20(#1399)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Light <> --- webui_pages/dialogue/dialogue.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/webui_pages/dialogue/dialogue.py b/webui_pages/dialogue/dialogue.py index 25b885b3..afefe32a 100644 --- a/webui_pages/dialogue/dialogue.py +++ b/webui_pages/dialogue/dialogue.py @@ -153,9 +153,10 @@ def dialogue_page(api: ApiRequest): for d in api.knowledge_base_chat(prompt, selected_kb, kb_top_k, score_threshold, history, model=llm_model): if error_msg := check_error_msg(d): # check whether error occured st.error(error_msg) - text += d["answer"] - chat_box.update_msg(text, 0) - chat_box.update_msg("\n\n".join(d["docs"]), 1, streaming=False) + else: + text += d["answer"] + chat_box.update_msg(text, 0) + chat_box.update_msg("\n\n".join(d["docs"]), 1, streaming=False) chat_box.update_msg(text, 0, streaming=False) elif dialogue_mode == "搜索引擎问答": chat_box.ai_say([