diff --git a/model/midjourney.go b/model/midjourney.go index 64b99349..c6e5c9a1 100644 --- a/model/midjourney.go +++ b/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 } diff --git a/web/src/components/MjLogsTable.js b/web/src/components/MjLogsTable.js index 63d14cd3..8a93d130 100644 --- a/web/src/components/MjLogsTable.js +++ b/web/src/components/MjLogsTable.js @@ -337,7 +337,8 @@ const LogsTable = () => {