加入了单轮对话功能

ChuanhuChatbot.py

@@ -83,10 +83,12 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
                     lora_select_dropdown = gr.Dropdown(
                         label="选择LoRA模型", choices=[], multiselect=False, interactive=True, visible=False
                     )
-                    use_streaming_checkbox = gr.Checkbox(
-                        label="实时传输回答", value=True, visible=ENABLE_STREAMING_OPTION
-                    )
-                    use_websearch_checkbox = gr.Checkbox(label="使用在线搜索", value=False)
+                    with gr.Row():
+                        use_streaming_checkbox = gr.Checkbox(
+                            label="实时传输回答", value=True, visible=ENABLE_STREAMING_OPTION
+                        )
+                        single_turn_checkbox = gr.Checkbox(label="单轮对话", value=False)
+                        use_websearch_checkbox = gr.Checkbox(label="使用在线搜索", value=False)
                     language_select_dropdown = gr.Dropdown(
                         label="选择回复语言（针对搜索&索引功能）",
                         choices=REPLY_LANGUAGES,
@@ -353,6 +355,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     # LLM Models
     keyTxt.change(current_model.value.set_key, keyTxt, [status_display]).then(**get_usage_args)
     keyTxt.submit(**get_usage_args)
+    single_turn_checkbox.change(current_model.value.set_single_turn, single_turn_checkbox, None)
     model_select_dropdown.change(current_model.value.get_model, [model_select_dropdown, lora_select_dropdown, keyTxt, temperature_slider, top_p_slider, systemPromptTxt], [status_display, lora_select_dropdown], show_progress=True)
     lora_select_dropdown.change(current_model.value.get_model, [model_select_dropdown, lora_select_dropdown, keyTxt, temperature_slider, top_p_slider, systemPromptTxt], [status_display], show_progress=True)
 

modules/base_model.py

@@ -71,6 +71,7 @@ class BaseLLMModel:
         self.system_prompt = system_prompt
         self.api_key = None
         self.need_api_key = False
+        self.single_turn = False
 
         self.temperature = temperature
         self.top_p = top_p
@@ -285,6 +286,9 @@ class BaseLLMModel:
             yield chatbot + [(inputs, "")], status_text
             return
 
+        if self.single_turn:
+            self.history = []
+            self.all_token_counts = []
         self.history.append(construct_user(inputs))
 
         try:
@@ -439,6 +443,9 @@ class BaseLLMModel:
         logging.info(msg)
         return msg
 
+    def set_single_turn(self, new_single_turn):
+        self.single_turn = new_single_turn
+
     def reset(self):
         self.history = []
         self.all_token_counts = []

modules/models.py

@@ -526,6 +526,9 @@ class ModelManager:
     def set_user_identifier(self, *args):
         self.model.set_user_identifier(*args)
 
+    def set_single_turn(self, *args):
+        self.model.set_single_turn(*args)
+