From 49400da0804fd0ab56056fbd0a2c9ee36cd091b6 Mon Sep 17 00:00:00 2001 From: Administrator <15274802129@163.com> Date: Mon, 15 Sep 2025 16:22:55 +0800 Subject: [PATCH] feat(ai): 增加公司 ID 筛选功能 --- src/main/java/cc/mrbird/febs/ai/controller/productQuestion/AiProductQuestionController.java | 25 ++++++++++++++++--------- 1 files changed, 16 insertions(+), 9 deletions(-) diff --git a/src/main/java/cc/mrbird/febs/ai/controller/productQuestion/AiProductQuestionController.java b/src/main/java/cc/mrbird/febs/ai/controller/productQuestion/AiProductQuestionController.java index faa94f0..497dc54 100644 --- a/src/main/java/cc/mrbird/febs/ai/controller/productQuestion/AiProductQuestionController.java +++ b/src/main/java/cc/mrbird/febs/ai/controller/productQuestion/AiProductQuestionController.java @@ -1,6 +1,5 @@ package cc.mrbird.febs.ai.controller.productQuestion; -import cc.mrbird.febs.ai.entity.AiProductPoint; import cc.mrbird.febs.ai.entity.AiProductQuestion; import cc.mrbird.febs.ai.req.AiProductQuestionAiDto; import cc.mrbird.febs.ai.service.AiProductQuestionService; @@ -31,6 +30,8 @@ @GetMapping("list") public FebsResponse list(AiProductQuestion dto, QueryRequest request) { + String companyId = getCurrentUserCompanyId(); + dto.setCompanyId(companyId); Map<String, Object> data = getDataTable(aiProductQuestionService.listInPage(dto, request)); return new FebsResponse().success().data(data); @@ -50,7 +51,18 @@ @ControllerEndpoint(operation = "新增", exceptionMessage = "操作失败") public FebsResponse add(@RequestBody @Valid AiProductQuestion dto) { + String companyId = getCurrentUserCompanyId(); + dto.setCompanyId(companyId); return aiProductQuestionService.add(dto); + } + + @PostMapping("aiAdd") + @ControllerEndpoint(operation = "AI生成题目", exceptionMessage = "操作失败") + public FebsResponse aiAdd(@RequestBody @Valid AiProductQuestionAiDto dto) { + + String companyId = getCurrentUserCompanyId(); + dto.setCompanyId(companyId); + return aiProductQuestionService.aiAdd(dto); } @PostMapping("update") @@ -69,16 +81,11 @@ return aiProductQuestionService.delete(id); } - @PostMapping("aiCreate/{id}") - @ControllerEndpoint(operation = "AI生成题目", exceptionMessage = "操作失败") - public FebsResponse aiCreate(@RequestBody @Valid AiProductQuestionAiDto dto) { - - return aiProductQuestionService.aiCreate(dto); - } - @GetMapping(value = "/questionTree") public FebsResponse questionTree() { - return new FebsResponse().success().data(aiProductQuestionService.questionTree()); + + String companyId = getCurrentUserCompanyId(); + return new FebsResponse().success().data(aiProductQuestionService.questionTree(companyId)); } } -- Gitblit v1.9.1