(relayMode int, midjRequest *dto.MidjourneyRequest)
| 27 | } |
| 28 | |
| 29 | func GetMjRequestModel(relayMode int, midjRequest *dto.MidjourneyRequest) (string, *dto.MidjourneyResponse, bool) { |
| 30 | action := "" |
| 31 | if relayMode == relayconstant.RelayModeMidjourneyAction { |
| 32 | // plus request |
| 33 | err := CoverPlusActionToNormalAction(midjRequest) |
| 34 | if err != nil { |
| 35 | return "", err, false |
| 36 | } |
| 37 | action = midjRequest.Action |
| 38 | } else { |
| 39 | switch relayMode { |
| 40 | case relayconstant.RelayModeMidjourneyImagine: |
| 41 | action = constant.MjActionImagine |
| 42 | case relayconstant.RelayModeMidjourneyVideo: |
| 43 | action = constant.MjActionVideo |
| 44 | case relayconstant.RelayModeMidjourneyEdits: |
| 45 | action = constant.MjActionEdits |
| 46 | case relayconstant.RelayModeMidjourneyDescribe: |
| 47 | action = constant.MjActionDescribe |
| 48 | case relayconstant.RelayModeMidjourneyBlend: |
| 49 | action = constant.MjActionBlend |
| 50 | case relayconstant.RelayModeMidjourneyShorten: |
| 51 | action = constant.MjActionShorten |
| 52 | case relayconstant.RelayModeMidjourneyChange: |
| 53 | action = midjRequest.Action |
| 54 | case relayconstant.RelayModeMidjourneyModal: |
| 55 | action = constant.MjActionModal |
| 56 | case relayconstant.RelayModeSwapFace: |
| 57 | action = constant.MjActionSwapFace |
| 58 | case relayconstant.RelayModeMidjourneyUpload: |
| 59 | action = constant.MjActionUpload |
| 60 | case relayconstant.RelayModeMidjourneySimpleChange: |
| 61 | params := ConvertSimpleChangeParams(midjRequest.Content) |
| 62 | if params == nil { |
| 63 | return "", MidjourneyErrorWrapper(constant.MjRequestError, "invalid_request"), false |
| 64 | } |
| 65 | action = params.Action |
| 66 | case relayconstant.RelayModeMidjourneyTaskFetch, relayconstant.RelayModeMidjourneyTaskFetchByCondition, relayconstant.RelayModeMidjourneyNotify: |
| 67 | return "", nil, true |
| 68 | default: |
| 69 | return "", MidjourneyErrorWrapper(constant.MjRequestError, "unknown_relay_action"), false |
| 70 | } |
| 71 | } |
| 72 | modelName := CoverActionToModelName(action) |
| 73 | return modelName, nil, true |
| 74 | } |
| 75 | |
| 76 | func CoverPlusActionToNormalAction(midjRequest *dto.MidjourneyRequest) *dto.MidjourneyResponse { |
| 77 | // "customId": "MJ::JOB::upsample::2::3dbbd469-36af-4a0f-8f02-df6c579e7011" |
no test coverage detected