Upload index.js

index.js

@@ -560,55 +560,69 @@ async function handleNormalResponse(response, model, res) {
         Utils.handleError(error, res);
     }
 }
+
 async function handleImageResponse(imageUrl) {
-    //对服务器发送图片请求
-    const MAX_RETRIES = 3;
-    let retryCount = 0;
-    let imageBase64Response;
+    try {
+        //对服务器发送图片请求
+        const MAX_RETRIES = 3;
+        let retryCount = 0;
+        let imageBase64Response;
 
-    while (retryCount < MAX_RETRIES) {
-        try {
-            //发送图片请求获取图片
-            imageBase64Response = await fetch(`https://assets.grok.com/${imageUrl}`, {
-                method: 'GET',
-                headers: {
-                    ...DEFAULT_HEADERS,
-                    ...CONFIG.API.SIGNATURE_COOKIE
-                }
-            });
+        while (retryCount < MAX_RETRIES) {
+            try {
+                //发送图片请求获取图片
+                imageBase64Response = await fetch(`https://assets.grok.com/${imageUrl}`, {
+                    method: 'GET',
+                    headers: {
+                        ...DEFAULT_HEADERS,
+                        ...CONFIG.API.SIGNATURE_COOKIE
+                    }
+                });
 
-            if (imageBase64Response.ok) {
-                break; // 如果请求成功，跳出重试循环
-            }
+                if (imageBase64Response.ok) {
+                    break; // 如果请求成功，跳出重试循环
+                }
 
-            retryCount++;
-            if (retryCount === MAX_RETRIES) {
-                throw new Error(`上游服务请求失败! status: ${imageBase64Response.status}`);
-            }
+                retryCount++;
+                if (retryCount === MAX_RETRIES) {
+                    throw new Error(`上游服务请求失败! status: ${imageBase64Response.status}`);
+                }
 
-            // 等待一段时间后重试（可以使用指数退避）
-            await new Promise(resolve => setTimeout(resolve, 500 * retryCount));
+                // 等待一段时间后重试（可以使用指数退避）
+                await new Promise(resolve => setTimeout(resolve, 500 * retryCount));
 
-        } catch (error) {
-            retryCount++;
-            if (retryCount === MAX_RETRIES) {
-                throw error;
+            } catch (error) {
+                retryCount++;
+                if (retryCount === MAX_RETRIES) {
+                    throw error;
+                }
+                // 等待一段时间后重试
+                await new Promise(resolve => setTimeout(resolve, 500 * retryCount));
             }
-            // 等待一段时间后重试
-            await new Promise(resolve => setTimeout(resolve, 500 * retryCount));
         }
-    }
 
-    const arrayBuffer = await imageBase64Response.arrayBuffer();
-    const imagebuffer = Buffer.from(arrayBuffer);
+        const arrayBuffer = await imageBase64Response.arrayBuffer();
+        const imagebuffer = Buffer.from(arrayBuffer);
 
-    const base64String = imagebuffer.toString('base64');
+        const base64String = imagebuffer.toString('base64');
 
-    // 获取图片类型（例如：image/jpeg, image/png）
-    const imageContentType = imageBase64Response.headers.get('content-type');
+        // 获取图片类型（例如：image/jpeg, image/png）
+        const imageContentType = imageBase64Response.headers.get('content-type');
 
-    // 构建 Data URL返回
-    return `data:${imageContentType};base64,${base64String}`;
+        // 修改返回结构，确保返回Markdown格式的图片标签
+        return {
+            type: "image_url",
+            image_url: {
+                url: `data:image/jpg;base64,${base64String}`
+            }
+        };
+
+        // 或者如果需要直接返回可显示的字符串
+        // return `![generated image](data:image/jpg;base64,${base64String})`;
+    } catch (error) {
+        console.error('图片处理失败:', error);
+        return '图片生成失败';
+    }
 }
 
 // 404处理