mirror of
https://gitee.com/hhyykk/ipms-sjy.git
synced 2025-10-31 18:28:43 +08:00
优化自动注入，可以创建多个 client
This commit is contained in:
cherishsince
@@ -1,77 +1,34 @@
|
||||
//package cn.iocoder.yudao.module.ai.controller.admin;
|
||||
//
|
||||
//import cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil;
|
||||
//import cn.iocoder.yudao.framework.common.pojo.CommonResult;
|
||||
//import cn.iocoder.yudao.module.ai.ErrorCodeConstants;
|
||||
//import cn.iocoder.yudao.module.ai.controller.admin.vo.AiChatReqVO;
|
||||
//import cn.iocoder.yudao.module.ai.enums.OpenAiModelEnum;
|
||||
//import io.swagger.v3.oas.annotations.Operation;
|
||||
//import io.swagger.v3.oas.annotations.tags.Tag;
|
||||
//import jakarta.servlet.http.HttpServletResponse;
|
||||
//import lombok.extern.slf4j.Slf4j;
|
||||
//import org.springframework.ai.chat.ChatClient;
|
||||
//import org.springframework.ai.chat.ChatResponse;
|
||||
//import org.springframework.ai.chat.prompt.Prompt;
|
||||
//import org.springframework.ai.openai.OpenAiChatClient;
|
||||
//import org.springframework.beans.factory.annotation.Autowired;
|
||||
//import org.springframework.context.ApplicationContext;
|
||||
//import org.springframework.validation.annotation.Validated;
|
||||
//import org.springframework.web.bind.annotation.PostMapping;
|
||||
//import org.springframework.web.bind.annotation.RequestBody;
|
||||
//import org.springframework.web.bind.annotation.RequestMapping;
|
||||
//import org.springframework.web.bind.annotation.RestController;
|
||||
//import reactor.core.publisher.Flux;
|
||||
//
|
||||
//import java.util.function.Consumer;
|
||||
//
|
||||
//// TODO done @fansili：有了 swagger 注释，就不用类注释了
|
||||
//@Tag(name = "AI模块")
|
||||
//@RestController
|
||||
//@RequestMapping("/ai-api")
|
||||
//@Slf4j
|
||||
//public class ChatController {
|
||||
package cn.iocoder.yudao.module.ai.controller.admin;
|
||||
|
||||
import cn.iocoder.yudao.framework.ai.chatqianwen.QianWenChatClient;
|
||||
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;
|
||||
|
||||
/**
|
||||
* @author fansili
|
||||
* @since 1.0
|
||||
* @time 2024/4/13 17:44
|
||||
*/
|
||||
@Tag(name = "AI模块")
|
||||
@RestController
|
||||
@RequestMapping("/ai-api")
|
||||
@Slf4j
|
||||
@AllArgsConstructor
|
||||
public class ChatController {
|
||||
|
||||
//
|
||||
// @Autowired
|
||||
// private ApplicationContext applicationContext;
|
||||
// private QianWenChatClient qianWenChatClient;
|
||||
//
|
||||
// @PostMapping("/chat")
|
||||
// @Operation(summary = "对话聊天", description = "简单的ai聊天")
|
||||
// public CommonResult chat(@RequestBody @Validated AiChatReqVO reqVO) {
|
||||
// ChatClient chatClient = getChatClient(reqVO.getAiModel());
|
||||
// String res;
|
||||
// try {
|
||||
// res = chatClient.call(reqVO.getPrompt());
|
||||
// } catch (Exception e) {
|
||||
// res = e.getMessage();
|
||||
// }
|
||||
// return CommonResult.success(res);
|
||||
// @GetMapping("/chat")
|
||||
// public String chat(@RequestParam("prompt") String prompt) {
|
||||
// return qianWenChatClient.call(prompt);
|
||||
// }
|
||||
//
|
||||
// @PostMapping("/chatStream")
|
||||
// @Operation(summary = "对话聊天chatStream", description = "简单的ai聊天")
|
||||
// public CommonResult chatStream(HttpServletResponse response, @RequestBody @Validated AiChatReqVO reqVO) throws InterruptedException {
|
||||
// OpenAiChatClient chatClient = applicationContext.getBean(OpenAiChatClient.class);
|
||||
// Flux<ChatResponse> chatResponse = chatClient.stream(new Prompt(reqVO.getPrompt()));
|
||||
// chatResponse.subscribe(new Consumer<ChatResponse>() {
|
||||
// @Override
|
||||
// public void accept(ChatResponse chatResponse) {
|
||||
// System.err.println(chatResponse.getResults().get(0).getOutput().getContent());
|
||||
// }
|
||||
// });
|
||||
// return CommonResult.success(null);
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 根据 ai模型 获取对于的 模型实现类
|
||||
// *
|
||||
// * @param aiModelEnum
|
||||
// * @return
|
||||
// */
|
||||
// private ChatClient getChatClient(OpenAiModelEnum aiModelEnum) {
|
||||
// if (OpenAiModelEnum.OPEN_AI_GPT_3_5 == aiModelEnum) {
|
||||
// return applicationContext.getBean(OpenAiChatClient.class);
|
||||
// }
|
||||
// // AI模型暂不支持
|
||||
// throw ServiceExceptionUtil.exception(ErrorCodeConstants.AI_MODULE_NOT_SUPPORTED);
|
||||
// }
|
||||
//}
|
||||
|
||||
}
|
||||
|
||||
@@ -0,0 +1,77 @@
|
||||
//package cn.iocoder.yudao.module.ai.controller.admin;
|
||||
//
|
||||
//import cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil;
|
||||
//import cn.iocoder.yudao.framework.common.pojo.CommonResult;
|
||||
//import cn.iocoder.yudao.module.ai.ErrorCodeConstants;
|
||||
//import cn.iocoder.yudao.module.ai.controller.admin.vo.AiChatReqVO;
|
||||
//import cn.iocoder.yudao.module.ai.enums.OpenAiModelEnum;
|
||||
//import io.swagger.v3.oas.annotations.Operation;
|
||||
//import io.swagger.v3.oas.annotations.tags.Tag;
|
||||
//import jakarta.servlet.http.HttpServletResponse;
|
||||
//import lombok.extern.slf4j.Slf4j;
|
||||
//import org.springframework.ai.chat.ChatClient;
|
||||
//import org.springframework.ai.chat.ChatResponse;
|
||||
//import org.springframework.ai.chat.prompt.Prompt;
|
||||
//import org.springframework.ai.openai.OpenAiChatClient;
|
||||
//import org.springframework.beans.factory.annotation.Autowired;
|
||||
//import org.springframework.context.ApplicationContext;
|
||||
//import org.springframework.validation.annotation.Validated;
|
||||
//import org.springframework.web.bind.annotation.PostMapping;
|
||||
//import org.springframework.web.bind.annotation.RequestBody;
|
||||
//import org.springframework.web.bind.annotation.RequestMapping;
|
||||
//import org.springframework.web.bind.annotation.RestController;
|
||||
//import reactor.core.publisher.Flux;
|
||||
//
|
||||
//import java.util.function.Consumer;
|
||||
//
|
||||
//// TODO done @fansili：有了 swagger 注释，就不用类注释了
|
||||
//@Tag(name = "AI模块")
|
||||
//@RestController
|
||||
//@RequestMapping("/ai-api")
|
||||
//@Slf4j
|
||||
//public class ChatController {
|
||||
//
|
||||
// @Autowired
|
||||
// private ApplicationContext applicationContext;
|
||||
//
|
||||
// @PostMapping("/chat")
|
||||
// @Operation(summary = "对话聊天", description = "简单的ai聊天")
|
||||
// public CommonResult chat(@RequestBody @Validated AiChatReqVO reqVO) {
|
||||
// ChatClient chatClient = getChatClient(reqVO.getAiModel());
|
||||
// String res;
|
||||
// try {
|
||||
// res = chatClient.call(reqVO.getPrompt());
|
||||
// } catch (Exception e) {
|
||||
// res = e.getMessage();
|
||||
// }
|
||||
// return CommonResult.success(res);
|
||||
// }
|
||||
//
|
||||
// @PostMapping("/chatStream")
|
||||
// @Operation(summary = "对话聊天chatStream", description = "简单的ai聊天")
|
||||
// public CommonResult chatStream(HttpServletResponse response, @RequestBody @Validated AiChatReqVO reqVO) throws InterruptedException {
|
||||
// OpenAiChatClient chatClient = applicationContext.getBean(OpenAiChatClient.class);
|
||||
// Flux<ChatResponse> chatResponse = chatClient.stream(new Prompt(reqVO.getPrompt()));
|
||||
// chatResponse.subscribe(new Consumer<ChatResponse>() {
|
||||
// @Override
|
||||
// public void accept(ChatResponse chatResponse) {
|
||||
// System.err.println(chatResponse.getResults().get(0).getOutput().getContent());
|
||||
// }
|
||||
// });
|
||||
// return CommonResult.success(null);
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 根据 ai模型 获取对于的 模型实现类
|
||||
// *
|
||||
// * @param aiModelEnum
|
||||
// * @return
|
||||
// */
|
||||
// private ChatClient getChatClient(OpenAiModelEnum aiModelEnum) {
|
||||
// if (OpenAiModelEnum.OPEN_AI_GPT_3_5 == aiModelEnum) {
|
||||
// return applicationContext.getBean(OpenAiChatClient.class);
|
||||
// }
|
||||
// // AI模型暂不支持
|
||||
// throw ServiceExceptionUtil.exception(ErrorCodeConstants.AI_MODULE_NOT_SUPPORTED);
|
||||
// }
|
||||
//}
|
||||
Reference in New Issue
Block a user