Update index.js

index.js

@@ -15,7 +15,7 @@ let currentIndex = 0;
 let sessionId = null;
 const CONFIG = {
   API: {
-    BASE_URL: process.env.DENO_URL || "https://partyrock.aws/stream/getCompletion",//如果需要多号循环，需要设置你自己的denourl
+    BASE_URL: "https://partyrock.aws/stream/getCompletion",//如果需要多号循环，需要设置你自己的denourl
     API_KEY: process.env.API_KEY || "sk-123456",//自定义你自己的认证密钥，记得修改
     RedisUrl: process.env.RedisUrl,
     RedisToken: process.env.RedisToken
@@ -35,9 +35,10 @@ const CONFIG = {
     'mistral-large': 'bedrock-mistral.mistral-large-2407-v1-0'
   },
   DEFAULT_HEADERS: {
+    "request-id": "",
     "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
     "Cache-Control": "no-cache, no-store",
-    "Connection": "close",
+    "pragma": "no-cache",
     "Accept": "text/event-stream",
     "Accept-Encoding": "gzip, deflate, br, zstd",
     "Content-Type": "application/json",
@@ -98,6 +99,7 @@ class TokenManager {
     CONFIG.DEFAULT_HEADERS["anti-csrftoken-a2z"] = Tokens[currentIndex].anti_csrftoken_a2z;
     CONFIG.DEFAULT_HEADERS.Cookie = `idToken=${Tokens[currentIndex].idToken}; pr_refresh_token=${Tokens[currentIndex].pr_refresh_token};aws-waf-token=${Tokens[currentIndex].aws_waf_token};cwr_s=${Tokens[currentIndex].cwr_s};cwr_u=${sessionId}`;
     CONFIG.DEFAULT_HEADERS.referer = Tokens[currentIndex].refreshUrl;
+    CONFIG.DEFAULT_HEADERS["request-id"] = `request-id-${Utils.uuidv4()}`;
   }
 
   async updateTokens(response, isWaf = false) {
@@ -107,10 +109,10 @@ class TokenManager {
         Tokens[currentIndex].aws_waf_token = wafToken;
         await this.updateCacheTokens();
         this.updateRedisTokens();
-        currentIndex = (currentIndex + 1) % Tokens.length;
+        //currentIndex = (currentIndex + 1) % Tokens.length;
         console.log("成功提取 aws-waf-token");
       } else {
-        currentIndex = (currentIndex + 1) % Tokens.length;
+        //currentIndex = (currentIndex + 1) % Tokens.length;
         await this.updateCacheTokens();
         console.log("提取aws-waf-token失败");
       }
@@ -127,7 +129,7 @@ class TokenManager {
         this.updateRedisTokens();//最后更新redis数据库缓存,异步
         console.log("更新缓存完毕");
       }
-      currentIndex = (currentIndex + 1) % Tokens.length;
+      //currentIndex = (currentIndex + 1) % Tokens.length;
     }
   }
 }