Update index.js

index.js

@@ -636,12 +636,12 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
 
             if (response.ok) {
                 Logger.info(`请求成功`, 'Server');
-                CONFIG.SIGNATURE_INDEX = (CONFIG.SIGNATURE_INDEX + 1) % tokenManager.getTokenCount();
                 Logger.info(`当前剩余可用令牌数: ${tokenManager.getTokenCount()}`, 'Server');
                 try {
                     await (req.body.stream
                         ? ResponseHandler.handleStreamResponse(response, req.body.model, res)
                         : ResponseHandler.handleNormalResponse(response, req.body.model, res));
+                    CONFIG.SIGNATURE_INDEX = (CONFIG.SIGNATURE_INDEX + 1) % tokenManager.getTokenCount();    
                     return; // 成功后直接返回
                 } catch (error) {
                     tokenManager.removeTokenByIndex(CONFIG.SIGNATURE_INDEX);
@@ -659,11 +659,10 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
                     if (!tokenManager.isRecoveryProcess) {
                         tokenManager.startTokenRecoveryProcess();
                     }
-                    Logger.warn(`当前令牌失效，已移除令牌，剩余令牌数: ${tokenManager.getTokenCount()}`, 'Server');
-                    // 更新签名索引
                     CONFIG.SIGNATURE_INDEX = (CONFIG.SIGNATURE_INDEX + 1) % tokenManager.getTokenCount();
-                    Logger.warn(`请求被限流，剩余重试次数: ${tokenManager.getTokenCount()}`, 'ChatAPI');
+                    Logger.warn(`当前令牌失效，已移除令牌，剩余令牌数: ${tokenManager.getTokenCount()}`, 'Server');
                 } else {
+                    CONFIG.SIGNATURE_INDEX = (CONFIG.SIGNATURE_INDEX + 1) % tokenManager.getTokenCount();
                     // 非429错误直接抛出
                     throw new Error(`上游服务请求失败! status: ${response.status}`);
                 }
@@ -672,8 +671,8 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
 
         // 如果重试次数用完仍然是429
         throw new Error('所有令牌都已耗尽，请求被限流');
-
     } catch (error) {
+        CONFIG.SIGNATURE_INDEX = (CONFIG.SIGNATURE_INDEX + 1) % tokenManager.getTokenCount();
         Logger.error('Chat Completions Request Error', error, 'ChatAPI');
         res.status(500).json({
             error: {