Browse Source

【新增】AI:聊天对话的更新模型

YunaiV 1 year ago
parent
commit
20657ccaf3

+ 1 - 4
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/chat/AiChatMessageController.java

@@ -18,24 +18,21 @@ import java.util.List;
 
 import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
 
-// TODO @芋艿:权限标识;
 @Tag(name = "管理后台 - 聊天消息")
 @RestController
 @RequestMapping("/ai/chat/message")
 @Slf4j
 public class AiChatMessageController {
+
     @Resource
     private AiChatService chatService;
 
     @Operation(summary = "发送消息(段式)", description = "一次性返回,响应较慢")
     @PostMapping("/send")
     public CommonResult<AiChatMessageRespVO> sendMessage(@Validated @RequestBody AiChatMessageSendReqVO sendReqVO) {
-        // TODO done @fan:使用 static import;这样就 success 就行了;
         return success(chatService.chat(sendReqVO));
     }
 
-    // TODO @芋艿:调用这个方法异常,Unable to handle the Spring Security Exception because the response is already committed.;可以再试试
-    // TODO @fan:要不要使用 Flux 来返回;可以使用 Flux<AiChatMessageRespVO>
     @Operation(summary = "发送消息(流式)", description = "流式返回,响应较快")
     @PostMapping(value = "/send-stream", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
     @PermitAll // 解决 SSE 最终响应的时候,会被 Access Denied 拦截的问题

+ 4 - 0
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/chat/vo/conversation/AiChatConversationRespVO.java

@@ -32,11 +32,15 @@ public class AiChatConversationRespVO implements VO {
     private Long roleId;
 
     @Schema(description = "模型编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1")
+    @Trans(type = TransType.SIMPLE, target = AiChatModelDO.class, fields = "name", ref = "modelName")
     private Long modelId;
 
     @Schema(description = "模型标志", requiredMode = Schema.RequiredMode.REQUIRED, example = "ERNIE-Bot-turbo-0922")
     private String model;
 
+    @Schema(description = "模型名字", requiredMode = Schema.RequiredMode.REQUIRED, example = "张三")
+    private String modelName;
+
     @Schema(description = "角色设定", example = "一个快乐的程序员")
     private String systemMessage;