From a2bd61947b618ca4d7b665bbc5f0e7744d893225 Mon Sep 17 00:00:00 2001 From: b_mountain <49973336+MRgenial@users.noreply.github.com> Date: Mon, 19 Aug 2024 00:48:34 +0800 Subject: [PATCH] [imporve] move ai package and improve ai code. (#2542) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: 刘进山 Co-authored-by: YuLuo Co-authored-by: linDong <56677297@qq.com> Co-authored-by: Calvin --- .../common/constants/AiTypeEnum.java | 13 ++-- .../manager/controller/AiController.java | 6 +- .../manager/service/{ => ai}/AiService.java | 2 +- .../{impl => ai}/AlibabaAiServiceImpl.java | 70 ++++++++----------- .../{impl => ai}/KimiAiServiceImpl.java | 44 +++++------- .../{impl => ai}/SparkDeskAiServiceImpl.java | 41 +++++------ .../{impl => ai}/ZhiPuServiceImpl.java | 42 +++++------ .../factory}/AiServiceFactoryImpl.java | 4 +- .../manager/controller/AiControllerTest.java | 4 +- .../manager/service/AiServiceFactoryTest.java | 3 +- .../manager/service/AlibabaAiServiceTest.java | 2 +- .../manager/service/KimiAiServiceTest.java | 2 +- .../service/SparkDeskAiServiceTest.java | 2 +- .../manager/service/ZhiPuServiceTest.java | 2 +- 14 files changed, 103 insertions(+), 134 deletions(-) rename manager/src/main/java/org/apache/hertzbeat/manager/service/{ => ai}/AiService.java (96%) rename manager/src/main/java/org/apache/hertzbeat/manager/service/{impl => ai}/AlibabaAiServiceImpl.java (64%) rename manager/src/main/java/org/apache/hertzbeat/manager/service/{impl => ai}/KimiAiServiceImpl.java (70%) rename manager/src/main/java/org/apache/hertzbeat/manager/service/{impl => ai}/SparkDeskAiServiceImpl.java (73%) rename manager/src/main/java/org/apache/hertzbeat/manager/service/{impl => ai}/ZhiPuServiceImpl.java (71%) rename manager/src/main/java/org/apache/hertzbeat/manager/service/{impl => ai/factory}/AiServiceFactoryImpl.java (95%) diff --git a/common/src/main/java/org/apache/hertzbeat/common/constants/AiTypeEnum.java b/common/src/main/java/org/apache/hertzbeat/common/constants/AiTypeEnum.java index 14e58decfbc..d1594033de1 100644 --- a/common/src/main/java/org/apache/hertzbeat/common/constants/AiTypeEnum.java +++ b/common/src/main/java/org/apache/hertzbeat/common/constants/AiTypeEnum.java @@ -17,6 +17,8 @@ package org.apache.hertzbeat.common.constants; +import java.util.Arrays; + /** * Ai type Enum */ @@ -48,13 +50,10 @@ public enum AiTypeEnum { * get type */ public static AiTypeEnum getTypeByName(String type) { - for (AiTypeEnum aiTypeEnum : values()) { - if (aiTypeEnum.name().equals(type)) { - return aiTypeEnum; - } - - } - return null; + return Arrays.stream(values()) + .filter(ai -> ai.name().equals(type)) + .findFirst() + .orElse(null); } } diff --git a/manager/src/main/java/org/apache/hertzbeat/manager/controller/AiController.java b/manager/src/main/java/org/apache/hertzbeat/manager/controller/AiController.java index 6c7408826d4..718d9c09410 100644 --- a/manager/src/main/java/org/apache/hertzbeat/manager/controller/AiController.java +++ b/manager/src/main/java/org/apache/hertzbeat/manager/controller/AiController.java @@ -22,8 +22,8 @@ import io.swagger.v3.oas.annotations.Parameter; import io.swagger.v3.oas.annotations.tags.Tag; import org.apache.hertzbeat.manager.config.AiProperties; -import org.apache.hertzbeat.manager.service.AiService; -import org.apache.hertzbeat.manager.service.impl.AiServiceFactoryImpl; +import org.apache.hertzbeat.manager.service.ai.AiService; +import org.apache.hertzbeat.manager.service.ai.factory.AiServiceFactoryImpl; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.http.codec.ServerSentEvent; import org.springframework.util.Assert; @@ -59,10 +59,8 @@ public class AiController { @Operation(summary = "Artificial intelligence questions and Answers", description = "Artificial intelligence questions and Answers") public Flux> requestAi(@Parameter(description = "Request text", example = "Who are you") @RequestParam("text") String text) { - Assert.notNull(aiServiceFactory, "please check that your type value is consistent with the documentation on the website"); AiService aiServiceImplBean = aiServiceFactory.getAiServiceImplBean(aiProperties.getType()); - return aiServiceImplBean.requestAi(text); } } diff --git a/manager/src/main/java/org/apache/hertzbeat/manager/service/AiService.java b/manager/src/main/java/org/apache/hertzbeat/manager/service/ai/AiService.java similarity index 96% rename from manager/src/main/java/org/apache/hertzbeat/manager/service/AiService.java rename to manager/src/main/java/org/apache/hertzbeat/manager/service/ai/AiService.java index 22ccd55d799..de8e683ed91 100644 --- a/manager/src/main/java/org/apache/hertzbeat/manager/service/AiService.java +++ b/manager/src/main/java/org/apache/hertzbeat/manager/service/ai/AiService.java @@ -15,7 +15,7 @@ * limitations under the License. */ -package org.apache.hertzbeat.manager.service; +package org.apache.hertzbeat.manager.service.ai; import org.apache.hertzbeat.common.constants.AiTypeEnum; diff --git a/manager/src/main/java/org/apache/hertzbeat/manager/service/impl/AlibabaAiServiceImpl.java b/manager/src/main/java/org/apache/hertzbeat/manager/service/ai/AlibabaAiServiceImpl.java similarity index 64% rename from manager/src/main/java/org/apache/hertzbeat/manager/service/impl/AlibabaAiServiceImpl.java rename to manager/src/main/java/org/apache/hertzbeat/manager/service/ai/AlibabaAiServiceImpl.java index 2cac8b4ef2b..0f0fc339be3 100644 --- a/manager/src/main/java/org/apache/hertzbeat/manager/service/impl/AlibabaAiServiceImpl.java +++ b/manager/src/main/java/org/apache/hertzbeat/manager/service/ai/AlibabaAiServiceImpl.java @@ -15,7 +15,7 @@ * limitations under the License. */ -package org.apache.hertzbeat.manager.service.impl; +package org.apache.hertzbeat.manager.service.ai; import java.util.List; import java.util.Objects; @@ -27,7 +27,6 @@ import org.apache.hertzbeat.manager.pojo.dto.AiMessage; import org.apache.hertzbeat.manager.pojo.dto.AliAiRequestParamDTO; import org.apache.hertzbeat.manager.pojo.dto.AliAiResponse; -import org.apache.hertzbeat.manager.service.AiService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; import org.springframework.http.HttpHeaders; @@ -77,46 +76,39 @@ public AiTypeEnum getType() { @Override public Flux> requestAi(String text) { checkParam(text, aiProperties.getModel(), aiProperties.getApiKey()); - try { - AliAiRequestParamDTO aliAiRequestParamDTO = AliAiRequestParamDTO.builder() - .model(aiProperties.getModel()) - .input(AliAiRequestParamDTO.Input.builder() - .messages(List.of(new AiMessage(AiConstants.AliAiConstants.REQUEST_ROLE, text))) - .build()) - .parameters(AliAiRequestParamDTO.Parameters.builder() - .maxTokens(AiConstants.AliAiConstants.MAX_TOKENS) - .temperature(AiConstants.AliAiConstants.TEMPERATURE) - .enableSearch(true) - .resultFormat("message") - .incrementalOutput(true) - .build()) - .build(); + AliAiRequestParamDTO aliAiRequestParamDTO = AliAiRequestParamDTO.builder() + .model(aiProperties.getModel()) + .input(AliAiRequestParamDTO.Input.builder() + .messages(List.of(new AiMessage(AiConstants.AliAiConstants.REQUEST_ROLE, text))) + .build()) + .parameters(AliAiRequestParamDTO.Parameters.builder() + .maxTokens(AiConstants.AliAiConstants.MAX_TOKENS) + .temperature(AiConstants.AliAiConstants.TEMPERATURE) + .enableSearch(true) + .resultFormat("message") + .incrementalOutput(true) + .build()) + .build(); - return webClient.post() - .body(BodyInserters.fromValue(aliAiRequestParamDTO)) - .retrieve() - .bodyToFlux(AliAiResponse.class) - .map(aliAiResponse -> { - if (Objects.nonNull(aliAiResponse)) { - List choices = aliAiResponse.getOutput().getChoices(); - if (CollectionUtils.isEmpty(choices)) { - return ServerSentEvent.builder().build(); - } - String content = choices.get(0).getMessage().getContent(); - return ServerSentEvent.builder() - .data(content) - .build(); + return webClient.post() + .body(BodyInserters.fromValue(aliAiRequestParamDTO)) + .retrieve() + .bodyToFlux(AliAiResponse.class) + .map(aliAiResponse -> { + if (Objects.nonNull(aliAiResponse)) { + List choices = aliAiResponse.getOutput().getChoices(); + if (CollectionUtils.isEmpty(choices)) { + return ServerSentEvent.builder().build(); } - return ServerSentEvent.builder().build(); - }) - .doOnError(error -> log.info("AiResponse Exception:{}", error.toString())); - - } catch (Exception e) { - log.info("KimiAiServiceImpl.requestAi exception:{}", e.toString()); - throw e; - } - + String content = choices.get(0).getMessage().getContent(); + return ServerSentEvent.builder() + .data(content) + .build(); + } + return ServerSentEvent.builder().build(); + }) + .doOnError(error -> log.info("AlibabaAiServiceImpl.requestAi exception:{}", error.getMessage())); } diff --git a/manager/src/main/java/org/apache/hertzbeat/manager/service/impl/KimiAiServiceImpl.java b/manager/src/main/java/org/apache/hertzbeat/manager/service/ai/KimiAiServiceImpl.java similarity index 70% rename from manager/src/main/java/org/apache/hertzbeat/manager/service/impl/KimiAiServiceImpl.java rename to manager/src/main/java/org/apache/hertzbeat/manager/service/ai/KimiAiServiceImpl.java index 29c46fa717c..031a33271b3 100644 --- a/manager/src/main/java/org/apache/hertzbeat/manager/service/impl/KimiAiServiceImpl.java +++ b/manager/src/main/java/org/apache/hertzbeat/manager/service/ai/KimiAiServiceImpl.java @@ -15,7 +15,7 @@ * limitations under the License. */ -package org.apache.hertzbeat.manager.service.impl; +package org.apache.hertzbeat.manager.service.ai; import java.util.List; import javax.annotation.PostConstruct; @@ -26,7 +26,6 @@ import org.apache.hertzbeat.manager.pojo.dto.AiMessage; import org.apache.hertzbeat.manager.pojo.dto.OpenAiRequestParamDTO; import org.apache.hertzbeat.manager.pojo.dto.OpenAiResponse; -import org.apache.hertzbeat.manager.service.AiService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; import org.springframework.http.HttpHeaders; @@ -71,37 +70,28 @@ public AiTypeEnum getType() { @Override public Flux> requestAi(String text) { - try { - checkParam(text, aiProperties.getModel(), aiProperties.getApiKey()); - OpenAiRequestParamDTO zhiPuRequestParamDTO = OpenAiRequestParamDTO.builder() - .model(aiProperties.getModel()) - .stream(Boolean.TRUE) - .maxTokens(AiConstants.KimiAiConstants.MAX_TOKENS) - .temperature(AiConstants.KimiAiConstants.TEMPERATURE) - .messages(List.of(new AiMessage(AiConstants.KimiAiConstants.REQUEST_ROLE, text))) - .build(); - - - return webClient.post() - .body(BodyInserters.fromValue(zhiPuRequestParamDTO)) - .retrieve() - .bodyToFlux(String.class) - .filter(aiResponse -> !"[DONE]".equals(aiResponse)) - .map(OpenAiResponse::convertToResponse) - .doOnError(error -> log.info("AiResponse Exception:{}", error.toString())); - - - } catch (Exception e) { - log.info("KimiAiServiceImpl.requestAi exception:{}", e.toString()); - throw e; - } + checkParam(text, aiProperties.getModel(), aiProperties.getApiKey()); + OpenAiRequestParamDTO zhiPuRequestParamDTO = OpenAiRequestParamDTO.builder() + .model(aiProperties.getModel()) + .stream(Boolean.TRUE) + .maxTokens(AiConstants.KimiAiConstants.MAX_TOKENS) + .temperature(AiConstants.KimiAiConstants.TEMPERATURE) + .messages(List.of(new AiMessage(AiConstants.KimiAiConstants.REQUEST_ROLE, text))) + .build(); + return webClient.post() + .body(BodyInserters.fromValue(zhiPuRequestParamDTO)) + .retrieve() + .bodyToFlux(String.class) + .filter(aiResponse -> !"[DONE]".equals(aiResponse)) + .map(OpenAiResponse::convertToResponse) + .doOnError(error -> log.info("KimiAiServiceImpl.requestAi exception:{}", error.getMessage())); } private void checkParam(String param, String model, String apiKey) { Assert.notNull(param, "text is null"); - Assert.notNull(param, "model is null"); + Assert.notNull(model, "model is null"); Assert.notNull(apiKey, "ai.api-key is null"); } } diff --git a/manager/src/main/java/org/apache/hertzbeat/manager/service/impl/SparkDeskAiServiceImpl.java b/manager/src/main/java/org/apache/hertzbeat/manager/service/ai/SparkDeskAiServiceImpl.java similarity index 73% rename from manager/src/main/java/org/apache/hertzbeat/manager/service/impl/SparkDeskAiServiceImpl.java rename to manager/src/main/java/org/apache/hertzbeat/manager/service/ai/SparkDeskAiServiceImpl.java index 3b1271d5111..4c2520f63a0 100644 --- a/manager/src/main/java/org/apache/hertzbeat/manager/service/impl/SparkDeskAiServiceImpl.java +++ b/manager/src/main/java/org/apache/hertzbeat/manager/service/ai/SparkDeskAiServiceImpl.java @@ -15,7 +15,7 @@ * limitations under the License. */ -package org.apache.hertzbeat.manager.service.impl; +package org.apache.hertzbeat.manager.service.ai; import java.util.List; import javax.annotation.PostConstruct; @@ -26,7 +26,6 @@ import org.apache.hertzbeat.manager.pojo.dto.AiMessage; import org.apache.hertzbeat.manager.pojo.dto.OpenAiRequestParamDTO; import org.apache.hertzbeat.manager.pojo.dto.OpenAiResponse; -import org.apache.hertzbeat.manager.service.AiService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; import org.springframework.http.HttpHeaders; @@ -78,33 +77,29 @@ public AiTypeEnum getType() { @Override public Flux> requestAi(String text) { + checkParam(text, aiProperties.getApiKey(), aiProperties.getModel()); + OpenAiRequestParamDTO zhiPuRequestParamDTO = OpenAiRequestParamDTO.builder() + .model(aiProperties.getModel()) + //sse + .stream(Boolean.TRUE) + .maxTokens(AiConstants.SparkDeskConstants.MAX_TOKENS) + .temperature(AiConstants.SparkDeskConstants.TEMPERATURE) + .messages(List.of(new AiMessage(AiConstants.SparkDeskConstants.REQUEST_ROLE, text))) + .build(); - try { - checkParam(text, aiProperties.getApiKey(), aiProperties.getModel()); - OpenAiRequestParamDTO zhiPuRequestParamDTO = OpenAiRequestParamDTO.builder() - .model(aiProperties.getModel()) - //sse - .stream(Boolean.TRUE) - .maxTokens(AiConstants.SparkDeskConstants.MAX_TOKENS) - .temperature(AiConstants.SparkDeskConstants.TEMPERATURE) - .messages(List.of(new AiMessage(AiConstants.SparkDeskConstants.REQUEST_ROLE, text))) - .build(); + return webClient.post() + .body(BodyInserters.fromValue(zhiPuRequestParamDTO)) + .retrieve() + .bodyToFlux(String.class) + .filter(aiResponse -> !"[DONE]".equals(aiResponse)) + .map(OpenAiResponse::convertToResponse) + .doOnError(error -> log.info("SparkDeskAiServiceImpl.requestAi exception:{}", error.getMessage())); - return webClient.post() - .body(BodyInserters.fromValue(zhiPuRequestParamDTO)) - .retrieve() - .bodyToFlux(String.class) - .filter(aiResponse -> !"[DONE]".equals(aiResponse)) - .map(OpenAiResponse::convertToResponse); - } catch (Exception e) { - log.info("SparkDeskAiServiceImpl.requestAi exception:{}", e.toString()); - throw e; - } } private void checkParam(String param, String apiKey, String model) { Assert.notNull(param, "text is null"); - Assert.notNull(param, "model is null"); + Assert.notNull(model, "model is null"); Assert.notNull(apiKey, "ai.api-key is null"); } } diff --git a/manager/src/main/java/org/apache/hertzbeat/manager/service/impl/ZhiPuServiceImpl.java b/manager/src/main/java/org/apache/hertzbeat/manager/service/ai/ZhiPuServiceImpl.java similarity index 71% rename from manager/src/main/java/org/apache/hertzbeat/manager/service/impl/ZhiPuServiceImpl.java rename to manager/src/main/java/org/apache/hertzbeat/manager/service/ai/ZhiPuServiceImpl.java index b6111c24720..75a55f1b8cd 100644 --- a/manager/src/main/java/org/apache/hertzbeat/manager/service/impl/ZhiPuServiceImpl.java +++ b/manager/src/main/java/org/apache/hertzbeat/manager/service/ai/ZhiPuServiceImpl.java @@ -15,7 +15,7 @@ * limitations under the License. */ -package org.apache.hertzbeat.manager.service.impl; +package org.apache.hertzbeat.manager.service.ai; import java.util.List; @@ -27,7 +27,6 @@ import org.apache.hertzbeat.manager.pojo.dto.AiMessage; import org.apache.hertzbeat.manager.pojo.dto.OpenAiRequestParamDTO; import org.apache.hertzbeat.manager.pojo.dto.OpenAiResponse; -import org.apache.hertzbeat.manager.service.AiService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; import org.springframework.http.HttpHeaders; @@ -73,33 +72,28 @@ public AiTypeEnum getType() { @Override public Flux> requestAi(String text) { - try { - checkParam(text, aiProperties.getModel(), aiProperties.getApiKey()); - OpenAiRequestParamDTO zhiPuRequestParamDTO = OpenAiRequestParamDTO.builder() - .model(aiProperties.getModel()) - //sse - .stream(Boolean.TRUE) - .maxTokens(AiConstants.ZhiPuConstants.MAX_TOKENS) - .temperature(AiConstants.ZhiPuConstants.TEMPERATURE) - .messages(List.of(new AiMessage(AiConstants.ZhiPuConstants.REQUEST_ROLE, text))) - .build(); - - return webClient.post() - .body(BodyInserters.fromValue(zhiPuRequestParamDTO)) - .retrieve() - .bodyToFlux(String.class) - .filter(aiResponse -> !"[DONE]".equals(aiResponse)) - .map(OpenAiResponse::convertToResponse) - .doOnError(error -> log.info("AiResponse Exception:{}", error.toString())); + checkParam(text, aiProperties.getModel(), aiProperties.getApiKey()); + OpenAiRequestParamDTO zhiPuRequestParamDTO = OpenAiRequestParamDTO.builder() + .model(aiProperties.getModel()) + //sse + .stream(Boolean.TRUE) + .maxTokens(AiConstants.ZhiPuConstants.MAX_TOKENS) + .temperature(AiConstants.ZhiPuConstants.TEMPERATURE) + .messages(List.of(new AiMessage(AiConstants.ZhiPuConstants.REQUEST_ROLE, text))) + .build(); - } catch (Exception e) { - log.info("ZhiPuServiceImpl.requestAi exception:{}", e.toString()); - throw e; - } + return webClient.post() + .body(BodyInserters.fromValue(zhiPuRequestParamDTO)) + .retrieve() + .bodyToFlux(String.class) + .filter(aiResponse -> !"[DONE]".equals(aiResponse)) + .map(OpenAiResponse::convertToResponse) + .doOnError(error -> log.info("ZhiPuServiceImpl.requestAi exception:{}", error.getMessage())); } private void checkParam(String param, String model, String apiKey) { Assert.notNull(param, "text is null"); + Assert.notNull(model, "model is null"); Assert.notNull(apiKey, "ai.api-key is null"); } diff --git a/manager/src/main/java/org/apache/hertzbeat/manager/service/impl/AiServiceFactoryImpl.java b/manager/src/main/java/org/apache/hertzbeat/manager/service/ai/factory/AiServiceFactoryImpl.java similarity index 95% rename from manager/src/main/java/org/apache/hertzbeat/manager/service/impl/AiServiceFactoryImpl.java rename to manager/src/main/java/org/apache/hertzbeat/manager/service/ai/factory/AiServiceFactoryImpl.java index 8a18e0ce1b5..8e0954c7adb 100644 --- a/manager/src/main/java/org/apache/hertzbeat/manager/service/impl/AiServiceFactoryImpl.java +++ b/manager/src/main/java/org/apache/hertzbeat/manager/service/ai/factory/AiServiceFactoryImpl.java @@ -15,7 +15,7 @@ * limitations under the License. */ -package org.apache.hertzbeat.manager.service.impl; +package org.apache.hertzbeat.manager.service.ai.factory; import java.util.HashMap; import java.util.List; @@ -24,7 +24,7 @@ import java.util.stream.Collectors; import javax.annotation.PostConstruct; import org.apache.hertzbeat.common.constants.AiTypeEnum; -import org.apache.hertzbeat.manager.service.AiService; +import org.apache.hertzbeat.manager.service.ai.AiService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; import org.springframework.stereotype.Component; diff --git a/manager/src/test/java/org/apache/hertzbeat/manager/controller/AiControllerTest.java b/manager/src/test/java/org/apache/hertzbeat/manager/controller/AiControllerTest.java index 7c1bfa5f27e..70f3e150e41 100644 --- a/manager/src/test/java/org/apache/hertzbeat/manager/controller/AiControllerTest.java +++ b/manager/src/test/java/org/apache/hertzbeat/manager/controller/AiControllerTest.java @@ -18,8 +18,8 @@ package org.apache.hertzbeat.manager.controller; import org.apache.hertzbeat.manager.config.AiProperties; -import org.apache.hertzbeat.manager.service.AiService; -import org.apache.hertzbeat.manager.service.impl.AiServiceFactoryImpl; +import org.apache.hertzbeat.manager.service.ai.AiService; +import org.apache.hertzbeat.manager.service.ai.factory.AiServiceFactoryImpl; import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.Test; import org.junit.jupiter.api.extension.ExtendWith; diff --git a/manager/src/test/java/org/apache/hertzbeat/manager/service/AiServiceFactoryTest.java b/manager/src/test/java/org/apache/hertzbeat/manager/service/AiServiceFactoryTest.java index 067585ce159..b0ac2089339 100644 --- a/manager/src/test/java/org/apache/hertzbeat/manager/service/AiServiceFactoryTest.java +++ b/manager/src/test/java/org/apache/hertzbeat/manager/service/AiServiceFactoryTest.java @@ -24,7 +24,8 @@ import java.util.function.Function; import java.util.stream.Collectors; import org.apache.hertzbeat.common.constants.AiTypeEnum; -import org.apache.hertzbeat.manager.service.impl.AiServiceFactoryImpl; +import org.apache.hertzbeat.manager.service.ai.AiService; +import org.apache.hertzbeat.manager.service.ai.factory.AiServiceFactoryImpl; import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.Test; import org.junit.jupiter.api.extension.ExtendWith; diff --git a/manager/src/test/java/org/apache/hertzbeat/manager/service/AlibabaAiServiceTest.java b/manager/src/test/java/org/apache/hertzbeat/manager/service/AlibabaAiServiceTest.java index ac6b7bbccec..ff0a5b0f331 100644 --- a/manager/src/test/java/org/apache/hertzbeat/manager/service/AlibabaAiServiceTest.java +++ b/manager/src/test/java/org/apache/hertzbeat/manager/service/AlibabaAiServiceTest.java @@ -17,7 +17,7 @@ package org.apache.hertzbeat.manager.service; -import org.apache.hertzbeat.manager.service.impl.AlibabaAiServiceImpl; +import org.apache.hertzbeat.manager.service.ai.AlibabaAiServiceImpl; /** * test case for {@link AlibabaAiServiceImpl} diff --git a/manager/src/test/java/org/apache/hertzbeat/manager/service/KimiAiServiceTest.java b/manager/src/test/java/org/apache/hertzbeat/manager/service/KimiAiServiceTest.java index 30f0eb790f9..a747c6f2ad9 100644 --- a/manager/src/test/java/org/apache/hertzbeat/manager/service/KimiAiServiceTest.java +++ b/manager/src/test/java/org/apache/hertzbeat/manager/service/KimiAiServiceTest.java @@ -18,7 +18,7 @@ package org.apache.hertzbeat.manager.service; -import org.apache.hertzbeat.manager.service.impl.KimiAiServiceImpl; +import org.apache.hertzbeat.manager.service.ai.KimiAiServiceImpl; /** * test case for {@link KimiAiServiceImpl} diff --git a/manager/src/test/java/org/apache/hertzbeat/manager/service/SparkDeskAiServiceTest.java b/manager/src/test/java/org/apache/hertzbeat/manager/service/SparkDeskAiServiceTest.java index f8dd26a998b..47d281bd778 100644 --- a/manager/src/test/java/org/apache/hertzbeat/manager/service/SparkDeskAiServiceTest.java +++ b/manager/src/test/java/org/apache/hertzbeat/manager/service/SparkDeskAiServiceTest.java @@ -17,7 +17,7 @@ package org.apache.hertzbeat.manager.service; -import org.apache.hertzbeat.manager.service.impl.SparkDeskAiServiceImpl; +import org.apache.hertzbeat.manager.service.ai.SparkDeskAiServiceImpl; /** * test case for {@link SparkDeskAiServiceImpl} diff --git a/manager/src/test/java/org/apache/hertzbeat/manager/service/ZhiPuServiceTest.java b/manager/src/test/java/org/apache/hertzbeat/manager/service/ZhiPuServiceTest.java index 68329d472b4..12fb3fd10b4 100644 --- a/manager/src/test/java/org/apache/hertzbeat/manager/service/ZhiPuServiceTest.java +++ b/manager/src/test/java/org/apache/hertzbeat/manager/service/ZhiPuServiceTest.java @@ -17,7 +17,7 @@ package org.apache.hertzbeat.manager.service; -import org.apache.hertzbeat.manager.service.impl.ZhiPuServiceImpl; +import org.apache.hertzbeat.manager.service.ai.ZhiPuServiceImpl; /** * test case for {@link ZhiPuServiceImpl}