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<String> 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());