package cc.mrbird.febs.ai.controller.memberTalk;
|
|
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.ApiMemberTalkStreamVoOld;
|
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.*;
|
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
|
*/
|
@Slf4j
|
@Validated
|
@RestController
|
@RequiredArgsConstructor
|
@RequestMapping(value = "/api/ai/memberTalk")
|
@Api(value = "ApiMemberTalkController", tags = "AI-用户陪练")
|
public class ApiMemberTalkController {
|
|
private final AiMemberTalkService aiMemberTalkService;
|
|
|
@ApiOperation(value = "开始陪练", notes = "开始陪练")
|
@ApiResponses({
|
@ApiResponse(code = 200, message = "success", response = ApiMemberTalkVo.class)
|
})
|
@PostMapping(value = "/start")
|
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);
|
}
|
|
|
@ApiOperation(value = "回答", notes = "回答")
|
@ApiResponses({
|
@ApiResponse(code = 200, message = "success", response = ApiMemberTalkVo.class)
|
})
|
@PostMapping(value = "/answer")
|
public FebsResponse answer(@RequestBody @Validated ApiMemberTalkAnswerDto dto) {
|
|
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;
|
}
|
|
// 修改接口定义
|
@ApiOperation("开始AI对话(流式)")
|
@ApiResponses({
|
@ApiResponse(code = 200, message = "流式响应", response = ApiMemberTalkStreamVoOld.class),
|
})
|
@GetMapping("/answer-stream")
|
public Flux<FebsResponse> answerStream(@RequestParam String question) {
|
return aiMemberTalkService.answerStream(question);
|
}
|
}
|
