yangtb24 commited on
Commit
7f8aae6
·
verified ·
1 Parent(s): 188d1ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -88,6 +88,7 @@ def get_credit_summary(api_key):
88
  logging.info(f"获取额度,API Key:{api_key},当前额度: {total_balance}")
89
  return {"total_balance": float(total_balance)}
90
  except requests.exceptions.Timeout as e:
 
91
  logging.error(f"获取额度信息失败,API Key:{api_key},尝试次数:{attempt+1}/{max_retries},错误信息:{e} (Timeout)")
92
  if attempt >= max_retries - 1:
93
  logging.error(f"获取额度信息失败,API Key:{api_key},所有重试次数均已失败 (Timeout)")
@@ -131,7 +132,7 @@ def test_model_availability(api_key, model_name, model_type="chat"):
131
  json=payload,
132
  timeout=timeout
133
  )
134
- return response.status_code == 200 or response.status_code == 429
135
  except requests.exceptions.RequestException as e:
136
  logging.error(
137
  f"测试{model_type}模型 {model_name} 可用性失败,"
 
88
  logging.info(f"获取额度,API Key:{api_key},当前额度: {total_balance}")
89
  return {"total_balance": float(total_balance)}
90
  except requests.exceptions.Timeout as e:
91
+
92
  logging.error(f"获取额度信息失败,API Key:{api_key},尝试次数:{attempt+1}/{max_retries},错误信息:{e} (Timeout)")
93
  if attempt >= max_retries - 1:
94
  logging.error(f"获取额度信息失败,API Key:{api_key},所有重试次数均已失败 (Timeout)")
 
132
  json=payload,
133
  timeout=timeout
134
  )
135
+ return response.status_code == 200
136
  except requests.exceptions.RequestException as e:
137
  logging.error(
138
  f"测试{model_type}模型 {model_name} 可用性失败,"