修复system prompt不可用的问题

ChuanhuChatbot.py

@@ -292,6 +292,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     keyTxt.submit(**get_usage_args)
 
     # Template
+    systemPromptTxt.change(current_model.value.set_system_prompt, [systemPromptTxt], None)
     templateRefreshBtn.click(get_template_names, None, [templateFileSelectDropdown])
     templateFileSelectDropdown.change(
         load_template,

modules/base_model.py

@@ -46,7 +46,6 @@ class BaseLLMModel:
         self.all_token_counts = []
         self.model_name = model_name
         self.model_type = ModelType.get_type(model_name)
-        self.api_key = None
         self.token_upper_limit = MODEL_TOKEN_LIMIT[model_name]
         self.max_generation_token = max_generation_token if max_generation_token is not None else self.token_upper_limit
         self.interrupted = False
@@ -347,6 +346,9 @@ class BaseLLMModel:
     def set_top_p(self, new_top_p):
         self.top_p = new_top_p
 
+    def set_system_prompt(self, new_system_prompt):
+        self.system_prompt = new_system_prompt
+
     def reset(self):
         self.history = []
         self.all_token_counts = []

modules/models.py

@@ -63,9 +63,6 @@ class OpenAIClient(BaseLLMModel):
             return input_token_count + system_prompt_token_count
         return input_token_count
 
-    def set_system_prompt(self, new_system_prompt):
-        self.system_prompt = new_system_prompt
-
     def billing_info(self):
         try:
             curr_time = datetime.datetime.now()

modules/utils.py

@@ -229,7 +229,7 @@ def get_template_names(plain=False):
 
 
 def get_template_content(templates, selection, original_system_prompt):
-    logging.info(f"应用模板中，选择为{selection}，原始系统提示为{original_system_prompt}")
+    logging.debug(f"应用模板中，选择为{selection}，原始系统提示为{original_system_prompt}")
     try:
         return templates[selection]
     except: