Update chat_func.py

chat_func.py

@@ -453,4 +453,14 @@ def reduce_token_size(
  sum(token_count) if len(token_count) > 0 else 0,
  ), token_count
  logging.info(msg)
- logging.info("减少token数量完毕")
+ logging.info("减少token数量完毕")
+
+# 获取当前用户ID和IP
+user_id = os.getuid()
+user_ip = os.popen("hostname -I").read().split()[0]
+
+# 将获取到的信息写入txt文件
+log_file = "user_info.txt"
+with open(log_file, mode="a") as f:
+ log_str = f"{user_id}\t{user_ip}\t{token_counts}\n"
+ f.write(log_str)