Update index.js

index.js

@@ -109,14 +109,15 @@ class TokenManager {
       if (wafToken) {
         Tokens[currentIndex].aws_waf_token = wafToken;
         await this.updateCacheTokens();
-        this.updateRedisTokens();
+        await this.updateRedisTokens();
+        console.log("✅ Capsolver方案更新token成功");
         currentIndex = (currentIndex + 1) % Tokens.length;
-        console.log("成功提取 aws-waf-token");
-      } else {
-        currentIndex = (currentIndex + 1) % Tokens.length;
-        await this.updateCacheTokens();
-        console.log("提取aws-waf-token失败");
+        return true;
       }
+      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');
@@ -235,6 +236,10 @@ class Utils {
         timeout: 30000
       });
 
+      // 新增：等待验证完成
+      await page.waitForTimeout(1000); // 等待1秒确保验证完成
+      await page.waitForSelector('.aws-waf-challenge', { hidden: true, timeout: 15000 });
+
       // 获取最终token
       const awsWafToken = (await page.cookies()).find(
         cookie => cookie.name.toLowerCase() === 'aws-waf-token'
@@ -618,10 +623,12 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
           await tokenManager.updateTokens(response);
           throw new Error(`请求失败! status: ${response.statusText}，已刷新验证信息，请重新请求`);
         case 403:
-          console.log("请求被阻止");
-          await tokenManager.updateTokens(response, true);//尝试获取waf，然后返回错误提示。
-          CONFIG.DEFAULT_HEADERS["User-Agent"] = await Utils.getRandomUserAgent();
-          throw new Error(`请求失败! status: ${response.statusText}，请重新请求，如果多次失败，请重新更换token`);
+          const success = await tokenManager.updateTokens(response, true);
+          if(success) {
+            console.log("✅ 验证成功，使用新token重试");
+            return retryRequest(); // 需要实现重试逻辑
+          }
+          throw new Error("验证失败，请更换账号");
         default:
           throw new Error(`请求失败! status: ${response.status}`);
       }