xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-19 17:06:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: reroll action error

Browse Source
This commit is contained in:
CaIon 2024-03-14 00:43:32 +08:00
parent 44361d75e8
commit a77fbc0fa2
3 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
model/ability.go
View File

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

7
service/midjourney.go
View File

@ -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
} 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
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
} }

2
web/src/pages/Channel/EditChannel.js
View File

@ -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',