huangge1199/vue-pro
1
0
Fork 0
mirror of https://gitee.com/huangge1199_admin/vue-pro.git synced 2024-11-30 11:11:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

增加 client name 枚举

Browse Source
This commit is contained in:
cherishsince 2024-04-14 16:05:29 +08:00
parent 009914ed73
commit 7024c5ab60
2 changed files with 35 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/AiClientNameEnum.java Normal file
View File

@ -0,0 +1,28 @@
package cn.iocoder.yudao.module.ai.enums;
import lombok.AllArgsConstructor;
import lombok.Getter;
/**
* ai client 名字
*
* 这个需要根据配置文件起的来决定
*
* @author fansili
* @time 2024/4/14 16:02
* @since 1.0
*/
@AllArgsConstructor
@Getter
public enum AiClientNameEnum {
QIAN_WEN("qianWen", "千问模型!"),
YI_YAN("yiYan", "一言模型!"),
XING_HUO("xingHuo", "星火模型!"),
;
private String name;
private String message;
}

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

@ -5,6 +5,8 @@ import cn.iocoder.yudao.framework.ai.chat.ChatResponse;
import cn.iocoder.yudao.framework.ai.chat.prompt.Prompt; import cn.iocoder.yudao.framework.ai.chat.prompt.Prompt;
import cn.iocoder.yudao.framework.ai.config.AiClient; import cn.iocoder.yudao.framework.ai.config.AiClient;
import cn.iocoder.yudao.framework.common.pojo.CommonResult; import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.module.ai.enums.AiClientNameEnum;
import cn.iocoder.yudao.module.ai.service.ChatService;
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 lombok.AllArgsConstructor; import lombok.AllArgsConstructor;
@ -22,6 +24,8 @@ import java.io.IOException;
import java.util.function.Consumer; import java.util.function.Consumer;
/** /**
* ia 模块
*
* @author fansili * @author fansili
* @time 2024/4/13 17:44 * @time 2024/4/13 17:44
* @since 1.0 * @since 1.0
@ -35,11 +39,12 @@ public class ChatController {
@Autowired @Autowired
private AiClient aiClient; private AiClient aiClient;
private final ChatService chatService;
@Operation(summary = "聊天-chat", description = "这个一般等待时间比较久，需要全部完成才会返回!") @Operation(summary = "聊天-chat", description = "这个一般等待时间比较久，需要全部完成才会返回!")
@GetMapping("/chat") @GetMapping("/chat")
public CommonResult<String> chat(@RequestParam("prompt") String prompt) { public CommonResult<String> chat(@RequestParam("prompt") String prompt) {
ChatResponse callRes = aiClient.call(new Prompt(prompt), "qianWen"); ChatResponse callRes = aiClient.call(new Prompt(prompt), AiClientNameEnum.QIAN_WEN.getName());
return CommonResult.success(callRes.getResult().getOutput().getContent()); return CommonResult.success(callRes.getResult().getOutput().getContent());
} }
@ -48,7 +53,7 @@ public class ChatController {
@GetMapping(value = "/chatStream", produces = MediaType.TEXT_EVENT_STREAM_VALUE) @GetMapping(value = "/chatStream", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
public SseEmitter chatStream(@RequestParam("prompt") String prompt) { public SseEmitter chatStream(@RequestParam("prompt") String prompt) {
Utf8SseEmitter sseEmitter = new Utf8SseEmitter(); Utf8SseEmitter sseEmitter = new Utf8SseEmitter();
Flux<ChatResponse> streamResponse = aiClient.stream(new Prompt(prompt), "qianWen"); Flux<ChatResponse> streamResponse = aiClient.stream(new Prompt(prompt), AiClientNameEnum.QIAN_WEN.getName());
streamResponse.subscribe( streamResponse.subscribe(
new Consumer<ChatResponse>() { new Consumer<ChatResponse>() {
@Override @Override