Update index.js

index.js

@@ -504,15 +504,10 @@ async function handleStreamResponse(response, model, res) {
             buffer = lines.pop() || '';
 
             for (const line of lines) {
-                if (!line.startsWith('data: ')) continue;
+                if (!line.trim()) continue;
                 const trimmedLine = line.trim();
                 if (trimmedLine.startsWith('data: ')) {
                     const content = trimmedLine.substring(6);
-                    if (content === '[DONE]') {
-                        res.write('data: [DONE]\n\n');
-                        continue;
-                    }
-
                     try {
                         const linejosn = JSON.parse(content);
                         if (linejosn?.error?.name === "RateLimitError") {
@@ -521,6 +516,7 @@ async function handleStreamResponse(response, model, res) {
                             CONFIG.API.SIGNATURE_COOKIE = null;
                             console.log("认证信息已删除");
                             res.write(`data: ${JSON.stringify(responseData)}\n\n`);
+                            res.write('data: [DONE]\n\n');
                             return res.end();
                         }
                         switch (model) {
@@ -561,6 +557,10 @@ async function handleStreamResponse(response, model, res) {
                 }
             }
         });
+        stream.on('end', () => {
+            res.write('data: [DONE]\n\n');
+            res.end();
+        });
         stream.on('error', (error) => {
             console.error('流处理错误:', error);
             res.write('data: [DONE]\n\n');
@@ -583,15 +583,13 @@ async function handleNormalResponse(response, model, res) {
         const lines = responseText.split('\n');
 
         for (const line of lines) {
-            if (!line.startsWith('data: ')) continue;
-
+            if (!line.trim()) continue;
             const data = line.slice(6);
             if (data === '[DONE]') continue;
             let linejosn = JSON.parse(data);
             if (linejosn?.error?.name === "RateLimitError") {
                 fullResponse = `${linejosn.error.name},请重新对话`;
                 CONFIG.API.SIGNATURE_COOKIE = null;
-                console.log("认证信息已删除");
                 return res.json(MessageProcessor.createChatResponse(fullResponse, model));
             }
             switch (model) {
@@ -701,7 +699,7 @@ async function handleImageResponse(imageUrl) {
 
 // 404处理
 app.use((req, res) => {
-    res.status(404).send('请求路径不存在');
+    res.status(200).send('api运行正常');
 });
 
 // 启动服务器