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 |    8 +++++---
 1 files changed, 5 insertions(+), 3 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 e03c29a..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,12 +50,13 @@
     @PostMapping("add")
     @ControllerEndpoint(operation = "新增", exceptionMessage = "操作失败")
     public FebsResponse add(@RequestBody @Valid AiMemberRole dto) {
-
+        String companyId = getCurrentUserCompanyId();
+        dto.setCompanyId(companyId);
         return aiMemberRoleService.memberRoleAdd(dto);
     }
 
     @PostMapping("update")
-    @ControllerEndpoint(operation = "分类-更新", exceptionMessage = "操作失败")
+    @ControllerEndpoint(operation = "更新", exceptionMessage = "操作失败")
     public FebsResponse update(@RequestBody @Valid AiMemberRole dto) {
 
         return aiMemberRoleService.updateMemberRole(dto);

--
Gitblit v1.9.1