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 | 19 +++++++++++++++++--
1 files changed, 17 insertions(+), 2 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 ca3cb00..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,7 +1,7 @@
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;
import cc.mrbird.febs.common.annotation.ControllerEndpoint;
import cc.mrbird.febs.common.controller.BaseController;
@@ -30,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);
@@ -49,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")
@@ -71,6 +84,8 @@
@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