diff --git a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/ChatController.java b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/chat/ChatController.java
index 8fbc6d9..90710a3 100644
--- a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/ChatController.java
+++ b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/chat/ChatController.java
@@ -1,14 +1,15 @@
-package cn.iocoder.yudao.module.ai.controller.admin;
+package cn.iocoder.yudao.module.ai.controller.admin.chat;
 
-import cn.iocoder.yudao.framework.ai.chatqianwen.QianWenChatClient;
+import cn.iocoder.yudao.framework.ai.chat.ChatResponse;
+import cn.iocoder.yudao.framework.ai.chat.prompt.Prompt;
+import cn.iocoder.yudao.framework.ai.config.AiClient;
+import cn.iocoder.yudao.framework.ai.config.YudaoAiClient;
+import cn.iocoder.yudao.framework.common.pojo.CommonResult;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import lombok.AllArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RequestParam;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
 
 /**
  * @author fansili
@@ -17,18 +18,17 @@ import org.springframework.web.bind.annotation.RestController;
  */
 @Tag(name = "AI模块")
 @RestController
-@RequestMapping("/ai-api")
+@RequestMapping("/admin-api/ai")
 @Slf4j
 @AllArgsConstructor
 public class ChatController {
 
-//
-//    @Autowired
-//    private QianWenChatClient qianWenChatClient;
-//
-//    @GetMapping("/chat")
-//    public String chat(@RequestParam("prompt") String prompt) {
-//        return qianWenChatClient.call(prompt);
-//    }
+    private final AiClient aiClient;
+
+    @PostMapping("/chat")
+    public CommonResult<String> chat(@RequestParam("prompt") String prompt) {
+        ChatResponse callRes = aiClient.call(new Prompt(prompt), "qianWen");
+        return CommonResult.success(callRes.getResult().getOutput().getContent());
+    }
 
 }
diff --git a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/app/vo/ChatRoleAddReq.java b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/app/chat/vo/ChatRoleAddReq.java
index b3957fa..b1c08e2 100644
--- a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/app/vo/ChatRoleAddReq.java
+++ b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/app/chat/vo/ChatRoleAddReq.java
@@ -1,4 +1,4 @@
-package cn.iocoder.yudao.module.ai.controller.app.vo;
+package cn.iocoder.yudao.module.ai.controller.app.chat.vo;
 
 import cn.iocoder.yudao.framework.common.pojo.PageParam;
 import io.swagger.v3.oas.annotations.media.Schema;
diff --git a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/app/vo/ChatRoleListReq.java b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/app/chat/vo/ChatRoleListReq.java
index 355f00c..18f3d46 100644
--- a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/app/vo/ChatRoleListReq.java
+++ b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/app/chat/vo/ChatRoleListReq.java
@@ -1,4 +1,4 @@
-package cn.iocoder.yudao.module.ai.controller.app.vo;
+package cn.iocoder.yudao.module.ai.controller.app.chat.vo;
 
 import cn.iocoder.yudao.framework.common.pojo.PageParam;
 import io.swagger.v3.oas.annotations.media.Schema;
diff --git a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/app/vo/ChatRoleListRes.java b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/app/chat/vo/ChatRoleListRes.java
index d788e78..6480296 100644
--- a/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/app/vo/ChatRoleListRes.java
+++ b/yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/app/chat/vo/ChatRoleListRes.java
@@ -1,4 +1,4 @@
-package cn.iocoder.yudao.module.ai.controller.app.vo;
+package cn.iocoder.yudao.module.ai.controller.app.chat.vo;
 
 import lombok.Data;
 import lombok.experimental.Accessors;