From c3a79d24dc10551b7a7830f1d11f66e3adb1430e Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Sat, 11 Oct 2025 09:54:11 +0800
Subject: [PATCH] feat(ai): 实现报告列表流式传输功能
---
src/main/java/cc/mrbird/febs/ai/controller/talk/ApiAiTalkController.java | 58 +++++++++++++++++++++++++++++++++++++++++-----------------
1 files changed, 41 insertions(+), 17 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/controller/talk/ApiAiTalkController.java b/src/main/java/cc/mrbird/febs/ai/controller/talk/ApiAiTalkController.java
index adf10b1..6ece427 100644
--- a/src/main/java/cc/mrbird/febs/ai/controller/talk/ApiAiTalkController.java
+++ b/src/main/java/cc/mrbird/febs/ai/controller/talk/ApiAiTalkController.java
@@ -1,30 +1,20 @@
package cc.mrbird.febs.ai.controller.talk;
-import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkAnswerDto;
-import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkDto;
-import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkItemPageDto;
-import cc.mrbird.febs.ai.req.talk.AiTalkAnswerStream;
-import cc.mrbird.febs.ai.req.talk.ApiTalkDto;
-import cc.mrbird.febs.ai.req.talk.ApiTalkItemPageDto;
-import cc.mrbird.febs.ai.req.talk.ApiTalkPageDto;
-import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkItemVo;
-import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkStreamVo;
-import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkVo;
+import cc.mrbird.febs.ai.req.talk.*;
+import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkStreamVoOld;
import cc.mrbird.febs.ai.res.talk.ApiTalkItemPageVo;
import cc.mrbird.febs.ai.res.talk.ApiTalkPageVo;
+import cc.mrbird.febs.ai.res.talk.ApiTalkQuestionVo;
import cc.mrbird.febs.ai.res.talk.ApiTalkVo;
-import cc.mrbird.febs.ai.service.AiMemberTalkService;
import cc.mrbird.febs.ai.service.AiTalkService;
import cc.mrbird.febs.common.entity.FebsResponse;
+import cn.hutool.core.util.StrUtil;
import io.swagger.annotations.*;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
-import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
import reactor.core.publisher.Flux;
-
-import java.io.IOException;
/**
* @author Administrator
@@ -38,6 +28,26 @@
public class ApiAiTalkController {
private final AiTalkService aiTalkService;
+
+ @ApiOperation(value = "题目轮播", notes = "题目轮播")
+ @ApiResponses({
+ @ApiResponse(code = 200, message = "success", response = ApiTalkQuestionVo.class)
+ })
+ @GetMapping(value = "/questionList")
+ public FebsResponse questionList() {
+
+ return aiTalkService.questionList();
+ }
+
+ @ApiOperation(value = "题目轮播", notes = "题目轮播")
+ @ApiResponses({
+ @ApiResponse(code = 200, message = "success", response = ApiTalkQuestionVo.class)
+ })
+ @PostMapping(value = "/questionListV2")
+ public FebsResponse questionListV2(@RequestBody @Validated ApiQuestionListDto dto) {
+
+ return aiTalkService.questionListV2(dto);
+ }
@ApiOperation(value = "用户AI问答", notes = "用户AI问答")
@ApiResponses({
@@ -71,7 +81,7 @@
@ApiOperation("提问AI(流式)")
@ApiResponses({
- @ApiResponse(code = 200, message = "流式响应", response = ApiMemberTalkStreamVo.class),
+ @ApiResponse(code = 200, message = "流式响应", response = ApiMemberTalkStreamVoOld.class),
})
@GetMapping("/answer-stream")
public Flux<FebsResponse> answerStream(@RequestParam String question) {
@@ -81,11 +91,25 @@
@ApiOperation("提问AI(流式)V2")
@ApiResponses({
- @ApiResponse(code = 200, message = "流式响应", response = ApiMemberTalkStreamVo.class),
+ @ApiResponse(code = 200, message = "流式响应", response = ApiMemberTalkStreamVoOld.class),
})
@PostMapping("/answer-streamV2")
public Flux<FebsResponse> answerStreamV2(@RequestBody @Validated AiTalkAnswerStream dto) {
-
+ if (StrUtil.isEmpty(dto.getQuestion())){
+ return Flux.just(new FebsResponse().fail().message("请输入问题"));
+ }
return aiTalkService.answerStreamV2(dto);
}
+
+ @ApiOperation("提问AI(流式带思考过程)V3")
+ @ApiResponses({
+ @ApiResponse(code = 200, message = "流式响应", response = ApiMemberTalkStreamVoOld.class),
+ })
+ @PostMapping("/answer-streamV3")
+ public Flux<FebsResponse> answerStreamV3(@RequestBody @Validated AiTalkAnswerStream dto) {
+ if (StrUtil.isEmpty(dto.getQuestion())){
+ return Flux.just(new FebsResponse().fail().message("请输入问题"));
+ }
+ return aiTalkService.answerStreamV3(dto);
+ }
}
--
Gitblit v1.9.1