Update index.js

index.js

@@ -109,15 +109,14 @@ class TokenManager {
       if (wafToken) {
         Tokens[currentIndex].aws_waf_token = wafToken;
         await this.updateCacheTokens();
-        await this.updateRedisTokens();
-        console.log("✅ Capsolver方案更新token成功");
+        this.updateRedisTokens();
         currentIndex = (currentIndex + 1) % Tokens.length;
-        return true;
+        console.log("成功提取 aws-waf-token");
+      } else {
+        currentIndex = (currentIndex + 1) % Tokens.length;
+        await this.updateCacheTokens();
+        console.log("提取aws-waf-token失败");
       }
-      console.log("❌ Capsolver方案失败，直接切换账号");
-      currentIndex = (currentIndex + 1) % Tokens.length;
-      await this.updateCacheTokens();
-      return false;
     } else {
       const newCsrfToken = response.headers.get('anti-csrftoken-a2z');
       const cookies = response.headers.get('set-cookie');
@@ -236,18 +235,6 @@ class Utils {
         timeout: 30000
       });
 
-      // 修改后 (兼容所有版本)
-      await new Promise(resolve => setTimeout(resolve, 2000)); // 替代 waitForTimeout
-      try {
-        await page.waitForFunction(
-          () => !document.querySelector('.aws-waf-challenge:not([hidden])'),
-          { timeout: 15000 }
-        );
-        console.log("✅ 验证挑战已完成");
-      } catch (error) {
-        console.error("❌ 验证挑战超时:", error);
-      }
-
       // 获取最终token
       const awsWafToken = (await page.cookies()).find(
         cookie => cookie.name.toLowerCase() === 'aws-waf-token'
@@ -631,12 +618,10 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
           await tokenManager.updateTokens(response);
           throw new Error(`请求失败! status: ${response.statusText}，已刷新验证信息，请重新请求`);
         case 403:
-          const success = await tokenManager.updateTokens(response, true);
-          if(success) {
-            console.log("✅ 验证成功，使用新token重试");
-            return retryRequest(); // 需要实现重试逻辑
-          }
-          throw new Error("验证失败，请更换账号");
+          console.log("请求被阻止");
+          await tokenManager.updateTokens(response, true);//尝试获取waf，然后返回错误提示。
+          CONFIG.DEFAULT_HEADERS["User-Agent"] = await Utils.getRandomUserAgent();
+          throw new Error(`请求失败! status: ${response.statusText}，请重新请求，如果多次失败，请重新更换token`);
         default:
           throw new Error(`请求失败! status: ${response.status}`);
       }