Update index.js

index.js

@@ -107,6 +107,8 @@ class TokenManager {
       if (wafToken) {
         Tokens[currentIndex].aws_waf_token = wafToken;
         await this.updateCacheTokens();
+        console.log("更新后的token",JSON.stringify(Tokens[currentIndex],null,2));
+        console.log("更新后cookie",CONFIG.DEFAULT_HEADERS.Cookie);
         this.updateRedisTokens();
         console.log("更新成功",)
         return;
@@ -514,26 +516,29 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
           },
           body: JSON.stringify(requestPayload)
         });
+        console.log("请求的头部",JSON.stringify(CONFIG.DEFAULT_HEADERS,null,2));
+        
         reqStatus = response.status;
         switch (reqStatus) {
           case 200:
             console.log("请求成功");
-            // 异步更新token
-            tokenManager.updateTokens(response)
             // 处理响应
             if (req.body.stream) {
               await ResponseHandler.handleStreamResponse(response, req.body.model, res);
             } else {
               await ResponseHandler.handleNormalResponse(response, req.body.model, res);
             }
+            await tokenManager.updateTokens(response);
             return;
           case 202:
             console.log("请求受限，更新WAF");
             await tokenManager.updateTokens(response, true);
             break;
           default:
+            console.log(await response.text());
             throw new Error(`请求失败! status: ${response.status}`);
         }
+        console.log("请求失败，重试中...");
         retryCount++;
         if (retryCount >= CONFIG.RETRY.MAX_ATTEMPTS) {
           throw new Error(`上游服务请求失败! status: ${response.status}`);