Update app.py

app.py

@@ -51,16 +51,23 @@ def get_credit_summary(api_key):
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json"
     }
-    try:
-        response = requests.get(API_ENDPOINT, headers=headers)
-        response.raise_for_status()
-        data = response.json().get("data", {})
-        total_balance = data.get("totalBalance", 0)
-        logging.info(f"获取额度，API Key：{api_key}，当前额度: {total_balance}")
-        return {"total_balance": float(total_balance)}
-    except requests.exceptions.RequestException as e:
-        logging.error(f"获取额度信息失败，API Key：{api_key}，错误信息：{e}")
-        return None
+    max_retries = 5
+    retry_delay = 5
+    for attempt in range(max_retries):
+        try:
+            response = requests.get(API_ENDPOINT, headers=headers, timeout=5)
+            response.raise_for_status()
+            data = response.json().get("data", {})
+            total_balance = data.get("totalBalance", 0)
+            logging.info(f"API Key：{api_key}，当前额度: {total_balance}")
+            return {"total_balance": float(total_balance)}
+        except requests.exceptions.RequestException as e:
+            logging.error(f"获取额度信息失败，API Key：{api_key}，尝试次数：{attempt+1}/{max_retries}，错误信息：{e}")
+            if attempt < max_retries - 1:
+                time.sleep(retry_delay)
+            else:
+                logging.error(f"获取额度信息失败，API Key：{api_key}，所有重试次数均已失败")
+    return None
 
 FREE_MODEL_TEST_KEY = (
     "sk-bmjbjzleaqfgtqfzmcnsbagxrlohriadnxqrzfocbizaxukw"