Просмотр исходного кода

Merge pull request #167 from weikecloud/main

增加MJ上游构图失败判断
Calcium-Ion 1 год назад
Родитель
Сommit
87919b032d
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      controller/midjourney.go

+ 1 - 1
controller/midjourney.go

@@ -147,7 +147,7 @@ func UpdateMidjourneyTaskBulk() {
 					task.Buttons = string(buttonStr)
 					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)
 					common.LogInfo(ctx, task.MjId+" 构建失败,"+task.FailReason)
 					task.Progress = "100%"
 					task.Progress = "100%"
 					err = model.CacheUpdateUserQuota(task.UserId)
 					err = model.CacheUpdateUserQuota(task.UserId)