From 5f366e546806a5690ff7a5770857048663e4521a Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 15 Sep 2025 16:17:11 +0800
Subject: [PATCH] feat(ai): 增加公司 ID 筛选功能
---
src/main/java/cc/mrbird/febs/ai/controller/productQuestion/AiProductQuestionController.java | 10 +++++++++-
1 files changed, 9 insertions(+), 1 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 591aa13..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
@@ -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,6 +51,8 @@
@ControllerEndpoint(operation = "新增", exceptionMessage = "操作失败")
public FebsResponse add(@RequestBody @Valid AiProductQuestion dto) {
+ String companyId = getCurrentUserCompanyId();
+ dto.setCompanyId(companyId);
return aiProductQuestionService.add(dto);
}
@@ -56,6 +60,8 @@
@ControllerEndpoint(operation = "AI生成题目", exceptionMessage = "操作失败")
public FebsResponse aiAdd(@RequestBody @Valid AiProductQuestionAiDto dto) {
+ String companyId = getCurrentUserCompanyId();
+ dto.setCompanyId(companyId);
return aiProductQuestionService.aiAdd(dto);
}
@@ -78,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