Update index.js

index.js

@@ -26,7 +26,7 @@ const CONFIG = {
         BODY_LIMIT: '5mb'
     },
     RETRY: {
-        MAX_ATTEMPTS: 3,//重试次数
+        MAX_ATTEMPTS: 1,//重试次数
         DELAY_BASE: 1000 // 基础延迟时间（毫秒）
     },
     ISSHOW_SEARCH_RESULTS: process.env.ISSHOW_SEARCH_RESULTS === 'true',//是否显示搜索结果,默认关闭
@@ -508,10 +508,14 @@ async function handleStreamResponse(response, model, res) {
                 const trimmedLine = line.trim();
                 if (trimmedLine.startsWith('data: ')) {
                     const data = trimmedLine.substring(6);
+                    if(data === "[DONE]"){
+                        console.log("流结束");
+                        res.write('data: [DONE]\n\n');
+                        return res.end();
+                    };
                     try {
-                        var content = data.trim();
-                        if(!content)continue;
-                        const linejosn = JSON.parse(content);
+                        if(!data.trim())continue;
+                        const linejosn = JSON.parse(data);
                         if (linejosn?.error?.name === "RateLimitError") {
                             var responseData = MessageProcessor.createChatResponse(`${linejosn.error.name},请重新对话`, model, true);
                             fs.unlinkSync(path.resolve(process.cwd(), 'signature.json'));
@@ -526,7 +530,7 @@ async function handleStreamResponse(response, model, res) {
                                 if (linejosn.response === "modelResponse" && linejosn?.modelResponse?.generatedImageUrls) {
                                     const dataImage = await handleImageResponse(linejosn.modelResponse.generatedImageUrls);
                                     const responseData = MessageProcessor.createChatResponse(dataImage, model, true);
-                                    res.write(`data: ${JSON.stringify(responseData)}\n\n`);
+                                    res.write(`data: ${JSON.stringify(responseData)}\n\n`);   
                                 }
                                 break;
                             case 'grok-latest':
@@ -559,10 +563,6 @@ 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');
@@ -638,7 +638,7 @@ async function handleImageResponse(imageUrl) {
     const MAX_RETRIES = 3;
     let retryCount = 0;
     let imageBase64Response;
-
+    
     while (retryCount < MAX_RETRIES) {
         try {
             //发送图片请求获取图片
@@ -691,7 +691,6 @@ async function handleImageResponse(imageUrl) {
         body: formData
     });
     if (!responseURL.ok) {
-        console.error("上传图床失败");
         return "生图失败，请查看图床密钥是否设置正确"
     } else {
         const result = await responseURL.json();