From d68e2f99592dc982a722d031219f1d0b4f87ed00 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 02 Sep 2025 10:11:05 +0800
Subject: [PATCH] feat(ai): 新增 AI 流式回答功能 V3 版本

---
 src/main/java/cc/mrbird/febs/ai/controller/memberTalk/ApiMemberTalkController.java |   33 ++++++++++++++++++++++++++-------
 1 files changed, 26 insertions(+), 7 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/controller/memberTalk/ApiMemberTalkController.java b/src/main/java/cc/mrbird/febs/ai/controller/memberTalk/ApiMemberTalkController.java
index a2d4df5..48488d4 100644
--- a/src/main/java/cc/mrbird/febs/ai/controller/memberTalk/ApiMemberTalkController.java
+++ b/src/main/java/cc/mrbird/febs/ai/controller/memberTalk/ApiMemberTalkController.java
@@ -1,9 +1,9 @@
 package cc.mrbird.febs.ai.controller.memberTalk;
 
-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.memberTalk.*;
 import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkItemVo;
+import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkReloadVo;
+import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkStreamVo;
 import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkVo;
 import cc.mrbird.febs.ai.service.AiMemberTalkService;
 import cc.mrbird.febs.common.entity.FebsResponse;
@@ -11,11 +11,9 @@
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 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 org.springframework.web.bind.annotation.*;
 import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
+import reactor.core.publisher.Flux;
 
 import javax.servlet.http.HttpServletRequest;
 import java.io.IOException;
@@ -43,6 +41,17 @@
     public FebsResponse start(@RequestBody @Validated ApiMemberTalkDto dto) {
 
         return aiMemberTalkService.start(dto);
+    }
+
+
+    @ApiOperation(value = "再练一次(重新回答)", notes = "再练一次(重新回答)")
+    @ApiResponses({
+            @ApiResponse(code = 200, message = "success", response = ApiMemberTalkReloadVo.class)
+    })
+    @PostMapping(value = "/reload")
+    public FebsResponse reload(@RequestBody @Validated ApiMemberTalkReloadDto dto) {
+
+        return aiMemberTalkService.reload(dto);
     }
 
 
@@ -93,4 +102,14 @@
 
         return emitter;
     }
+
+    // 修改接口定义
+    @ApiOperation("开始AI对话(流式)")
+    @ApiResponses({
+            @ApiResponse(code = 200, message = "流式响应", response = ApiMemberTalkStreamVo.class),
+    })
+    @GetMapping("/answer-stream")
+    public Flux<FebsResponse> answerStream(@RequestParam String question) {
+        return aiMemberTalkService.answerStream(question);
+    }
 }

--
Gitblit v1.9.1