Update index.js

index.js

@@ -107,19 +107,17 @@ class TokenManager {
     if (isWaf) {
       var wafToken = await Utils.extractWaf();
       if (wafToken) {
-        // 更新当前token的aws_waf_token
         Tokens[currentIndex].aws_waf_token = wafToken;
-        // 立即更新缓存headers
         await this.updateCacheTokens();
-        // 更新Redis存储
         await this.updateRedisTokens();
-        console.log("✅ 成功更新aws-waf-token到缓存");
+        console.log("✅ Capsolver方案更新token成功");
         currentIndex = (currentIndex + 1) % Tokens.length;
-      } else {
-        console.log("❌ 提取aws-waf-token失败，切换账号");
-        currentIndex = (currentIndex + 1) % Tokens.length;
-        await this.updateCacheTokens();
+        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');
@@ -625,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}`);
       }