添加使用逻辑，每访问一次，可用次数减一，区分gpt-4、gpt3.5

redis/test_user_redis.py

@@ -171,6 +171,7 @@ def purchase():
 @app.route('/validate', methods=['POST'])
 def validate():
     token = parse_token(request)
+    model = request.json.get('model')
 
     # 验证令牌
     if not validate_token(token):
@@ -187,7 +188,7 @@ def validate():
         return jsonify({'code': 400, 'message': 'User has not purchased any package'})
 
     # 检查用户聊天次数是否超过限制
-    if exceeded_chat_limit(user_id, package):
+    if exceeded_chat_limit(user_id, package, model):
         return jsonify({'code': 400, 'message': 'Chat limit exceeded'})
 
     return jsonify({'code': 0, 'message': 'Chat limit not exceeded'})
@@ -195,29 +196,33 @@ def validate():
 
 @app.route('/chat/completions', methods=['POST'])
 def proxy_chat_completions():
-    # token = parse_token(request)
+    token = parse_token(request)
+    model = request.json.get('model')
 
-    # # 验证令牌
-    # if not validate_token(token):
-    #     return jsonify({'code': 401, 'message': 'Invalid token'})
+    # 验证令牌
+    if not validate_token(token):
+        return jsonify({'code': 401, 'message': 'Invalid token'})
 
-    # user_id = get_user_id_from_token(token)
+    user_id = get_user_id_from_token(token)
 
-    # if not user_id:
-    #     return jsonify({'code': 400, 'message': 'User not found'})
+    if not user_id:
+        return jsonify({'code': 400, 'message': 'User not found'})
 
-    # # 获取用户套餐信息
-    # package = get_user_package(user_id)
-    # if not package:
-    #     return jsonify({'code': 400, 'message': 'User has not purchased any package'})
+    # 获取用户套餐信息
+    package = get_user_package(user_id)
+    if not package:
+        return jsonify({'code': 400, 'message': 'User has not purchased any package'})
 
-    # # 检查用户聊天次数是否超过限制
-    # if exceeded_chat_limit(user_id, package):
-    #     return jsonify({'code': 400, 'message': 'Chat limit exceeded'})
+    # 检查用户聊天次数是否超过限制
+    if exceeded_chat_limit(user_id, package, model):
+        if model == 'gpt-3.5-turbo':
+            return jsonify({'code': 400, 'message': 'model3.5基础访问次数已用完'})
+        if model == 'gpt-4':
+            return jsonify({'code': 400, 'message': 'model4高级访问次数已用完'})
 
     # 获取请求数据
     data = request.get_json()
-    
+
     # 设置请求头部信息
     headers = {
         'Authorization': f'Bearer {MY_OPENAI_API_KEY}',
@@ -230,6 +235,9 @@ def proxy_chat_completions():
     # 获取 OpenAI API 的响应数据
     result = response.json()
 
+    user_package_key = f'user:{user_id}:package'
+    redis.hincrby(user_package_key, 'basic_chat_limit', -1)
+
     # 返回 OpenAI API 的响应给客户端
     return result, response.status_code
 
@@ -374,22 +382,22 @@ def get_package_expiration(user_id):
     expiration = redis.ttl(user_package_key)
     return expiration
 
-# 检查用户聊天次数是否超过限制
-
-
-def exceeded_chat_limit(user_id, package):
-    user_basic_chat_key = f'user:{user_id}:basic_chat'
-    user_advanced_chat_key = f'user:{user_id}:advanced_chat'
 
-    basic_chat_limit = int(package.get(b'basic_chat_limit', 0).decode('utf-8'))
-    advanced_chat_limit = int(package.get(
-        b'advanced_chat_limit', 0).decode('utf-8'))
-
-    if basic_chat_limit >= 0 and int(redis.get(user_basic_chat_key) or 0) >= basic_chat_limit:
-        return True
-
-    if advanced_chat_limit >= 0 and int(redis.get(user_advanced_chat_key) or 0) >= advanced_chat_limit:
-        return True
+# 检查用户聊天次数是否超过限制
+def exceeded_chat_limit(user_id, package, model):
+    if model == 'gpt-3.5-turbo':
+        user_basic_chat_key = f'user:{user_id}:basic_chat'
+        basic_chat_limit = int(package.get(
+            b'basic_chat_limit', 0).decode('utf-8'))
+        if basic_chat_limit >= 0 and int(redis.get(user_basic_chat_key) or 0) >= basic_chat_limit:
+            return True
+
+    if model == 'gpt-4':
+        user_advanced_chat_key = f'user:{user_id}:advanced_chat'
+        advanced_chat_limit = int(package.get(
+            b'advanced_chat_limit', 0).decode('utf-8'))
+        if advanced_chat_limit >= 0 and int(redis.get(user_advanced_chat_key) or 0) >= advanced_chat_limit:
+            return True
 
     return False