فهرست منبع

fix: reroll action error

CaIon 2 سال پیش
والد
کامیت
a77fbc0fa2
3فایلهای تغییر یافته به همراه12 افزوده شده و 4 حذف شده
  1. 6 1
      model/ability.go
  2. 5 2
      service/midjourney.go
  3. 1 1
      web/src/pages/Channel/EditChannel.js

+ 6 - 1
model/ability.go

@@ -147,7 +147,12 @@ func FixAbility() (int, error) {
 		return 0, err
 	}
 	var channels []Channel
-	err = DB.Where("id NOT IN (?)", abilityChannelIds).Find(&channels).Error
+
+	if len(abilityChannelIds) == 0 {
+		err = DB.Find(&channels).Error
+	} else {
+		err = DB.Where("id NOT IN (?)", abilityChannelIds).Find(&channels).Error
+	}
 	if err != nil {
 		return 0, err
 	}

+ 5 - 2
service/midjourney.go

@@ -45,7 +45,7 @@ func GetMjRequestModel(relayMode int, midjRequest *dto.MidjourneyRequest) (strin
 		case relayconstant.RelayModeMidjourneyTaskFetch, relayconstant.RelayModeMidjourneyTaskFetchByCondition, relayconstant.RelayModeMidjourneyNotify:
 			return "", nil, true
 		default:
-			return "", MidjourneyErrorWrapper(constant.MjRequestError, "unknown_action"), false
+			return "", MidjourneyErrorWrapper(constant.MjRequestError, "unknown_relay_action"), false
 		}
 	}
 	modelName := CoverActionToModelName(action)
@@ -93,6 +93,9 @@ func CoverPlusActionToNormalAction(midjRequest *dto.MidjourneyRequest) *dto.Midj
 	} else if strings.Contains(action, "pan") {
 		midjRequest.Action = constant.MjActionPan
 		midjRequest.Index = 1
+	} else if strings.Contains(action, "reroll") {
+		midjRequest.Action = constant.MjActionReRoll
+		midjRequest.Index = 1
 	} else if action == "Outpaint" || action == "CustomZoom" {
 		midjRequest.Action = constant.MjActionZoom
 		midjRequest.Index = 1
@@ -100,7 +103,7 @@ func CoverPlusActionToNormalAction(midjRequest *dto.MidjourneyRequest) *dto.Midj
 		midjRequest.Action = constant.MjActionInPaintPre
 		midjRequest.Index = 1
 	} else {
-		return MidjourneyErrorWrapper(constant.MjRequestError, "unknown_action")
+		return MidjourneyErrorWrapper(constant.MjRequestError, "unknown_action:"+customId)
 	}
 	return nil
 }

+ 1 - 1
web/src/pages/Channel/EditChannel.js

@@ -110,7 +110,7 @@ const EditChannel = (props) => {
                         'mj_zoom',
                         'mj_shorten',
                         'mj_inpaint_pre',
-                        'mj_inpaint_pre',
+                        'mj_inpaint',
                         'mj_high_variation',
                         'mj_low_variation',
                         'mj_pan',