feat: 更换模型时保持对话上下文

ChuanhuChatbot.py

@@ -70,7 +70,7 @@ with gr.Blocks(theme=small_and_beautiful_theme) as demo:
                                 uploadFileBtn = gr.UploadButton(
                                     interactive=True, label="", file_types=[".json"], elem_id="gr-history-upload-btn")
                                 historyRefreshBtn = gr.Button("", elem_id="gr-history-refresh-btn")
-                            
+
 
                     with gr.Row(elem_id="chuanhu-history-body"):
                         with gr.Column(scale=6, elem_id="history-select-wrap"):
@@ -372,11 +372,11 @@ with gr.Blocks(theme=small_and_beautiful_theme) as demo:
                             label=i18n("实时传输回答"), value=True, visible=ENABLE_STREAMING_OPTION, elem_classes="switch-checkbox"
                         )
                         name_chat_method = gr.Dropdown(
-                            label=i18n("对话命名方式"), 
-                            choices=HISTORY_NAME_METHODS, 
+                            label=i18n("对话命名方式"),
+                            choices=HISTORY_NAME_METHODS,
                             multiselect=False,
                             interactive=True,
-                            value=HISTORY_NAME_METHODS[chat_name_method_index], 
+                            value=HISTORY_NAME_METHODS[chat_name_method_index],
                         )
                         single_turn_checkbox = gr.Checkbox(label=i18n(
                             "单轮对话"), value=False, elem_classes="switch-checkbox", elem_id="gr-single-session-cb", visible=False)
@@ -638,12 +638,12 @@ with gr.Blocks(theme=small_and_beautiful_theme) as demo:
     keyTxt.submit(**get_usage_args)
     single_turn_checkbox.change(
         set_single_turn, [current_model, single_turn_checkbox], None)
-    model_select_dropdown.change(get_model, [model_select_dropdown, lora_select_dropdown, user_api_key, temperature_slider, top_p_slider, systemPromptTxt, user_name], [
+    model_select_dropdown.change(get_model, [model_select_dropdown, lora_select_dropdown, user_api_key, temperature_slider, top_p_slider, systemPromptTxt, user_name, current_model], [
                                  current_model, status_display, chatbot, lora_select_dropdown, user_api_key, keyTxt], show_progress=True, api_name="get_model")
     model_select_dropdown.change(toggle_like_btn_visibility, [model_select_dropdown], [
                                  like_dislike_area], show_progress=False)
     lora_select_dropdown.change(get_model, [model_select_dropdown, lora_select_dropdown, user_api_key, temperature_slider,
-                                top_p_slider, systemPromptTxt, user_name], [current_model, status_display, chatbot], show_progress=True)
+                                top_p_slider, systemPromptTxt, user_name, current_model], [current_model, status_display, chatbot], show_progress=True)
 
     # Template
     systemPromptTxt.change(set_system_prompt, [

modules/models/models.py

@@ -228,7 +228,7 @@ class OpenAIClient(BaseLLMModel):
         ret = super().set_key(new_access_key)
         self._refresh_header()
         return ret
-    
+
     def _single_query_at_once(self, history, temperature=1.0):
         timeout = TIMEOUT_ALL
         headers = {
@@ -255,7 +255,7 @@ class OpenAIClient(BaseLLMModel):
 
         return response
 
-    
+
     def auto_name_chat_history(self, name_chat_method, user_question, chatbot, user_name, single_turn_checkbox):
         if len(self.history) == 2 and not single_turn_checkbox:
             user_question = self.history[0]["content"]
@@ -601,7 +601,8 @@ def get_model(
     temperature=None,
     top_p=None,
     system_prompt=None,
-    user_name=""
+    user_name="",
+    original_model = None
 ) -> BaseLLMModel:
     msg = i18n("模型设置为了：") + f" {model_name}"
     model_type = ModelType.get_type(model_name)
@@ -671,7 +672,7 @@ def get_model(
             access_key = os.environ.get("GOOGLE_PALM_API_KEY", access_key)
             model = Google_PaLM_Client(model_name, access_key, user_name=user_name)
         elif model_type == ModelType.LangchainChat:
-            from .azure import Azure_OpenAI_Client
+            from .Azure import Azure_OpenAI_Client
             model = Azure_OpenAI_Client(model_name, user_name=user_name)
         elif model_type == ModelType.Midjourney:
             from .midjourney import Midjourney_Client
@@ -688,6 +689,8 @@ def get_model(
         traceback.print_exc()
         msg = f"{STANDARD_ERROR_MSG}: {e}"
     presudo_key = hide_middle_chars(access_key)
+    if original_model is not None:
+        model.history = original_model.history
     if dont_change_lora_selector:
         return model, msg, chatbot, gr.update(), access_key, presudo_key
     else: