From 7358084198dc544d1ea6944f2ed3948cf17de322 Mon Sep 17 00:00:00 2001 From: cherishsince Date: Wed, 12 Jun 2024 13:42:29 +0800 Subject: [PATCH] =?UTF-8?q?=E3=80=90=E4=BC=98=E5=8C=96=E3=80=91=E9=87=8D?= =?UTF-8?q?=E5=91=BD=E5=90=8D=20midjourney=20properties?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ...djourneyProperties.java => MidjourneyConfig.java} | 11 ++++++++++- .../ai/core/model/midjourney/api/MidjourneyApi.java | 12 ++++++------ 2 files changed, 16 insertions(+), 7 deletions(-) rename yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/midjourney/{MidjourneyProperties.java => MidjourneyConfig.java} (65%) diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/midjourney/MidjourneyProperties.java b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/midjourney/MidjourneyConfig.java similarity index 65% rename from yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/midjourney/MidjourneyProperties.java rename to yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/midjourney/MidjourneyConfig.java index fc381a1c4d..7a4dd9e1d6 100644 --- a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/midjourney/MidjourneyProperties.java +++ b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/midjourney/MidjourneyConfig.java @@ -10,9 +10,18 @@ import lombok.Data; * @since 1.0 */ @Data -public class MidjourneyProperties { +public class MidjourneyConfig { + /** + * keys + */ private String key; + /** + * 请求地址 + */ private String url; + /** + * 通知回调地址 + */ private String notifyUrl; } diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/midjourney/api/MidjourneyApi.java b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/midjourney/api/MidjourneyApi.java index ee5d40691b..aa0fbda001 100644 --- a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/midjourney/api/MidjourneyApi.java +++ b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/midjourney/api/MidjourneyApi.java @@ -1,6 +1,6 @@ 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 com.google.common.collect.ImmutableMap; 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_LIST_BY_CONDITION = "/task/list-by-condition"; private final WebClient webClient; - private final MidjourneyProperties midjourneyProperties; + private final MidjourneyConfig midjourneyConfig; - public MidjourneyApi(MidjourneyProperties midjourneyProperties) { - this.midjourneyProperties = midjourneyProperties; + public MidjourneyApi(MidjourneyConfig midjourneyConfig) { + this.midjourneyConfig = midjourneyConfig; this.webClient = WebClient.builder() - .baseUrl(midjourneyProperties.getUrl()) - .defaultHeaders(ApiUtils.getJsonContentHeaders(midjourneyProperties.getKey())) + .baseUrl(midjourneyConfig.getUrl()) + .defaultHeaders(ApiUtils.getJsonContentHeaders(midjourneyConfig.getKey())) .build(); }