Merge pull request #167 from weikecloud/main

增加MJ上游构图失败判断

Merge pull request #167 from weikecloud/main
增加MJ上游构图失败判断
a8433823 · Calcium-Ion · GitHub · 130d2751 · c30cba8b · a8433823
Commit a8433823 authored Mar 30, 2024 by Calcium-Ion Committed by GitHub Mar 30, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

controller/midjourney.go
+1 -1

No files found.
--- a/controller/midjourney.go
+++ b/controller/midjourney.go
@@ -147,7 +147,7 @@ func UpdateMidjourneyTaskBulk() {
 					task.Buttons = string(buttonStr)
 				}

-				if task.Progress != "100%" && responseItem.FailReason != "" {
+				if (task.Progress != "100%" && responseItem.FailReason != "") || (task.Progress == "100%" && task.Status == "FAILURE") {
 					common.LogInfo(ctx, task.MjId+" 构建失败，"+task.FailReason)
 					task.Progress = "100%"
 					err = model.CacheUpdateUserQuota(task.UserId)