Update app.py
Browse files
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
|
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
|
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:
|