huangge1199/vue-pro
1
0
Fork 0
mirror of https://gitee.com/huangge1199_admin/vue-pro.git synced 2025-02-17 09:40:34 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

注释 chat controller，现在没用了，需要修改

Browse Source
This commit is contained in:
cherishsince 2024-04-11 21:51:54 +08:00
parent 238ef89826
commit 1ea94d9245
1 changed files with 77 additions and 77 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

154
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/ChatController.java
View File

@ -1,77 +1,77 @@
package cn.iocoder.yudao.module.ai.controller.admin; //package cn.iocoder.yudao.module.ai.controller.admin;
//
import cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil; //import cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil;
import cn.iocoder.yudao.framework.common.pojo.CommonResult; //import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.module.ai.ErrorCodeConstants; //import cn.iocoder.yudao.module.ai.ErrorCodeConstants;
import cn.iocoder.yudao.module.ai.controller.admin.vo.AiChatReqVO; //import cn.iocoder.yudao.module.ai.controller.admin.vo.AiChatReqVO;
import cn.iocoder.yudao.module.ai.enums.OpenAiModelEnum; //import cn.iocoder.yudao.module.ai.enums.OpenAiModelEnum;
import io.swagger.v3.oas.annotations.Operation; //import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag; //import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.servlet.http.HttpServletResponse; //import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j; //import lombok.extern.slf4j.Slf4j;
import org.springframework.ai.chat.ChatClient; //import org.springframework.ai.chat.ChatClient;
import org.springframework.ai.chat.ChatResponse; //import org.springframework.ai.chat.ChatResponse;
import org.springframework.ai.chat.prompt.Prompt; //import org.springframework.ai.chat.prompt.Prompt;
import org.springframework.ai.openai.OpenAiChatClient; //import org.springframework.ai.openai.OpenAiChatClient;
import org.springframework.beans.factory.annotation.Autowired; //import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext; //import org.springframework.context.ApplicationContext;
import org.springframework.validation.annotation.Validated; //import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.PostMapping; //import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody; //import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping; //import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController; //import org.springframework.web.bind.annotation.RestController;
import reactor.core.publisher.Flux; //import reactor.core.publisher.Flux;
//
import java.util.function.Consumer; //import java.util.function.Consumer;
//
// TODO done @fansili：有了 swagger 注释，就不用类注释了 //// TODO done @fansili：有了 swagger 注释，就不用类注释了
@Tag(name = "AI模块") //@Tag(name = "AI模块")
@RestController //@RestController
@RequestMapping("/ai-api") //@RequestMapping("/ai-api")
@Slf4j //@Slf4j
public class ChatController { //public class ChatController {
//
@Autowired // @Autowired
private ApplicationContext applicationContext; // private ApplicationContext applicationContext;
//
@PostMapping("/chat") // @PostMapping("/chat")
@Operation(summary = "对话聊天", description = "简单的ai聊天") // @Operation(summary = "对话聊天", description = "简单的ai聊天")
public CommonResult chat(@RequestBody @Validated AiChatReqVO reqVO) { // public CommonResult chat(@RequestBody @Validated AiChatReqVO reqVO) {
ChatClient chatClient = getChatClient(reqVO.getAiModel()); // ChatClient chatClient = getChatClient(reqVO.getAiModel());
String res; // String res;
try { // try {
res = chatClient.call(reqVO.getPrompt()); // res = chatClient.call(reqVO.getPrompt());
} catch (Exception e) { // } catch (Exception e) {
res = e.getMessage(); // res = e.getMessage();
} // }
return CommonResult.success(res); // return CommonResult.success(res);
} // }
//
@PostMapping("/chatStream") // @PostMapping("/chatStream")
@Operation(summary = "对话聊天chatStream", description = "简单的ai聊天") // @Operation(summary = "对话聊天chatStream", description = "简单的ai聊天")
public CommonResult chatStream(HttpServletResponse response, @RequestBody @Validated AiChatReqVO reqVO) throws InterruptedException { // public CommonResult chatStream(HttpServletResponse response, @RequestBody @Validated AiChatReqVO reqVO) throws InterruptedException {
OpenAiChatClient chatClient = applicationContext.getBean(OpenAiChatClient.class); // OpenAiChatClient chatClient = applicationContext.getBean(OpenAiChatClient.class);
Flux<ChatResponse> chatResponse = chatClient.stream(new Prompt(reqVO.getPrompt())); // Flux<ChatResponse> chatResponse = chatClient.stream(new Prompt(reqVO.getPrompt()));
chatResponse.subscribe(new Consumer<ChatResponse>() { // chatResponse.subscribe(new Consumer<ChatResponse>() {
@Override // @Override
public void accept(ChatResponse chatResponse) { // public void accept(ChatResponse chatResponse) {
System.err.println(chatResponse.getResults().get(0).getOutput().getContent()); // System.err.println(chatResponse.getResults().get(0).getOutput().getContent());
} // }
}); // });
return CommonResult.success(null); // return CommonResult.success(null);
} // }
//
/** // /**
* 根据 ai模型 获取对于的 模型实现类 // * 根据 ai模型 获取对于的 模型实现类
* // *
* @param aiModelEnum // * @param aiModelEnum
* @return // * @return
*/ // */
private ChatClient getChatClient(OpenAiModelEnum aiModelEnum) { // private ChatClient getChatClient(OpenAiModelEnum aiModelEnum) {
if (OpenAiModelEnum.OPEN_AI_GPT_3_5 == aiModelEnum) { // if (OpenAiModelEnum.OPEN_AI_GPT_3_5 == aiModelEnum) {
return applicationContext.getBean(OpenAiChatClient.class); // return applicationContext.getBean(OpenAiChatClient.class);
} // }
// AI模型暂不支持 // // AI模型暂不支持
throw ServiceExceptionUtil.exception(ErrorCodeConstants.AI_MODULE_NOT_SUPPORTED); // throw ServiceExceptionUtil.exception(ErrorCodeConstants.AI_MODULE_NOT_SUPPORTED);
} // }
} //}