feat(ai): 新增公司知识库初始化功能

Administrator

9 hours ago 451101e29d1d48a7ed06b6b31fe3f6bdda5dd88a

feat(ai): 新增公司知识库初始化功能

- 在新增公司时自动创建对应的知识库分类
- 集成阿里云百炼API实现知识库索引创建
- 新增知识库文件实体类及相应Mapper和服务层接口
- 修改工具类访问修饰符以支持外部调用
- 添加事务注解确保数据一致性
- 处理父分类ID为空的情况

 src/main/java/cc/mrbird/febs/ai/entity/AiKnowledgeFile.java

New file
@@ -0,0 +1,29 @@
package cc.mrbird.febs.ai.entity;

import cc.mrbird.febs.common.entity.AiBaseEntity;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;

@Data
@TableName("ai_knowledge_file")
public class AiKnowledgeFile extends AiBaseEntity {
    /**
     * 文件ID
     */
    private String fileId;

    /**
     * 文件存储路径
     */
    private String savePath;

    /**
     * 文件名称
     */
    private String name;

    /**
     * 公司ID
     */
    private String companyId;
}

 src/main/java/cc/mrbird/febs/ai/mapper/AiKnowledgeFileMapper.java

New file
@@ -0,0 +1,7 @@
package cc.mrbird.febs.ai.mapper;

import cc.mrbird.febs.ai.entity.AiKnowledgeFile;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;

public interface AiKnowledgeFileMapper extends BaseMapper<AiKnowledgeFile> {
}

 src/main/java/cc/mrbird/febs/ai/service/AiKnowledgeFileService.java

New file
@@ -0,0 +1,7 @@
package cc.mrbird.febs.ai.service;

import cc.mrbird.febs.ai.entity.AiKnowledgeFile;
import com.baomidou.mybatisplus.extension.service.IService;

public interface AiKnowledgeFileService extends IService<AiKnowledgeFile> {
}

 src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java

@@ -6,12 +6,16 @@
import cc.mrbird.febs.ai.mapper.AiCompanyMapper;
import cc.mrbird.febs.ai.mapper.AiCompanyWorkflowMapper;
import cc.mrbird.febs.ai.service.AiCompanyService;
import cc.mrbird.febs.ai.util.KnowledgeBaseUtil;
import cc.mrbird.febs.ai.util.UUID;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.entity.QueryRequest;
import cc.mrbird.febs.common.exception.FebsException;
import cc.mrbird.febs.common.utils.ShareCodeUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil;
import com.aliyun.bailian20231229.Client;
import com.aliyun.bailian20231229.models.CreateIndexResponse;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
@@ -20,6 +24,7 @@
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.*;
import java.util.stream.Collectors;
@@ -45,6 +50,7 @@
    }

    @Override
    @Transactional
    public FebsResponse add(AiCompany dto) {
        Date createTime = new Date();
        AiCompany entity = new AiCompany();
@@ -63,6 +69,34 @@
        this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.REPORT.getCode(),null,createTime);
        this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.ADMIN_QUESTION.getCode(),null,createTime);

        String categoryId = "";
        try {
            categoryId = KnowledgeBaseUtil.AddCategory(entity.getName(), null);
        } catch (Exception e) {
            throw new FebsException("初始化分类失败");
        }

        String knowledgeId = "";
        try {
            Client client = KnowledgeBaseUtil.createClient();
            String workspaceId = KnowledgeBaseUtil.WORKSPACE_ID;
            String name = entity.getName();
            String sourceType = "DATA_CENTER_FILE";
            String structureType = "unstructured";
            String sinkType = "DEFAULT";
            String fileId = "file_b7d0a5b2df1745b9bfb7402f5fe1054f_12629554";
            CreateIndexResponse indexResponse = KnowledgeBaseUtil.createIndex(client, workspaceId, fileId, name, structureType, sourceType, sinkType);
            knowledgeId = indexResponse.getBody().getData().getId();
        } catch (Exception e) {
            throw new FebsException("初始化知识库失败");
        }

        this.update(null,
                Wrappers.lambdaUpdate(AiCompany.class)
                        .set(AiCompany::getKnowledgeId, knowledgeId)
                        .set(AiCompany::getCategoryId, categoryId)
                        .eq(AiCompany::getId, entity.getId()));

        return new FebsResponse().success().message("操作成功");
    }


 src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java

New file
@@ -0,0 +1,17 @@
package cc.mrbird.febs.ai.service.impl;

import cc.mrbird.febs.ai.entity.AiKnowledgeFile;
import cc.mrbird.febs.ai.mapper.AiKnowledgeFileMapper;
import cc.mrbird.febs.ai.service.AiKnowledgeFileService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

@Slf4j
@Service
@RequiredArgsConstructor
public class AiKnowledgeFileServiceImpl extends ServiceImpl<AiKnowledgeFileMapper, AiKnowledgeFile> implements AiKnowledgeFileService {

    private final AiKnowledgeFileMapper aiKnowledgeFileMapper;
}

 src/main/java/cc/mrbird/febs/ai/util/KnowledgeBaseUtil.java

@@ -17,10 +17,10 @@
import java.util.concurrent.TimeUnit;

public class KnowledgeBaseUtil {
    private static String       ACCESS_KEY_ID     = "LTAI5tCyQRwhZ2eimxCFKbdq";
    private static String       ACCESS_KEY_SECRET = "fs1mEwLXg2j9XuKJsFoW8ThQbJFqHl";
    private static String       WORKSPACE_ID = "llm-4bcr09yfxlgz0b0t";
    private static String       ENDPOINT  = "bailian.cn-beijing.aliyuncs.com";
    public static final String       ACCESS_KEY_ID     = "LTAI5tCyQRwhZ2eimxCFKbdq";
    public static final String       ACCESS_KEY_SECRET = "fs1mEwLXg2j9XuKJsFoW8ThQbJFqHl";
    public static final String       WORKSPACE_ID = "llm-4bcr09yfxlgz0b0t";
    public static final String       ENDPOINT  = "bailian.cn-beijing.aliyuncs.com";

    /**
     * <b>description</b> :
@@ -44,6 +44,9 @@
     * @throws Exception 当API调用失败或其他异常情况时抛出
     */
    public static String AddCategory(String categoryName, String parentCategoryId) throws Exception {
        if (StrUtil.isBlank(parentCategoryId)){
            parentCategoryId = "";
        }
        // 创建阿里云百炼客户端
        com.aliyun.bailian20231229.Client client = KnowledgeBaseUtil.createClient();
        // 构造添加分类请求参数