feat: 优化自动命名提示词

modules/models/base_model.py

@@ -682,7 +682,7 @@ class BaseLLMModel:
     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"]
-            filename = user_question[:16] + ".json"
+            filename = replace_special_symbols(user_question)[:16] + ".json"
             return self.rename_chat_history(filename, chatbot, user_name)
         else:
             return gr.update()

modules/models/models.py

@@ -264,18 +264,18 @@ class OpenAIClient(BaseLLMModel):
                 try:
                     history = [
                         { "role": "system", "content": SUMMARY_CHAT_SYSTEM_PROMPT},
-                        { "role": "user", "content": f"User: {user_question}\nAssistant: {ai_answer}"}
+                        { "role": "user", "content": f"Please write a title based on the following conversation:\n---\nUser: {user_question}\nAssistant: {ai_answer}"}
                     ]
                     response = self._single_query_at_once(history, temperature=0.0)
                     response = json.loads(response.text)
                     content = response["choices"][0]["message"]["content"]
-                    filename = content + ".json"
+                    filename = replace_special_symbols(content) + ".json"
                 except Exception as e:
                     logging.info(f"自动命名失败。{e}")
-                    filename = user_question[:16] + ".json"
+                    filename = replace_special_symbols(user_question)[:16] + ".json"
                 return self.rename_chat_history(filename, chatbot, user_name)
             elif name_chat_method == i18n("第一条提问"):
-                filename = user_question[:16] + ".json"
+                filename = replace_special_symbols(user_question)[:16] + ".json"
                 return self.rename_chat_history(filename, chatbot, user_name)
             else:
                 return gr.update()

modules/utils.py

@@ -717,3 +717,11 @@ def get_file_hash(file_src=None, file_paths=None):
 
 def myprint(**args):
     print(args)
+
+def replace_special_symbols(string, replace_string=" "):
+    # 定义正则表达式，匹配所有特殊符号
+    pattern = r'[!@#$%^&*()<>?/\|}{~:]'
+
+    new_string = re.sub(pattern, replace_string, string)
+
+    return new_string