Update app.py

app.py

@@ -479,30 +479,6 @@ def list_models():
         "data": detailed_models
     })
 
-def get_billing_info():
-    keys = valid_keys_global + unverified_keys_global
-    total_balance = 0
-
-    with concurrent.futures.ThreadPoolExecutor(
-        max_workers=10000
-    ) as executor:
-        futures = [
-            executor.submit(get_credit_summary, key) for key in keys
-        ]
-
-        for future in concurrent.futures.as_completed(futures):
-            try:
-                credit_summary = future.result()
-                if credit_summary:
-                    total_balance += credit_summary.get(
-                        "total_balance",
-                        0
-                    )
-            except Exception as exc:
-                logging.error(f"获取额度信息生成异常: {exc}")
-
-    return total_balance
-
 @app.route('/handsome/v1/dashboard/billing/usage', methods=['GET'])
 def billing_usage():
     if not check_authorization(request):
@@ -521,7 +497,23 @@ def billing_subscription():
     if not check_authorization(request):
         return jsonify({"error": "Unauthorized"}), 401
 
-    total_balance = get_billing_info()
+    keys = valid_keys_global + unverified_keys_global
+    total_balance = 0
+
+    with concurrent.futures.ThreadPoolExecutor(
+        max_workers=10000
+    ) as executor:
+        futures = [
+            executor.submit(get_credit_summary, key) for key in keys
+        ]
+
+        for future in concurrent.futures.as_completed(futures):
+            try:
+                credit_summary = future.result()
+                if credit_summary:
+                    total_balance += credit_summary.get("total_balance", 0)
+            except Exception as exc:
+                logging.error(f"获取额度信息生成异常: {exc}")
 
     return jsonify({
         "object": "billing_subscription",
@@ -552,15 +544,7 @@ def handsome_embeddings():
     api_key = select_key(request_type, model_name)
 
     if not api_key:
-        return jsonify(
-            {
-                "error": (
-                    "No available API key for this "
-                    "request type or all keys have "
-                    "reached their limits"
-                )
-            }
-        ), 429
+        return jsonify({"error": ("No available API key for this request type or all keys have reached their limits")}), 429
 
     headers = {
         "Authorization": f"Bearer {api_key}",