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