Update app.py

app.py

@@ -264,18 +264,7 @@ def key_is_valid(key, request_type):
     if request_type == "invalid":
         return False
 
-    credit_summary = get_credit_summary(key)
-    if credit_summary is None:
-        return False
-
-    total_balance = credit_summary.get("total_balance", 0)
-
-    if request_type == "free":
-        return True
-    elif request_type == "paid" or request_type == "unverified":
-        return total_balance > 0
-    else:
-        return False
+    return total_balance > 0
 
 def check_authorization(request):
     """
@@ -320,17 +309,13 @@ def index():
 @app.route('/check_tokens', methods=['POST'])
 def check_tokens():
     tokens = request.json.get('tokens', [])
-    test_model = os.environ.get(
-        "TEST_MODEL",
-        "Pro/google/gemma-2-9b-it"
-    )
 
     with concurrent.futures.ThreadPoolExecutor(
         max_workers=1000
     ) as executor:
         future_to_token = {
             executor.submit(
-                process_key, token, test_model
+                process_key, token
             ): token for token in tokens
         }
 
@@ -353,24 +338,6 @@ def check_tokens():
                             "message": "无法获取额度信息"
                         }
                     )
-                elif key_type == "free":
-                    results.append(
-                        {
-                            "token": token,
-                            "type": "免费 KEY",
-                            "balance": balance,
-                            "message": "额度不足"
-                        }
-                    )
-                elif key_type == "unverified":
-                    results.append(
-                        {
-                            "token": token,
-                            "type": "未实名 KEY",
-                            "balance": balance,
-                            "message": "无法使用指定模型"
-                        }
-                    )
                 elif key_type == "valid":
                     results.append(
                         {