移动目录

cherishsince
1 parent 98319f68
Showing 4 changed files with 18 additions and 18 deletions
-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());
+    }
  
 }
-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;
-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;
-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;