Update chat_func.py

chat_func.py

@@ -456,11 +456,16 @@ def reduce_token_size(
  logging.info(msg)
  logging.info("减少token数量完毕")
 
-# 获取当前用户ID和IP
+# 获取当前用户ID
 user_id = os.getuid()
 
 # 将获取到的信息写入txt文件
-all_token_counts = 10
-log_file = "user_info.txt"
-log_str = f"{user_id}\t{all_token_counts}\n"
-np.savetxt(log_file, [log_str], delimiter="", fmt="%s")
+# all_token_counts = 10
+
+def write_to_file(user_id, all_token_counts):
+ log_file = "user_info.txt"
+ log_str = f"{user_id}\t{all_token_counts}\n"
+ with open(log_file, 'a') as f:
+ f.write(log_str)
+
+write_to_file(user_id,all_token_counts)