hhyykk/ipms-sjy
1
1
Fork 0
mirror of https://gitee.com/hhyykk/ipms-sjy.git synced 2025-02-02 03:34:58 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

【代码评审】AI：AI 绘图的接入

Browse Source
This commit is contained in:
YunaiV 2024-06-17 20:06:56 +08:00
parent 7c1e949d77
commit a331ed546d
4 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/AiImageController.java
View File

@ -85,7 +85,7 @@ public class AiImageController {
public CommonResult<Boolean> midjourneyAction(@RequestParam("id") Long imageId, public CommonResult<Boolean> midjourneyAction(@RequestParam("id") Long imageId,
@RequestParam("customId") String customId) { @RequestParam("customId") String customId) {
imageService.midjourneyAction(getLoginUserId(), imageId, customId); imageService.midjourneyAction(getLoginUserId(), imageId, customId);
return success(null); return success(true);
} }
} }

1
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/image/AiImageDO.java
View File

@ -102,6 +102,7 @@ public class AiImageDO extends BaseDO {
@TableField(typeHandler = JacksonTypeHandler.class) @TableField(typeHandler = JacksonTypeHandler.class)
private MidjourneyNotifyReqVO response; private MidjourneyNotifyReqVO response;
// TODO @fan：这个建议 Double
/** /**
* mj 进度(10%、50%、100%) * mj 进度(10%、50%、100%)
*/ */

2
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/job/MidjourneyJob.java
View File

@ -29,7 +29,7 @@ import java.util.Map;
public class MidjourneyJob implements JobHandler { public class MidjourneyJob implements JobHandler {
// TODO @fan：@Resource // TODO @fan：@Resource
@Autowired @Autowired(required = false)
private MidjourneyApi midjourneyApi; private MidjourneyApi midjourneyApi;
@Autowired @Autowired
private AiImageMapper imageMapper; private AiImageMapper imageMapper;

2
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/image/AiImageServiceImpl.java
View File

@ -55,7 +55,7 @@ public class AiImageServiceImpl implements AiImageService {
private FileApi fileApi; private FileApi fileApi;
@Resource @Resource
private AiApiKeyService apiKeyService; private AiApiKeyService apiKeyService;
@Autowired @Autowired(required = false)
private MidjourneyApi midjourneyApi; private MidjourneyApi midjourneyApi;
@Value("${ai.midjourney-proxy.notifyUrl:http://127.0.0.1:48080/admin-api/ai/image/midjourney-notify}") @Value("${ai.midjourney-proxy.notifyUrl:http://127.0.0.1:48080/admin-api/ai/image/midjourney-notify}")
private String midjourneyNotifyUrl; private String midjourneyNotifyUrl;