dan92 commited on
Commit
7cde6d0
·
verified ·
1 Parent(s): 840c33b

Upload index.js

Browse files
Files changed (1) hide show
  1. index.js +19 -2
index.js CHANGED
@@ -403,6 +403,19 @@ class MessageProcessor {
403
  model: model
404
  };
405
 
 
 
 
 
 
 
 
 
 
 
 
 
 
406
  return {
407
  ...baseResponse,
408
  object: 'chat.completion',
@@ -410,7 +423,7 @@ class MessageProcessor {
410
  index: 0,
411
  message: {
412
  role: 'assistant',
413
- content: message // 直接返回原始消息
414
  },
415
  finish_reason: 'stop'
416
  }],
@@ -615,7 +628,11 @@ async function handleImageResponse(imageUrl) {
615
  const imageContentType = imageBase64Response.headers.get('content-type');
616
 
617
  // 修改返回结构,确保返回Markdown格式的图片标签
618
- return `![图片](data:image/jpg;base64,${base64String})`; // Markdown格式
 
 
 
 
619
  } catch (error) {
620
  console.error('图片处理失败:', error);
621
  return '图片生成失败';
 
403
  model: model
404
  };
405
 
406
+ // 处理图片响应
407
+ const processContent = (content) => {
408
+ if (content?.image_url?.url) {
409
+ return [{
410
+ type: "image_url",
411
+ image_url: {
412
+ url: content.image_url.url
413
+ }
414
+ }];
415
+ }
416
+ return content;
417
+ };
418
+
419
  return {
420
  ...baseResponse,
421
  object: 'chat.completion',
 
423
  index: 0,
424
  message: {
425
  role: 'assistant',
426
+ content: processContent(message)
427
  },
428
  finish_reason: 'stop'
429
  }],
 
628
  const imageContentType = imageBase64Response.headers.get('content-type');
629
 
630
  // 修改返回结构,确保返回Markdown格式的图片标签
631
+ return {
632
+ image_url: {
633
+ url: `data:image/jpg;base64,${base64String}`
634
+ }
635
+ };
636
  } catch (error) {
637
  console.error('图片处理失败:', error);
638
  return '图片生成失败';