Upload index.js

index.js

@@ -403,6 +403,19 @@ class MessageProcessor {
             model: model
         };
 
+        // 处理图片响应
+        const processContent = (content) => {
+            if (typeof content === 'string' && content.startsWith('data:image')) {
+                return [{
+                    type: "image_url",
+                    image_url: {
+                        url: content
+                    }
+                }];
+            }
+            return content;
+        };
+
         return {
             ...baseResponse,
             object: 'chat.completion',
@@ -410,9 +423,7 @@ class MessageProcessor {
                 index: 0,
                 message: {
                     role: 'assistant',
-                    content: Array.isArray(message) ? 
-                        message.map(item => item.image_url?.url || item) : // 提取图片URL
-                        message
+                    content: processContent(message)
                 },
                 finish_reason: 'stop'
             }],
@@ -591,7 +602,12 @@ async function handleImageResponse(imageUrl) {
         const imageContentType = imageBase64Response.headers.get('content-type');
 
         // 修改返回结构，确保返回Markdown格式的图片标签
-        return `data:image/jpg;base64,${base64String}`; // 直接返回Data URL字符串
+        return {
+            type: "image_url",
+            image_url: {
+                url: `data:image/jpg;base64,${base64String}`
+            }
+        };
     } catch (error) {
         console.error('图片处理失败:', error);
         return '图片生成失败';