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:
|