Update index.js

index.js

@@ -15,7 +15,7 @@ let currentIndex = 0;
 let sessionId = null;
 const CONFIG = {
   API: {
-    BASE_URL: "https://partyrock.aws/stream/getCompletion",//如果需要多号循环，需要设置你自己的denourl
+    BASE_URL: process.env.DENO_URL || "https://partyrock.aws/stream/getCompletion",//如果需要多号循环，需要设置你自己的denourl
     API_KEY: process.env.API_KEY || "sk-123456",//自定义你自己的认证密钥，记得修改
     RedisUrl: process.env.RedisUrl,
     RedisToken: process.env.RedisToken
@@ -35,7 +35,6 @@ const CONFIG = {
     'mistral-large': 'bedrock-mistral.mistral-large-2407-v1-0'
   },
   DEFAULT_HEADERS: {
-    "host": "partyrock.aws",
     "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",
@@ -101,7 +100,6 @@ class TokenManager {
     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()}`;
-    CONFIG.DEFAULT_HEADERS["User-Agent"] = await Utils.getRandomUserAgent();
   }
 
   async updateTokens(response, isWaf = false) {
@@ -527,6 +525,7 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
     await tokenManager.updateCacheTokens();
     const apiClient = new ApiClient(req.body.model);
     const requestPayload = await apiClient.transformMessages(req.body);
+    console.log(JSON.stringify(CONFIG.DEFAULT_HEADERS, null, 2));
     
     try {
       console.log("开始请求");
@@ -566,7 +565,7 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
         case 403:
           console.log("请求被阻止");
           await tokenManager.updateTokens(response, true);//尝试获取waf，然后返回错误提示。
-          //CONFIG.DEFAULT_HEADERS["User-Agent"] = await Utils.getRandomUserAgent();
+          CONFIG.DEFAULT_HEADERS["User-Agent"] = await Utils.getRandomUserAgent();
           throw new Error(`请求失败! status: ${response.statusText}，请重新请求，如果多次失败，请重新更换token`);
         default:
           throw new Error(`请求失败! status: ${response.status}`);