Commit bc2b1112 by CaIon

fix: reroll action error

parent da55c56c
...@@ -147,7 +147,12 @@ func FixAbility() (int, error) { ...@@ -147,7 +147,12 @@ func FixAbility() (int, error) {
return 0, err return 0, err
} }
var channels []Channel 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 { if err != nil {
return 0, err return 0, err
} }
......
...@@ -45,7 +45,7 @@ func GetMjRequestModel(relayMode int, midjRequest *dto.MidjourneyRequest) (strin ...@@ -45,7 +45,7 @@ func GetMjRequestModel(relayMode int, midjRequest *dto.MidjourneyRequest) (strin
case relayconstant.RelayModeMidjourneyTaskFetch, relayconstant.RelayModeMidjourneyTaskFetchByCondition, relayconstant.RelayModeMidjourneyNotify: case relayconstant.RelayModeMidjourneyTaskFetch, relayconstant.RelayModeMidjourneyTaskFetchByCondition, relayconstant.RelayModeMidjourneyNotify:
return "", nil, true return "", nil, true
default: default:
return "", MidjourneyErrorWrapper(constant.MjRequestError, "unknown_action"), false return "", MidjourneyErrorWrapper(constant.MjRequestError, "unknown_relay_action"), false
} }
} }
modelName := CoverActionToModelName(action) modelName := CoverActionToModelName(action)
...@@ -93,6 +93,9 @@ func CoverPlusActionToNormalAction(midjRequest *dto.MidjourneyRequest) *dto.Midj ...@@ -93,6 +93,9 @@ func CoverPlusActionToNormalAction(midjRequest *dto.MidjourneyRequest) *dto.Midj
} else if strings.Contains(action, "pan") { } else if strings.Contains(action, "pan") {
midjRequest.Action = constant.MjActionPan midjRequest.Action = constant.MjActionPan
midjRequest.Index = 1 midjRequest.Index = 1
} else if strings.Contains(action, "reroll") {
midjRequest.Action = constant.MjActionReRoll
midjRequest.Index = 1
} else if action == "Outpaint" || action == "CustomZoom" { } else if action == "Outpaint" || action == "CustomZoom" {
midjRequest.Action = constant.MjActionZoom midjRequest.Action = constant.MjActionZoom
midjRequest.Index = 1 midjRequest.Index = 1
...@@ -100,7 +103,7 @@ func CoverPlusActionToNormalAction(midjRequest *dto.MidjourneyRequest) *dto.Midj ...@@ -100,7 +103,7 @@ func CoverPlusActionToNormalAction(midjRequest *dto.MidjourneyRequest) *dto.Midj
midjRequest.Action = constant.MjActionInPaintPre midjRequest.Action = constant.MjActionInPaintPre
midjRequest.Index = 1 midjRequest.Index = 1
} else { } else {
return MidjourneyErrorWrapper(constant.MjRequestError, "unknown_action") return MidjourneyErrorWrapper(constant.MjRequestError, "unknown_action:"+customId)
} }
return nil return nil
} }
......
...@@ -110,7 +110,7 @@ const EditChannel = (props) => { ...@@ -110,7 +110,7 @@ const EditChannel = (props) => {
'mj_zoom', 'mj_zoom',
'mj_shorten', 'mj_shorten',
'mj_inpaint_pre', 'mj_inpaint_pre',
'mj_inpaint_pre', 'mj_inpaint',
'mj_high_variation', 'mj_high_variation',
'mj_low_variation', 'mj_low_variation',
'mj_pan', 'mj_pan',
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment