From c0bec6f44f75cdd1fe5acc6bc84ec8e1650c45a7 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Wed, 17 Sep 2025 16:15:43 +0800
Subject: [PATCH] feat(ai): 为 AiMemberTalk 实体添加新字段并更新相关页面

---
 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