Update chat_func.py

chat_func.py

@@ -197,13 +197,13 @@ def stream_predict(
                 yield get_return_value()
     return all_token_counts          
 
-def main():
+#def main():
     # 调用stream_predict函数获取结果...
-    with open("token_counts.txt", "a") as f:
-        for token_count in all_token_counts:
-            f.write(str(token_count) + "\n")
-if __name__ == '__main__':
-    main()
+#    with open("token_counts.txt", "a") as f:
+#        for token_count in all_token_counts:
+#            f.write(str(token_count) + "\n")
+#if __name__ == '__main__':
+#    main()
     
 def predict_all(
     openai_api_key,
@@ -465,7 +465,7 @@ def reduce_token_size(
 
 # 获取当前用户ID
 user_id = os.getuid()
-
+print(user_id)
 # 将获取到的信息写入txt文件
 #all_token_counts = 10