blame | history | patch | commit | commitdiff | ignore whitespace
Administrator
2025-08-19 0b7f5d136824a74f491ee3566088402f5223589b 
 src/main/java/cc/mrbird/febs/ai/controller/memberTalk/ApiMemberTalkController.java


@@ -2,13 +2,12 @@




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.res.memberTalk.ApiMemberTalkItemVo;


import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkVo;


import cc.mrbird.febs.ai.service.AiMemberTalkService;


import cc.mrbird.febs.common.entity.FebsResponse;


import io.swagger.annotations.Api;


import io.swagger.annotations.ApiOperation;


import io.swagger.annotations.ApiResponse;


import io.swagger.annotations.ApiResponses;


import io.swagger.annotations.*;


import lombok.RequiredArgsConstructor;


import lombok.extern.slf4j.Slf4j;


import org.springframework.validation.annotation.Validated;


@@ -16,6 +15,11 @@


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.servlet.mvc.method.annotation.SseEmitter;




import javax.servlet.http.HttpServletRequest;


import java.io.IOException;


import java.util.List;




/**


 * @author Administrator


@@ -51,4 +55,42 @@




        return aiMemberTalkService.answer(dto);


    }




    @ApiOperation(value = "对话记录分页查询", notes = "对话记录分页查询")


    @ApiResponses({


            @ApiResponse(code = 200, message = "success", response = ApiMemberTalkItemVo.class)


    })


    @PostMapping(value = "/historyPage")


    public FebsResponse historyPage(@RequestBody @Validated ApiMemberTalkItemPageDto dto) {




        return aiMemberTalkService.historyPage(dto);


    }




    @PostMapping("/start-stream")


    @ApiOperation("开始AI对话(流式)")


    @ApiResponses({


            @ApiResponse(code = 200, message = "流式响应", response = ApiMemberTalkVo.class),


            @ApiResponse(code = 500, message = "系统错误")


    })


    public SseEmitter startStream(


            @ApiParam(value = "对话请求参数", required = true)


            @RequestBody ApiMemberTalkDto dto) {




        SseEmitter emitter = new SseEmitter(0L); // 0表示永不超时




        aiMemberTalkService.startStream(dto, response -> {


            try {


                emitter.send(SseEmitter.event().data(response));


                // 如果包含report，说明是最终结果，关闭连接


                if (response.getCode() != null &&


                        "200".equals(response.getCode())) {


                    emitter.complete();


                }


            } catch (IOException e) {


                emitter.completeWithError(e);


            }


        });




        return emitter;


    }


}