次数校验

chatgpt-next-web/service.py

@@ -420,8 +420,8 @@ def validate():
 
     # 获取用户套餐信息
     package = get_user_package(user_id)
-    if not package:
-        return jsonify({'code': 400, 'message': 'User has not purchased any package'})
+    if not package & get_free_count(user_id) <= 0:
+        return jsonify({'code': 400, 'message': '聊天次数已用完，请购买套餐'})
 
     # 检查用户聊天次数是否超过限制
     if exceeded_chat_limit(user_id, package, model):