Update index.js

index.js

@@ -608,9 +608,11 @@ async function handleResponse(response, model, res, isStream) {
                             if (!data.trim()) continue;
                             if (data === "[DONE]") continue;                     
                             const linejosn = JSON.parse(data);
-                            Logger.error(JSON.stringify(linejosn, null, 2), 'Server');
-                            if (linejosn?.error) {
+                            if (linejosn?.error) {                          
                                 Logger.error(JSON.stringify(linejosn, null, 2), 'Server');
+                                if(linejosn.error?.name === "RateLimitError"){
+                                    CONFIG.API.TEMP_COOKIE = null;
+                                }
                                 stream.destroy();
                                 reject(new Error("RateLimitError"));
                                 return;
@@ -640,6 +642,7 @@ async function handleResponse(response, model, res, isStream) {
                             })();
                             dataPromises.push(processPromise);
                         } catch (error) {
+                            Logger.error(error, 'Server');
                             continue;
                         }
                     }
@@ -661,11 +664,13 @@ async function handleResponse(response, model, res, isStream) {
                     CONFIG.IS_IMG_GEN2 = false;
                     resolve();
                 } catch (error) {
+                    Logger.error(error, 'Server');
                     reject(error);
                 }
             });
 
             stream.on('error', (error) => {
+                Logger.error(error, 'Server');
                 reject(error);
             });
         });
@@ -673,7 +678,7 @@ async function handleResponse(response, model, res, isStream) {
         Logger.error(error, 'Server');
         CONFIG.IS_IMG_GEN = false;
         CONFIG.IS_IMG_GEN2 = false;
-        throw error;
+        throw new Error(error);
     }
 }
 
@@ -700,6 +705,7 @@ async function handleImageResponse(imageUrl) {
             await new Promise(resolve => setTimeout(resolve, CONFIG.API.RETRY_TIME * retryCount));
 
         } catch (error) {
+            Logger.error(error, 'Server');
             retryCount++;
             if (retryCount === MAX_RETRIES) {
                 throw error;
@@ -762,12 +768,7 @@ async function handleImageResponse(imageUrl) {
         } else {
             Logger.info("生图成功", 'Server');
             const result = await responseURL.json();
-            try {
-              var url = result.data.links.url
-              return `![image](${result.data.links.url})`
-            } catch (error) {
-              return error
-            }
+            return `![image](${result.data.links.url})`
         }
     }
 }
@@ -786,7 +787,7 @@ app.use(cors({
     allowedHeaders: ['Content-Type', 'Authorization']
 }));
 
-app.get('/hf/v1/models', (req, res) => {
+app.get('/v1/models', (req, res) => {
     res.json({
         object: "list",
         data: Object.keys(CONFIG.MODELS).map((model, index) => ({
@@ -799,7 +800,7 @@ app.get('/hf/v1/models', (req, res) => {
 });
 
 
-app.post('/hf/v1/chat/completions', async (req, res) => {
+app.post('/v1/chat/completions', async (req, res) => {
     try {
         const authToken = req.headers.authorization?.replace('Bearer ', '');
         if (CONFIG.API.IS_CUSTOM_SSO) {
@@ -880,8 +881,10 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
                 Logger.info(`当前剩余可用令牌数: ${tokenManager.getTokenCount()}`, 'Server');
                 try {
                     await handleResponse(response, req.body.model, res, req.body.stream);
+                    Logger.info(`请求结束`, 'Server');
                     return;
                 } catch (error) {
+                    Logger.error(error, 'Server');
                     if (isTempCookie) {
                         await Utils.get_signature();
                     } else {
@@ -900,7 +903,7 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
                 if (response.status === 429) {
                     if (isTempCookie) {
                         await Utils.get_signature();
-                    } else {
+                    } else {                      
                         tokenManager.setModelLimit(CONFIG.SSO_INDEX, req.body.model);
                         for (let i = 1; i <= tokenManager.getTokenCount(); i++) {
                             CONFIG.SSO_INDEX = (CONFIG.SSO_INDEX + 1) % tokenManager.getTokenCount();
@@ -911,7 +914,7 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
                             }
                         }
                     }
-                } else {
+                } else {                
                     // 非429错误直接抛出
                     if (isTempCookie) {
                         await Utils.get_signature();