package cc.mrbird.febs.ai.controller.agent;
|
|
import cc.mrbird.febs.ai.req.agent.*;
|
import cc.mrbird.febs.ai.req.productCategory.ApiProductCategoryAllDto;
|
import cc.mrbird.febs.ai.res.agent.AiAgentInitVo;
|
import cc.mrbird.febs.ai.res.agent.ApiAgentCategoryVo;
|
import cc.mrbird.febs.ai.res.agent.ApiAgentVo;
|
import cc.mrbird.febs.ai.res.productCategory.ApiProductCategoryVo;
|
import cc.mrbird.febs.ai.service.AiAgentService;
|
import cc.mrbird.febs.common.entity.FebsResponse;
|
import cn.hutool.core.util.StrUtil;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import io.swagger.annotations.ApiResponse;
|
import io.swagger.annotations.ApiResponses;
|
import lombok.RequiredArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.validation.annotation.Validated;
|
import org.springframework.web.bind.annotation.*;
|
import reactor.core.publisher.Flux;
|
|
import javax.servlet.http.HttpServletRequest;
|
|
/**
|
* @author Administrator
|
*/
|
@Slf4j
|
@Validated
|
@RestController
|
@RequiredArgsConstructor
|
@RequestMapping(value = "/api/ai/agent")
|
@Api(value = "ApiAgentController", tags = "AI-智能体")
|
public class ApiAgentController {
|
|
private final AiAgentService service;
|
|
@ApiOperation(value = "全部分类(带公司ID)", notes = "全部分类(带公司ID)")
|
@ApiResponses({
|
@ApiResponse(code = 200, message = "success", response = ApiAgentCategoryVo.class)
|
})
|
@PostMapping(value = "/allCategoryList")
|
public FebsResponse allCategoryList(@RequestBody ApiAgentCategoryAllDto dto) {
|
|
return service.allCategoryList(dto);
|
}
|
|
|
@ApiOperation(value = "列表", notes = "列表")
|
@ApiResponses({
|
@ApiResponse(code = 200, message = "success", response = ApiAgentVo.class)
|
})
|
@PostMapping(value = "/list")
|
public FebsResponse list(@RequestBody @Validated ApiAgentPageDto dto) {
|
|
return service.agentList(dto);
|
}
|
|
@ApiOperation(value = "详情", notes = "详情")
|
@ApiResponses({
|
@ApiResponse(code = 200, message = "success", response = AiAgentInitVo.class)
|
})
|
@PostMapping(value = "/initAgent", produces = "application/json")
|
public FebsResponse initAgent(@RequestBody @Validated AiAgentInitDto dto) {
|
|
return service.initAgent(dto);
|
}
|
|
@ApiOperation(value = "初始化对话", notes = "初始化对话")
|
@ApiResponses({
|
@ApiResponse(code = 200, message = "success", response = AgentSendInitVo.class)
|
})
|
@PostMapping(value = "/initSend", produces = "application/json")
|
public FebsResponse initSend(@RequestBody @Validated AgentInitDto dto) {
|
|
return service.initSend(dto);
|
}
|
|
@ApiOperation(value = "保存对话消息", notes = "保存对话消息")
|
@PostMapping(value = "/saveContext")
|
public FebsResponse saveContext(@RequestBody @Validated AgentSaveContextDto dto) {
|
|
return service.saveContext(dto);
|
}
|
|
@ApiOperation("AI回答(流式)")
|
@ApiResponses({
|
@ApiResponse(code = 200, message = "流式响应", response = AitalkItemStreamVo.class),
|
})
|
@PostMapping("/aiAnswer")
|
public Flux<FebsResponse> aiAnswer(@RequestBody @Validated AitalkItemStreamDto dto) {
|
if (StrUtil.isEmpty(dto.getTalkId()) || StrUtil.isEmpty(dto.getReqContext())|| StrUtil.isEmpty(dto.getReqContext())){
|
return Flux.just(new FebsResponse().fail().message("参数异常"));
|
}
|
return service.aiAnswer(dto);
|
}
|
|
|
|
}
|
