From 68f3f0d050f21ff4b18a7461e6123be0fa199405 Mon Sep 17 00:00:00 2001 From: huangge1199 Date: Wed, 14 May 2025 13:31:44 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=EF=BC=9AAI=E5=A4=A7=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B=E6=8E=A5=E5=85=A5-sdk=E6=8E=A5=E5=85=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../aiagent/Service/InvokeService.java | 15 +++++++++++++++ .../impl/InvokeServiceImpl.java} | 16 ++++++++-------- .../controller/invoke/InvokeController.java | 6 +++--- 3 files changed, 26 insertions(+), 11 deletions(-) create mode 100644 src/main/java/com/huangge1199/aiagent/Service/InvokeService.java rename src/main/java/com/huangge1199/aiagent/{demo/invoke/SdkAiInvoke.java => Service/impl/InvokeServiceImpl.java} (81%) diff --git a/src/main/java/com/huangge1199/aiagent/Service/InvokeService.java b/src/main/java/com/huangge1199/aiagent/Service/InvokeService.java new file mode 100644 index 0000000..3fc2e73 --- /dev/null +++ b/src/main/java/com/huangge1199/aiagent/Service/InvokeService.java @@ -0,0 +1,15 @@ +package com.huangge1199.aiagent.Service; + +import com.alibaba.dashscope.aigc.generation.GenerationResult; +import com.alibaba.dashscope.exception.InputRequiredException; +import com.alibaba.dashscope.exception.NoApiKeyException; + +/** + * InvokeService + * + * @author huangge1199 + * @since 2025/5/14 13:27:08 + */ +public interface InvokeService { + GenerationResult callWithMessage() throws NoApiKeyException, InputRequiredException; +} diff --git a/src/main/java/com/huangge1199/aiagent/demo/invoke/SdkAiInvoke.java b/src/main/java/com/huangge1199/aiagent/Service/impl/InvokeServiceImpl.java similarity index 81% rename from src/main/java/com/huangge1199/aiagent/demo/invoke/SdkAiInvoke.java rename to src/main/java/com/huangge1199/aiagent/Service/impl/InvokeServiceImpl.java index f618d79..b44370b 100644 --- a/src/main/java/com/huangge1199/aiagent/demo/invoke/SdkAiInvoke.java +++ b/src/main/java/com/huangge1199/aiagent/Service/impl/InvokeServiceImpl.java @@ -1,18 +1,17 @@ -package com.huangge1199.aiagent.demo.invoke; +package com.huangge1199.aiagent.Service.impl; import java.util.Arrays; -import java.lang.System; + import com.alibaba.dashscope.aigc.generation.Generation; import com.alibaba.dashscope.aigc.generation.GenerationParam; import com.alibaba.dashscope.aigc.generation.GenerationResult; import com.alibaba.dashscope.common.Message; import com.alibaba.dashscope.common.Role; -import com.alibaba.dashscope.exception.ApiException; import com.alibaba.dashscope.exception.InputRequiredException; import com.alibaba.dashscope.exception.NoApiKeyException; -import com.alibaba.dashscope.utils.JsonUtils; +import com.huangge1199.aiagent.Service.InvokeService; import org.springframework.beans.factory.annotation.Value; -import org.springframework.stereotype.Component; +import org.springframework.stereotype.Service; /** * SdkAiInvoke @@ -20,13 +19,14 @@ import org.springframework.stereotype.Component; * @author huangge1199 * @since 2025/5/14 12:27:47 */ -@Component -public class SdkAiInvoke { +@Service +public class InvokeServiceImpl implements InvokeService { @Value("${bailian.API-KEY}") private String baiLianKey; - public GenerationResult callWithMessage() throws ApiException, NoApiKeyException, InputRequiredException { + @Override + public GenerationResult callWithMessage() throws NoApiKeyException, InputRequiredException { Generation gen = new Generation(); Message systemMsg = Message.builder() .role(Role.SYSTEM.getValue()) diff --git a/src/main/java/com/huangge1199/aiagent/controller/invoke/InvokeController.java b/src/main/java/com/huangge1199/aiagent/controller/invoke/InvokeController.java index 67e87fa..b4bcba7 100644 --- a/src/main/java/com/huangge1199/aiagent/controller/invoke/InvokeController.java +++ b/src/main/java/com/huangge1199/aiagent/controller/invoke/InvokeController.java @@ -5,8 +5,8 @@ import com.alibaba.dashscope.exception.ApiException; import com.alibaba.dashscope.exception.InputRequiredException; import com.alibaba.dashscope.exception.NoApiKeyException; import com.alibaba.dashscope.utils.JsonUtils; +import com.huangge1199.aiagent.Service.InvokeService; import com.huangge1199.aiagent.common.R; -import com.huangge1199.aiagent.demo.invoke.SdkAiInvoke; import io.swagger.v3.oas.annotations.Operation; import io.swagger.v3.oas.annotations.tags.Tag; import jakarta.annotation.Resource; @@ -26,13 +26,13 @@ import org.springframework.web.bind.annotation.RestController; public class InvokeController { @Resource - private SdkAiInvoke sdkAiInvoke; + private InvokeService invokeService; @PostMapping("/sdk") @Operation(summary = "sdk接入") public R sdkAiInvoke() { try { - GenerationResult result = sdkAiInvoke.callWithMessage(); + GenerationResult result = invokeService.callWithMessage(); return R.ok(JsonUtils.toJson(result)); } catch (ApiException | NoApiKeyException | InputRequiredException e) { return R.fail(e.getMessage());