hhyykk/ipms-sjy
1
1
Fork 0
mirror of https://gitee.com/hhyykk/ipms-sjy.git synced 2025-10-29 09:18:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

【优化】重命名 midjourney properties

Browse Source
This commit is contained in:
cherishsince
2024-06-12 13:42:29 +08:00
parent 9ae25aa0b8
commit 7358084198
2 changed files with 16 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/midjourney/MidjourneyProperties.java → yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/midjourney/MidjourneyConfig.java
View File

@@ -10,9 +10,18 @@ import lombok.Data;
* @since 1.0 * @since 1.0
*/ */
@Data @Data
public class MidjourneyProperties { public class MidjourneyConfig {
/**
* keys
*/
private String key; private String key;
/**
* 请求地址
*/
private String url; private String url;
/**
* 通知回调地址
*/
private String notifyUrl; private String notifyUrl;
} }

12
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/midjourney/api/MidjourneyApi.java
View File

@@ -1,6 +1,6 @@
package cn.iocoder.yudao.framework.ai.core.model.midjourney.api; package cn.iocoder.yudao.framework.ai.core.model.midjourney.api;
import cn.iocoder.yudao.framework.ai.core.model.midjourney.MidjourneyProperties; import cn.iocoder.yudao.framework.ai.core.model.midjourney.MidjourneyConfig;
import cn.iocoder.yudao.framework.common.util.json.JsonUtils; import cn.iocoder.yudao.framework.common.util.json.JsonUtils;
import com.google.common.collect.ImmutableMap; import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Lists; import com.google.common.collect.Lists;
@@ -29,13 +29,13 @@ public class MidjourneyApi {
private static final String URI_ACTON = "/submit/action"; private static final String URI_ACTON = "/submit/action";
private static final String URI_LIST_BY_CONDITION = "/task/list-by-condition"; private static final String URI_LIST_BY_CONDITION = "/task/list-by-condition";
private final WebClient webClient; private final WebClient webClient;
private final MidjourneyProperties midjourneyProperties; private final MidjourneyConfig midjourneyConfig;
public MidjourneyApi(MidjourneyProperties midjourneyProperties) { public MidjourneyApi(MidjourneyConfig midjourneyConfig) {
this.midjourneyProperties = midjourneyProperties; this.midjourneyConfig = midjourneyConfig;
this.webClient = WebClient.builder() this.webClient = WebClient.builder()
.baseUrl(midjourneyProperties.getUrl()) .baseUrl(midjourneyConfig.getUrl())
.defaultHeaders(ApiUtils.getJsonContentHeaders(midjourneyProperties.getKey())) .defaultHeaders(ApiUtils.getJsonContentHeaders(midjourneyConfig.getKey()))
.build(); .build();
} }