From 8298b88330abc838dd6484c96317de68b95aeb6a Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 16 Sep 2025 16:49:44 +0800
Subject: [PATCH] feat(ai): 新增 AI 用户选择角色关联产品功能

---
 src/main/java/cc/mrbird/febs/ai/controller/memberRole/AiMemberRoleController.java |    6 ++++--
 1 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/controller/memberRole/AiMemberRoleController.java b/src/main/java/cc/mrbird/febs/ai/controller/memberRole/AiMemberRoleController.java
index 4300e07..5c9d160 100644
--- a/src/main/java/cc/mrbird/febs/ai/controller/memberRole/AiMemberRoleController.java
+++ b/src/main/java/cc/mrbird/febs/ai/controller/memberRole/AiMemberRoleController.java
@@ -31,7 +31,8 @@
 
     @GetMapping("list")
     public FebsResponse list(AiMemberRole dto, QueryRequest request) {
-
+        String companyId = getCurrentUserCompanyId();
+        dto.setCompanyId(companyId);
         Map<String, Object> data = getDataTable(aiMemberRoleService.listInPage(dto, request));
         return new FebsResponse().success().data(data);
     }
@@ -49,7 +50,8 @@
     @PostMapping("add")
     @ControllerEndpoint(operation = "新增", exceptionMessage = "操作失败")
     public FebsResponse add(@RequestBody @Valid AiMemberRole dto) {
-
+        String companyId = getCurrentUserCompanyId();
+        dto.setCompanyId(companyId);
         return aiMemberRoleService.memberRoleAdd(dto);
     }
 

--
Gitblit v1.9.1