package cc.mrbird.febs.ai.controller.memberTalk;

import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkItemPageDto;
import cc.mrbird.febs.ai.req.memberTalkStream.ApiMemberTalkReloadStreamDto;
import cc.mrbird.febs.ai.req.memberTalkStream.ApiMemberTalkStreamDto;
import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkItemVo;
import cc.mrbird.febs.ai.res.memberTalkStream.ApiMemberTalkReloadStreamVo;
import cc.mrbird.febs.ai.res.memberTalkStream.ApiMemberTalkStreamVo;
import cc.mrbird.febs.ai.service.ApiMemberTalkStreamService;
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.*;

/**
 * @author Administrator
 */
@Slf4j
@Validated
@RestController
@RequiredArgsConstructor
@RequestMapping(value = "/api/ai/memberTalkStream")
@Api(value = "ApiMemberTalkStreamController", tags = "AI-用户陪练(流式)")
public class ApiMemberTalkStreamController {

    private final ApiMemberTalkStreamService apiMemberTalkStreamService;

    @ApiOperation("生成题目")
    @ApiResponses({
            @ApiResponse(code = 200, message = "流式响应", response = ApiMemberTalkStreamVo.class),
    })
    @PostMapping("/start")
    public FebsResponse start(@RequestBody @Validated ApiMemberTalkStreamDto dto) {
        return apiMemberTalkStreamService.start(dto);
    }

    @ApiOperation(value = "再练一次（重新回答）", notes = "再练一次（重新回答）")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = ApiMemberTalkReloadStreamVo.class)
    })
    @PostMapping(value = "/reload")
    public FebsResponse reload(@RequestBody @Validated ApiMemberTalkReloadStreamDto dto) {

        return apiMemberTalkStreamService.reload(dto);
    }

    @ApiOperation(value = "对话记录分页查询", notes = "对话记录分页查询")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = ApiMemberTalkItemVo.class)
    })
    @PostMapping(value = "/historyPage")
    public FebsResponse historyPage(@RequestBody @Validated ApiMemberTalkItemPageDto dto) {

        return apiMemberTalkStreamService.historyPage(dto);
    }

}