BJ_ZTEITS-Developers / parking-cloud-yudao

Commit 27f052de0a7bf9f61d17eea59427aee8fcf79c47

Authored by cherishsince
1 parent 74620ebc

【增加】midjourney 增加 --ar 参数设置

Inline Side-by-side
Showing 2 changed files with 9 additions and 2 deletions
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/vo/AiImageMidjourneyImagineReqVO.java
  View file @27f052d
... ... @@ -25,6 +25,12 @@ public class AiImageMidjourneyImagineReqVO {
25 25 @Schema(description = "模型(midjourney、niji)")
26 26 private String model;
27 27  
  28 + @Schema(description = "图片宽度")
  29 + private String width;
  30 +
  31 + @Schema(description = "图片高度")
  32 + private String height;
  33 +
28 34 @Schema(description = "垫图(参考图)base64数组")
29 35 private List<String> base64Array;
30 36 }
... ...
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/image/AiImageServiceImpl.java
  View file @27f052d
... ... @@ -12,8 +12,8 @@ import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
12 12 import cn.iocoder.yudao.module.ai.AiCommonConstants;
13 13 import cn.iocoder.yudao.module.ai.ErrorCodeConstants;
14 14 import cn.iocoder.yudao.module.ai.client.MidjourneyProxyClient;
  15 +import cn.iocoder.yudao.module.ai.client.enums.MidjourneySubmitCodeEnum;
15 16 import cn.iocoder.yudao.module.ai.client.vo.MidjourneyImagineReqVO;
16   -import cn.iocoder.yudao.module.ai.client.vo.MidjourneySubmitCodeEnum;
17 17 import cn.iocoder.yudao.module.ai.client.vo.MidjourneySubmitRespVO;
18 18 import cn.iocoder.yudao.module.ai.controller.admin.image.vo.*;
19 19 import cn.iocoder.yudao.module.ai.dal.dataobject.image.AiImageDO;
... ... @@ -157,7 +157,8 @@ public class AiImageServiceImpl implements AiImageService {
157 157 // 3、调用 MidjourneyProxy 提交任务
158 158 MidjourneyImagineReqVO imagineReqVO = BeanUtils.toBean(req, MidjourneyImagineReqVO.class);
159 159 imagineReqVO.setNotifyHook(midjourneyNotifyUrl);
160   - imagineReqVO.setState(String.valueOf(aiImageDO.getId()));
  160 + // 设置 midjourney 扩展参数，通过 --ar 来设置尺寸
  161 + imagineReqVO.setState(String.format("--ar %s:%s", req.getWidth(), req.getHeight()));
161 162 MidjourneySubmitRespVO submitRespVO = midjourneyProxyClient.imagine(imagineReqVO);
162 163  
163 164 // 4、保存任务 id (状态码: 1(提交成功), 21(已存在), 22(排队中), other(错误))
... ...