Update app.py

app.py

@@ -24,7 +24,7 @@ text_models = []
 invalid_keys_global = []
 valid_keys_global = []
 
-executor = concurrent.futures.ThreadPoolExecutor(max_workers=1000)
+executor = concurrent.futures.ThreadPoolExecutor(max_workers=10000)
 model_key_indices = {}
 
 request_timestamps = []
@@ -60,21 +60,21 @@ def get_credit_summary(api_key):
             exchange_rate = get_usd_to_cny_rate()
             if exchange_rate is not None:
                 total_balance_cny += usd_balance * exchange_rate
-                logging.info(f"获取美元兑人民币汇率成功，当前总额度(CNY): {total_balance_cny}")
+                logging.info(f"获取美元兑人民币汇率成功，API Key：{api_key}，当前总额度(CNY): {total_balance_cny}")
             else:
-                 logging.warning(f"获取美元兑人民币汇率失败，无法转换美元余额，API Key：{api_key}")
-                 total_balance_cny += usd_balance * 7.2
+                logging.warning(f"获取美元兑人民币汇率失败，无法转换美元余额，API Key：{api_key}")
+                total_balance_cny += usd_balance * 7.2
         except Exception as e:
-             logging.error(f"获取美元兑人民币汇率失败，API Key：{api_key}，错误信息：{e}")
-             total_balance_cny += usd_balance * 7.2 
+            logging.error(f"获取美元兑人民币汇率失败，API Key：{api_key}，错误信息：{e}")
+            total_balance_cny += usd_balance * 7.2 
 
         return {"total_balance": float(total_balance_cny)}
     except requests.exceptions.RequestException as e:
         logging.error(f"获取额度信息失败，API Key：{api_key}，错误信息：{e}")
         return None
     except Exception as e:
-         logging.error(f"处理额度信息失败，API Key：{api_key}，错误信息：{e}")
-         return None
+        logging.error(f"处理额度信息失败，API Key：{api_key}，错误信息：{e}")
+        return None
 
 def get_usd_to_cny_rate():
     try:
@@ -100,7 +100,7 @@ def load_keys():
     logging.info(f"加载的 keys：{unique_keys}")
 
     with concurrent.futures.ThreadPoolExecutor(
-        max_workers=1000
+        max_workers=10000
     ) as executor:
         future_to_key = {
             executor.submit(
@@ -191,54 +191,6 @@ def index():
         tpm = sum(token_counts)
 
     return jsonify({"rpm": rpm, "tpm": tpm})
-
-@app.route('/check_tokens', methods=['POST'])
-def check_tokens():
-    tokens = request.json.get('tokens', [])
-
-    with concurrent.futures.ThreadPoolExecutor(
-        max_workers=1000
-    ) as executor:
-        future_to_token = {
-            executor.submit(
-                process_key, token
-            ): token for token in tokens
-        }
-
-        results = []
-        for future in concurrent.futures.as_completed(future_to_token):
-            token = future_to_token[future]
-            try:
-                key_type = future.result()
-                credit_summary = get_credit_summary(token)
-                balance = (
-                    credit_summary.get("total_balance", 0)
-                    if credit_summary else 0
-                )
-                if key_type == "invalid":
-                    results.append(
-                        {
-                            "token": token,
-                            "type": "无效 KEY",
-                            "balance": balance,
-                            "message": "无法获取额度信息"
-                        }
-                    )
-                elif key_type == "valid":
-                    results.append(
-                        {
-                            "token": token,
-                            "type": "有效 KEY",
-                            "balance": balance,
-                            "message": "可以使用指定模型"
-                        }
-                    )
-            except Exception as exc:
-                logging.error(
-                    f"处理 Token {token} 生成异常: {exc}"
-                )
-
-    return jsonify(results)
           
 @app.route('/handsome/v1/models', methods=['GET'])
 def list_models():
@@ -251,22 +203,6 @@ def list_models():
             "object": "model",
             "created": 1678888888,
             "owned_by": "openai",
-            "permission": [
-                {
-                    "id": f"modelperm-{uuid.uuid4().hex}",
-                    "object": "model_permission",
-                    "created": 1678888888,
-                    "allow_create_engine": False,
-                    "allow_sampling": True,
-                    "allow_logprobs": True,
-                    "allow_search_indices": False,
-                    "allow_view": True,
-                    "allow_fine_tuning": False,
-                    "organization": "*",
-                    "group": None,
-                    "is_blocking": False
-                }
-            ],
             "root": "deepseek-chat",
             "parent": None
         },
@@ -275,22 +211,6 @@ def list_models():
             "object": "model",
             "created": 1678888889,
             "owned_by": "openai",
-            "permission": [
-                {
-                    "id": f"modelperm-{uuid.uuid4().hex}",
-                    "object": "model_permission",
-                    "created": 1678888889,
-                    "allow_create_engine": False,
-                    "allow_sampling": True,
-                    "allow_logprobs": True,
-                    "allow_search_indices": False,
-                    "allow_view": True,
-                    "allow_fine_tuning": False,
-                    "organization": "*",
-                    "group": None,
-                    "is_blocking": False
-                }
-            ],
             "root": "deepseek-reasoner",
             "parent": None
         }