Upload 33 files

modules/index_func.py

@@ -16,7 +16,7 @@ def get_index_name(file_src):
 
     md5_hash = hashlib.md5()
     for file_path in file_paths:
-        with open(file_path, "rb", encoding="utf-8") as f:
+        with open(file_path, "rb") as f:
             while chunk := f.read(8192):
                 md5_hash.update(chunk)
 

modules/models/ChuanhuAgent.py

@@ -63,7 +63,7 @@ class ChuanhuAgent_Client(BaseLLMModel):
         self.index_summary = None
         self.index = None
         if "Pro" in self.model_name:
-            self.tools = load_tools(["google-search-results-json", "llm-math", "arxiv", "wikipedia", "wolfram-alpha"], llm=self.llm)
+            self.tools = load_tools(["serpapi", "google-search-results-json", "llm-math", "arxiv", "wikipedia", "wolfram-alpha"], llm=self.llm)
         else:
             self.tools = load_tools(["ddg-search", "llm-math", "arxiv", "wikipedia"], llm=self.llm)
             self.tools.append(

modules/models/base_model.py

@@ -375,7 +375,7 @@ class BaseLLMModel:
 
         status_text = "开始生成回答……"
         logging.info(
-            "输入为：" + colorama.Fore.BLUE + f"{inputs}" + colorama.Style.RESET_ALL
+             "用户" + f"{self.user_identifier}" + "的输入为：" + colorama.Fore.BLUE + f"{inputs}" + colorama.Style.RESET_ALL
         )
         if should_check_token_count:
             yield chatbot + [(inputs, "")], status_text
@@ -481,6 +481,7 @@ class BaseLLMModel:
         if len(self.history) > 0:
             inputs = self.history[-2]["content"]
             del self.history[-2:]
+        if len(self.all_token_counts) > 0:
             self.all_token_counts.pop()
         elif len(chatbot) > 0:
             inputs = chatbot[-1][0]