yangtb24 commited on
Commit
64c7900
·
verified ·
1 Parent(s): 2fff224

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -66,14 +66,11 @@ def get_credit_summary(api_key):
66
  return {"total_balance": float(total_balance)}
67
  except requests.exceptions.Timeout as e:
68
  logging.error(f"获取额度信息失败,API Key:{api_key},尝试次数:{attempt+1}/{max_retries},错误信息:{e} (Timeout)")
69
- if attempt < max_retries - 1:
70
- pass
71
- else:
72
  logging.error(f"获取额度信息失败,API Key:{api_key},所有重试次数均已失败 (Timeout)")
73
  except requests.exceptions.RequestException as e:
74
  logging.error(f"获取额度信息失败,API Key:{api_key},错误信息:{e}")
75
  return None
76
- return None
77
 
78
  FREE_MODEL_TEST_KEY = (
79
  "sk-bmjbjzleaqfgtqfzmcnsbagxrlohriadnxqrzfocbizaxukw"
@@ -415,8 +412,8 @@ def key_is_valid(key, request_type):
415
  def check_authorization(request):
416
  authorization_key = os.environ.get("AUTHORIZATION_KEY")
417
  if not authorization_key:
418
- logging.warning("环境变量 AUTHORIZATION_KEY 未设置,请设置后重试。")
419
- return False
420
 
421
  auth_header = request.headers.get('Authorization')
422
  if not auth_header:
 
66
  return {"total_balance": float(total_balance)}
67
  except requests.exceptions.Timeout as e:
68
  logging.error(f"获取额度信息失败,API Key:{api_key},尝试次数:{attempt+1}/{max_retries},错误信息:{e} (Timeout)")
69
+ if attempt >= max_retries - 1:
 
 
70
  logging.error(f"获取额度信息失败,API Key:{api_key},所有重试次数均已失败 (Timeout)")
71
  except requests.exceptions.RequestException as e:
72
  logging.error(f"获取额度信息失败,API Key:{api_key},错误信息:{e}")
73
  return None
 
74
 
75
  FREE_MODEL_TEST_KEY = (
76
  "sk-bmjbjzleaqfgtqfzmcnsbagxrlohriadnxqrzfocbizaxukw"
 
412
  def check_authorization(request):
413
  authorization_key = os.environ.get("AUTHORIZATION_KEY")
414
  if not authorization_key:
415
+ logging.warning("环境变量 AUTHORIZATION_KEY 未设置,此时无需鉴权即可使用,建议进行设置后再使用。")
416
+ return True
417
 
418
  auth_header = request.headers.get('Authorization')
419
  if not auth_header: