From e2a0ea3a333e46b22152d1e90356db4d2cda5f66 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 29 Sep 2025 14:03:18 +0800
Subject: [PATCH] feat(ai): 新增阿里云百炼知识库工具类 - 添加 KnowledgeBaseUtil 工具类,封装阿里云百炼知识库操作 - 实现文件上传、知识库创建与更新等核心功能 - 支持文件MD5计算、分类管理、索引任务提交与状态查询 - 集成阿里云百炼SDK依赖,版本为2.5.0 - 提供完整的知识库初始化与文件更新流程示例 - 添加文件解析状态轮询与错误处理机制 - 支持知识库中文档的追加导入与旧文档删除功能

---
 src/main/java/cc/mrbird/febs/ai/service/impl/AiProductRoleServiceImpl.java |   14 +++++++++++---
 1 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductRoleServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductRoleServiceImpl.java
index 0f361cb..4d2a6bc 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductRoleServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductRoleServiceImpl.java
@@ -8,6 +8,7 @@
 import cc.mrbird.febs.common.entity.FebsResponse;
 import cc.mrbird.febs.common.entity.QueryRequest;
 import cn.hutool.core.util.ObjectUtil;
+import cn.hutool.core.util.StrUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
@@ -44,6 +45,9 @@
 
         Page<AiProductRole> page = new Page<>(request.getPageNum(), request.getPageSize());
         LambdaQueryWrapper<AiProductRole> query = Wrappers.lambdaQuery(AiProductRole.class);
+        if (StrUtil.isNotEmpty(dto.getCompanyId())){
+            query.eq(AiProductRole::getCompanyId, dto.getCompanyId());
+        }
         Page<AiProductRole> pages = aiProductRoleMapper.selectPage(page, query);
         return pages;
     }
@@ -73,7 +77,6 @@
         if (ObjectUtil.isNotNull( entity)){
             this.update(null,
                     Wrappers.lambdaUpdate(AiProductRole.class)
-                            .set(AiProductRole::getCompanyId, dto.getCompanyId())
                             .set(AiProductRole::getName, dto.getName())
                             .set(AiProductRole::getDescription, dto.getDescription())
                             .set(AiProductRole::getPromptHead, dto.getPromptHead())
@@ -101,8 +104,13 @@
     }
 
     @Override
-    public List<AiProductRole> productRoleTree() {
-        return aiProductRoleMapper.selectList(null);
+    public List<AiProductRole> productRoleTree(String companyId) {
+
+        LambdaQueryWrapper<AiProductRole> query = Wrappers.lambdaQuery(AiProductRole.class);
+        if (StrUtil.isNotEmpty(companyId)){
+            query.eq(AiProductRole::getCompanyId, companyId);
+        }
+        return aiProductRoleMapper.selectList(query);
     }
 
 }

--
Gitblit v1.9.1