重新加入在UI中设置API-Key的功能

ChuanhuChatbot.py

@@ -21,7 +21,6 @@ with open("assets/custom.css", "r", encoding="utf-8") as f:
 with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     user_name = gr.State("")
     promptTemplates = gr.State(load_template(get_template_names(plain=True)[0], mode=2))
-    user_api_key = gr.State(my_api_key)
     user_question = gr.State("")
     current_model = gr.State(get_model(MODELS[0], my_api_key)[0])
 
@@ -288,7 +287,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     two_column.change(update_doc_config, [two_column], None)
 
     # LLM Models
-    keyTxt.change(submit_key, keyTxt, [user_api_key, status_display]).then(**get_usage_args)
+    keyTxt.change(current_model.value.set_key, keyTxt, [status_display]).then(**get_usage_args)
     keyTxt.submit(**get_usage_args)
     model_select_dropdown.change(get_model, [model_select_dropdown, keyTxt, temperature_slider, top_p_slider, systemPromptTxt], [current_model, status_display], show_progress=True)
 

modules/base_model.py

@@ -52,6 +52,7 @@ class BaseLLMModel:
         self.temperature = temperature
         self.top_p = top_p
         self.system_prompt = system_prompt
+        self.api_key = None
 
 
     def get_answer_stream_iter(self):
@@ -349,6 +350,12 @@ class BaseLLMModel:
     def set_system_prompt(self, new_system_prompt):
         self.system_prompt = new_system_prompt
 
+    def set_key(self, new_access_key):
+        self.api_key = new_access_key.strip()
+        msg = f"API密钥更改为了{hide_middle_chars(self.api_key)}"
+        logging.info(msg)
+        return msg
+
     def reset(self):
         self.history = []
         self.all_token_counts = []