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/productPoint/AiProductPointController.java |    9 +++++++--
 1 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/controller/productPoint/AiProductPointController.java b/src/main/java/cc/mrbird/febs/ai/controller/productPoint/AiProductPointController.java
index 0c53cf9..12a8535 100644
--- a/src/main/java/cc/mrbird/febs/ai/controller/productPoint/AiProductPointController.java
+++ b/src/main/java/cc/mrbird/febs/ai/controller/productPoint/AiProductPointController.java
@@ -29,7 +29,8 @@
 
     @GetMapping("list")
     public FebsResponse list(AiProductPoint dto, QueryRequest request) {
-
+        String companyId = getCurrentUserCompanyId();
+        dto.setCompanyId(companyId);
         Map<String, Object> data = getDataTable(service.listInPage(dto, request));
         return new FebsResponse().success().data(data);
     }
@@ -38,6 +39,8 @@
     @ControllerEndpoint(operation = "新增", exceptionMessage = "操作失败")
     public FebsResponse add(@RequestBody @Valid AiProductPoint dto) {
 
+        String companyId = getCurrentUserCompanyId();
+        dto.setCompanyId(companyId);
         return service.add(dto);
     }
 
@@ -60,6 +63,8 @@
     @GetMapping(value = "/pointTree")
     public FebsResponse pointTree() {
 
-        return new FebsResponse().success().data(service.pointTree());
+
+        String companyId = getCurrentUserCompanyId();
+        return new FebsResponse().success().data(service.pointTree(companyId));
     }
 }

--
Gitblit v1.9.1