yangtb24 commited on
Commit
f144607
·
verified ·
1 Parent(s): 1211b2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -330,8 +330,16 @@ def handsome_models():
330
  """
331
  if not check_authorization(request):
332
  return jsonify({"error": "Unauthorized"}), 401
333
-
334
- return jsonify({"data": [{"id": model} for model in all_models]})
 
 
 
 
 
 
 
 
335
 
336
  @app.route('/handsome/v1/balance', methods=['GET'])
337
  def handsome_balance():
 
330
  """
331
  if not check_authorization(request):
332
  return jsonify({"error": "Unauthorized"}), 401
333
+
334
+ # 使用 valid_keys_global 中的第一个 key 获取完整模型列表
335
+ if valid_keys_global:
336
+ models = get_all_models(valid_keys_global[0])
337
+ if models:
338
+ return jsonify({"data": [{"id": model} for model in models]})
339
+ else:
340
+ return jsonify({"error": "获取模型列表失败"}), 500
341
+ else:
342
+ return jsonify({"error": "没有有效的key,无法获取完整模型列表。"}), 500
343
 
344
  @app.route('/handsome/v1/balance', methods=['GET'])
345
  def handsome_balance():