Update index.js

index.js

@@ -20,10 +20,6 @@ const CONFIG = {
     RedisUrl: process.env.RedisUrl,
     RedisToken: process.env.RedisToken
   },
-  RETRY: {
-    MAX_ATTEMPTS: 1,
-    DELAY_BASE: 1000
-  },
   SERVER: {
     PORT: process.env.PORT || 3000,
     BODY_LIMIT: '5mb'
@@ -101,7 +97,7 @@ class TokenManager {
     sessionId = Utils.uuidv4();
     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.referer = Tokens[currentIndex].refererUrl;
   }
 
   async updateTokens(response, isWaf = false) {
@@ -171,7 +167,7 @@ class Utils {
       await page.setUserAgent(
         CONFIG.DEFAULT_HEADERS["User-Agent"]
       )
-      await page.goto(Tokens[currentIndex].refreshUrl, {
+      await page.goto(Tokens[currentIndex].refererUrl, {
         waitUntil: 'networkidle2',
         timeout: 30000
       });
@@ -256,7 +252,7 @@ async function initializeService() {
           ? JSON.parse(checkRedis)
           : checkRedis;
         Tokens.push({
-          refreshUrl: parsedRedis.refreshUrl,
+          refererUrl: parsedRedis.refererUrl,
           anti_csrftoken_a2z: parsedRedis.anti_csrftoken_a2z,
           pr_refresh_token: parsedRedis.pr_refresh_token,
           aws_waf_token: parsedRedis.aws_waf_token,
@@ -269,18 +265,18 @@ async function initializeService() {
       }
     } else {
       console.log(index, '没有缓存，开始提取环境变量');
-      const refreshUrl = process.env[`AUTH_TOKENS_${index}_REFRESH_URL`];
+      const refererUrl = process.env[`AUTH_TOKENS_${index}_REFRESH_URL`];
       const anti_csrftoken_a2z = process.env[`AUTH_TOKENS_${index}_ANTI_CSRF_TOKEN`];
       const cookie = process.env[`AUTH_TOKENS_${index}_COOKIE`];
 
-      if (!refreshUrl && !anti_csrftoken_a2z && !cookie) {
+      if (!refererUrl && !anti_csrftoken_a2z && !cookie) {
         break;
       }
       const cookies = await Utils.extractTokens(cookie);
 
-      if (refreshUrl && anti_csrftoken_a2z && cookie) {
+      if (refererUrl && anti_csrftoken_a2z && cookie) {
         Tokens.push({
-          refreshUrl,
+          refererUrl,
           anti_csrftoken_a2z,
           pr_refresh_token: cookies["pr_refresh_token"],
           aws_waf_token: cookies["aws-waf-token"],
@@ -530,7 +526,7 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
     try {
       console.log("开始请求");
       //发送请求
-      var response = await fetch(`https://strong-tuna-37.deno.dev`, {
+      var response = await fetch(`https://partyrock.aws/stream/getCompletion`, {
         method: "POST",
         headers: {
           ...CONFIG.DEFAULT_HEADERS
@@ -564,9 +560,7 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
           throw new Error(`请求失败! status: ${response.statusText}，已刷新验证信息，请重新请求`);
         case 403:
           console.log("请求被阻止");
-          console.log(JSON.stringify(CONFIG.DEFAULT_HEADERS, null, 2));
           CONFIG.DEFAULT_HEADERS["User-Agent"] = await Utils.getRandomUserAgent();
-          console.log(response.headers);
           throw new Error(`请求失败! status: ${response.statusText}，请重新请求，如果多次失败，请重新更换token`);
         default:
           throw new Error(`请求失败! status: ${response.status}`);