From 451101e29d1d48a7ed06b6b31fe3f6bdda5dd88a Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 29 Sep 2025 15:45:51 +0800
Subject: [PATCH] feat(ai): 新增公司知识库初始化功能
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java | 154 ++++++++++++++++++++++++++++++++++++++++++++++++++-
1 files changed, 151 insertions(+), 3 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java
index ffeb9e1..37f0506 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java
@@ -1,13 +1,21 @@
package cc.mrbird.febs.ai.service.impl;
import cc.mrbird.febs.ai.entity.AiCompany;
-import cc.mrbird.febs.ai.entity.AiMemberRole;
+import cc.mrbird.febs.ai.entity.AiCompanyWorkflow;
+import cc.mrbird.febs.ai.enums.LlmApplicationAppIdEnum;
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;
@@ -16,8 +24,10 @@
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
-import java.util.Date;
+import java.util.*;
+import java.util.stream.Collectors;
@Slf4j
@Service
@@ -25,6 +35,7 @@
public class AiCompanyServiceImpl extends ServiceImpl<AiCompanyMapper, AiCompany> implements AiCompanyService {
private final AiCompanyMapper aiCompanyMapper;
+ private final AiCompanyWorkflowMapper aiCompanyWorkflowMapper;
@Override
public AiCompany getById(String id) {
@@ -32,13 +43,67 @@
}
@Override
+ public List<AiCompany> getListById(List<String> ids) {
+ LambdaQueryWrapper<AiCompany> aiCompanyLambdaQueryWrapper = Wrappers.lambdaQuery(AiCompany.class);
+ aiCompanyLambdaQueryWrapper.in(AiCompany::getId, ids);
+ return aiCompanyMapper.selectList(aiCompanyLambdaQueryWrapper);
+ }
+
+ @Override
+ @Transactional
public FebsResponse add(AiCompany dto) {
+ Date createTime = new Date();
AiCompany entity = new AiCompany();
entity.setId(UUID.getSimpleUUIDString());
entity.setName(dto.getName());
- entity.setCreatedTime(new Date());
+ entity.setCode(ShareCodeUtil.toSerialCode(1L));
+ entity.setCreatedTime(createTime);
this.baseMapper.insert( entity);
+
+
+
+ this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.HIGH_LIGHT.getCode(),null,createTime);
+ this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.SUGGESTION.getCode(),null,createTime);
+ this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.REFERENCE_ANSWER.getCode(),null,createTime);
+ this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.KEY_KNOWLEDGE.getCode(),null,createTime);
+ 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("操作成功");
+ }
+
+ public static void main(String[] args) {
+ for (int i = 0; i < 10; i++){
+ System.out.println(ShareCodeUtil.toSerialCode(1L));
+ }
}
@Override
@@ -64,4 +129,87 @@
Page<AiCompany> pages = this.baseMapper.selectPage(page, query);
return pages;
}
+
+ @Override
+ public AiCompanyWorkflow getWorkflowById(String id) {
+ return aiCompanyWorkflowMapper.selectById( id);
+ }
+
+ @Override
+ public AiCompanyWorkflow getWorkflowByTypeAndCompanyId(int type, String companyId) {
+
+ LambdaQueryWrapper<AiCompanyWorkflow> query = Wrappers.lambdaQuery(AiCompanyWorkflow.class);
+ query.eq(AiCompanyWorkflow::getCompanyId, companyId);
+ query.eq(AiCompanyWorkflow::getType, type);
+ query.last("limit 1");
+ return aiCompanyWorkflowMapper.selectOne(query);
+ }
+
+ @Override
+ public IPage<AiCompanyWorkflow> workflowListInPage(AiCompanyWorkflow dto, QueryRequest request) {
+ Page<AiCompanyWorkflow> page = new Page<>(request.getPageNum(), request.getPageSize());
+ LambdaQueryWrapper<AiCompanyWorkflow> query = Wrappers.lambdaQuery(AiCompanyWorkflow.class);
+ query.orderByDesc(AiCompanyWorkflow::getCreatedTime);
+ Page<AiCompanyWorkflow> pages = aiCompanyWorkflowMapper.selectPage(page, query);
+
+ List<AiCompanyWorkflow> records = pages.getRecords();
+ if (CollUtil.isNotEmpty( records)){
+ Set<String> collect = records.stream().map(AiCompanyWorkflow::getCompanyId).collect(Collectors.toSet());
+
+
+ Map<String, AiCompany> map = new HashMap<>();
+ if (CollUtil.isNotEmpty( collect)){
+ //set转list
+ List<String> collect1 = new ArrayList<>(collect);
+ List<AiCompany> listById = this.getListById((collect1));
+ map = listById.stream().collect(Collectors.toMap(AiCompany::getId, aiCompany -> aiCompany));
+ }
+
+ for (AiCompanyWorkflow aiCompanyWorkflow : records){
+ if (map.containsKey(aiCompanyWorkflow.getCompanyId())){
+ aiCompanyWorkflow.setCompanyName(map.get(aiCompanyWorkflow.getCompanyId()).getName());
+ }
+ }
+
+ }
+ return pages;
+ }
+
+ @Override
+ public FebsResponse addWorkflow(AiCompanyWorkflow dto) {
+ AiCompanyWorkflow entity = new AiCompanyWorkflow();
+ entity.setId(UUID.getSimpleUUIDString());
+ entity.setType(dto.getType());
+ entity.setCode(dto.getCode());
+ entity.setCreatedTime(new Date());
+ aiCompanyWorkflowMapper.insert( entity);
+ return new FebsResponse().success().message("操作成功");
+ }
+
+ @Override
+ public FebsResponse updateWorkflow(AiCompanyWorkflow dto) {
+ String id = dto.getId();
+ AiCompanyWorkflow entity = this.getWorkflowById(id);
+ if(ObjectUtil.isNotNull( entity)){
+ aiCompanyWorkflowMapper.update(null,
+ Wrappers.lambdaUpdate(AiCompanyWorkflow.class)
+ .set(AiCompanyWorkflow::getType, dto.getType())
+ .set(AiCompanyWorkflow::getCode, dto.getCode())
+ .set(AiCompanyWorkflow::getUpdatedTime, new Date())
+ .eq(AiCompanyWorkflow::getId, id));
+
+ }
+ return new FebsResponse().success().message("操作成功");
+ }
+
+ @Override
+ public void addWorkflow(String companyId, int type,String code, Date createTime) {
+ AiCompanyWorkflow entity = new AiCompanyWorkflow();
+ entity.setId(UUID.getSimpleUUIDString());
+ entity.setCompanyId(companyId);
+ entity.setType(type);
+ entity.setCode(code);
+ entity.setCreatedTime(createTime);
+ aiCompanyWorkflowMapper.insert( entity);
+ }
}
--
Gitblit v1.9.1