Update app.py

app.py

@@ -19,7 +19,7 @@ app = Flask(__name__)
 all_models = []
 free_models = []
 
-# 全局的 key 列表
+# 全局变量，用于存储各种类型的 KEY 列表
 invalid_keys_global = []
 free_keys_global = []
 unverified_keys_global = []
@@ -70,7 +70,7 @@ def test_model_availability(api_key, model_name):
         # 检查是否是429错误
         if response.status_code == 429:
             return True
-        
+      
         response.json()  # 尝试解析 JSON 响应
         return True
     except requests.exceptions.RequestException as e:
@@ -114,7 +114,7 @@ def load_keys():
         logging.info(f"免费 KEY：{free_keys}")
         logging.info(f"未实名 KEY：{unverified_keys}")
         logging.info(f"有效 KEY：{valid_keys}")
-        
+      
         # 更新全局的 key 列表
         global invalid_keys_global, free_keys_global, unverified_keys_global, valid_keys_global
         invalid_keys_global = invalid_keys
@@ -155,7 +155,7 @@ def refresh_models():
     刷新模型列表和免费模型列表。
     """
     global all_models, free_models
-    
+  
     # 使用 valid_keys_global 中的第一个 key 获取完整模型列表
     if valid_keys_global:
         all_models = get_all_models(valid_keys_global[0])
@@ -226,17 +226,6 @@ def check_authorization(request):
 
     return True
 
-def get_total_balance():
-    """
-    获取所有大于等于0的key的额度的和。
-    """
-    total_balance = 0
-    for key in free_keys_global + unverified_keys_global + valid_keys_global:
-        credit_summary = get_credit_summary(key)
-        if credit_summary is not None:
-            total_balance += credit_summary.get("total_balance", 0)
-    return total_balance
-
 # 创建一个后台调度器
 scheduler = BackgroundScheduler()
 
@@ -310,7 +299,7 @@ def handsome_chat_completions():
             stream=data.get("stream", False),
             timeout=60
         )
-        
+      
         # 检查是否是429错误
         if response.status_code == 429:
             return jsonify(response.json()), 429
@@ -324,25 +313,41 @@ def handsome_chat_completions():
         return jsonify({"error": str(e)}), 500
 
 @app.route('/handsome/v1/models', methods=['GET'])
-def handsome_models():
+def list_models():
     """
-    处理 /handsome/v1/models 路由的请求，添加鉴权，返回所有模型列表。
+    处理 /handsome/v1/models 路由的请求，返回模型列表。
     """
     if not check_authorization(request):
         return jsonify({"error": "Unauthorized"}), 401
 
-    return jsonify({"data": [{"id": model} for model in all_models]})
+    # 返回模型列表和免费模型列表
+    return jsonify({
+        "data": [{"id": model, "object": "model"} for model in all_models],
+        "free_models": free_models
+    })
 
-@app.route('/handsome/v1/dashboard/billing/subscription', methods=['GET'])
-def handsome_balance():
+@app.route('/handsome/v1/dashboard/billing', methods=['GET'])
+def billing_info():
     """
-    处理 /handsome/v1/balance 路由的请求，添加鉴权，返回所有可用KEY的额度总和。
+    处理 /handsome/v1/dashboard/billing 路由的请求，返回所有余额大于0的key的总和。
     """
     if not check_authorization(request):
         return jsonify({"error": "Unauthorized"}), 401
 
-    total_balance = get_total_balance()
-    return jsonify({"total_balance": total_balance})
+    total_balance = 0
+    # 遍历 valid_keys_global 和 unverified_keys_global 获取总余额
+    for key in valid_keys_global + unverified_keys_global:
+        credit_summary = get_credit_summary(key)
+        if credit_summary and credit_summary.get("total_balance", 0) > 0:
+            total_balance += credit_summary["total_balance"]
+    
+    return jsonify({
+        "total_granted": total_balance,  # 使用 total_granted 字段
+        "total_usage": 0,  # 暂时设置为0，因为没有使用量数据
+        "total_available": total_balance,
+        "has_payment_method": False,
+        "object": "billing_summary"
+    })
 
 if __name__ == '__main__':
     # 打印所有环境变量，方便调试