package cc.mrbird.febs.ai.service.impl;
|
|
import cc.mrbird.febs.ai.entity.AiCompany;
|
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;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
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;
|
|
@Slf4j
|
@Service
|
@RequiredArgsConstructor
|
public class AiCompanyServiceImpl extends ServiceImpl<AiCompanyMapper, AiCompany> implements AiCompanyService {
|
|
private final AiCompanyMapper aiCompanyMapper;
|
private final AiCompanyWorkflowMapper aiCompanyWorkflowMapper;
|
|
@Override
|
public AiCompany getById(String id) {
|
return aiCompanyMapper.selectById( id);
|
}
|
|
@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.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
|
public FebsResponse update(AiCompany dto) {
|
String id = dto.getId();
|
AiCompany entity = this.getById(id);
|
if(ObjectUtil.isNotNull( entity)){
|
this.update(null,
|
Wrappers.lambdaUpdate(AiCompany.class)
|
.set(AiCompany::getName, dto.getName())
|
.set(AiCompany::getUpdatedTime, new Date())
|
.eq(AiCompany::getId, id));
|
|
}
|
return new FebsResponse().success().message("操作成功");
|
}
|
|
@Override
|
public IPage<AiCompany> listInPage(AiCompany dto, QueryRequest request) {
|
Page<AiCompany> page = new Page<>(request.getPageNum(), request.getPageSize());
|
LambdaQueryWrapper<AiCompany> query = Wrappers.lambdaQuery(AiCompany.class);
|
query.orderByDesc(AiCompany::getCreatedTime);
|
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);
|
}
|
}
|
