|
@@ -205,7 +205,7 @@ public class MidjourneyServiceImpl implements MidjourneyService {
|
|
|
// }
|
|
// }
|
|
|
|
|
|
|
|
@Override
|
|
@Override
|
|
|
- public SubmitResult submitAction(MidjourneyUser user, Long taskId, String customId, Long num, String botType) throws Exception {
|
|
|
|
|
|
|
+ public Object submitAction(MidjourneyUser user, Long taskId, String customId, Long num, String botType) throws Exception {
|
|
|
MidjourneyUserConversation conversation = conversationMapper.selectOne(Wrappers.lambdaQuery(MidjourneyUserConversation.class).eq(MidjourneyUserConversation::getTaskId, taskId).last("limit 1"));
|
|
MidjourneyUserConversation conversation = conversationMapper.selectOne(Wrappers.lambdaQuery(MidjourneyUserConversation.class).eq(MidjourneyUserConversation::getTaskId, taskId).last("limit 1"));
|
|
|
if (conversation == null) {
|
|
if (conversation == null) {
|
|
|
throw BusinessRuntimeException.getInstance("关联任务不存在或已失效");
|
|
throw BusinessRuntimeException.getInstance("关联任务不存在或已失效");
|
|
@@ -239,15 +239,13 @@ public class MidjourneyServiceImpl implements MidjourneyService {
|
|
|
.build();
|
|
.build();
|
|
|
SubmitResult result = submit(user.getMode(),"action", conversation.getInstanceId(),param);
|
|
SubmitResult result = submit(user.getMode(),"action", conversation.getInstanceId(),param);
|
|
|
if (result.getCode() == 21) {
|
|
if (result.getCode() == 21) {
|
|
|
- // 以上操作有弹窗确认,恢复次数
|
|
|
|
|
- recoverUserLimit(user.getId(), user.getMode(),num);
|
|
|
|
|
if (user.getMode() == 2 && modalFlag.get()){
|
|
if (user.getMode() == 2 && modalFlag.get()){
|
|
|
- SubmitModalDTO modalParam = new SubmitModalDTO();
|
|
|
|
|
- modalParam.setPrompt(conversation.getPrompt() + " --v 6");
|
|
|
|
|
- modalParam.setTaskId(Long.valueOf(result.getResult()));
|
|
|
|
|
- String modal = HttpUtil.post(midjourneyHost + (EnvCommonService.active.equals(envCommonService.getEnv()) ? "/8081":"/8082") + "/applets/midjourney/submit/modal", Jsons.toJson(modalParam));
|
|
|
|
|
- log.info("modal:{}", modal);
|
|
|
|
|
- return result;
|
|
|
|
|
|
|
+ MidjourneyUserConversation modal = submitModal(user, Long.valueOf(result.getResult()), conversation.getPrompt() + " --v 6", null);
|
|
|
|
|
+ log.info("action-modal:{}", modal);
|
|
|
|
|
+ return modal;
|
|
|
|
|
+ }else {
|
|
|
|
|
+ // 以上操作有弹窗确认,恢复次数
|
|
|
|
|
+ recoverUserLimit(user.getId(), user.getMode(),num);
|
|
|
}
|
|
}
|
|
|
}else {
|
|
}else {
|
|
|
//syncUser(user.getId(), user.getMode(),num);
|
|
//syncUser(user.getId(), user.getMode(),num);
|