feat(ai):为公司实体添加编码字段- 在 AiCompany 实体中新增 code 字段 - 使用 ShareCodeUtil生成唯一编码 - ...

Administrator

yesterday d22072910b38c3f9c71f01f02f68094426c543c8

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

@@ -2,12 +2,14 @@

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.UUID;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.entity.QueryRequest;
import cc.mrbird.febs.common.utils.ShareCodeUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
@@ -48,18 +50,26 @@
        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(),1,null,createTime);
        this.addWorkflow(entity.getId(),2,null,createTime);
        this.addWorkflow(entity.getId(),3,null,createTime);
        this.addWorkflow(entity.getId(),4,null,createTime);
        this.addWorkflow(entity.getId(),6,null,createTime);
        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);

        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
@@ -92,6 +102,16 @@
    }

    @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);