فهرست منبع

feat: 改为转发的方式获取midjourney图片

CaIon 2 سال پیش
والد
کامیت
d9deda18c1
2فایلهای تغییر یافته به همراه10 افزوده شده و 1 حذف شده
  1. 8 0
      model/midjourney.go
  2. 2 1
      web/src/components/MjLogsTable.js

+ 8 - 0
model/midjourney.go

@@ -1,5 +1,7 @@
 package model
 
+import "one-api/common"
+
 type Midjourney struct {
 	Id          int    `json:"id"`
 	Code        int    `json:"code"`
@@ -27,6 +29,9 @@ func GetAllUserTask(userId int, startIdx int, num int) []*Midjourney {
 	if err != nil {
 		return nil
 	}
+	for _, task := range tasks {
+		task.ImageUrl = common.ServerAddress + "/mj/image/" + task.MjId
+	}
 	return tasks
 }
 
@@ -37,6 +42,9 @@ func GetAllTasks(startIdx int, num int) []*Midjourney {
 	if err != nil {
 		return nil
 	}
+	for _, task := range tasks {
+		task.ImageUrl = common.ServerAddress + "/mj/image/" + task.MjId
+	}
 	return tasks
 }
 

+ 2 - 1
web/src/components/MjLogsTable.js

@@ -337,7 +337,8 @@ const LogsTable = () => {
                     <Table.Cell>
                       {
                         log.image_url ? (
-                            <Link to={log.image_url} target='_blank'>点击查看</Link>
+                            // <Link to={log.image_url} target='_blank'>点击查看</Link>
+                            <a href={log.image_url} target='_blank'>点击查看</a>
                         ) : '暂未生成图片'
                       }
                     </Table.Cell>