feat(ai): 添加AI产品对话轮数功能

Administrator

1 days ago e1583cd7e9c87f8b008757372debc5217e4e3d6e

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

@@ -55,6 +55,9 @@
        if (StrUtil.isNotEmpty(dto.getProductCategoryId())){
            query.eq(AiProduct::getProductCategoryId, dto.getProductCategoryId());
        }
        if (StrUtil.isNotEmpty(dto.getCompanyId())){
            query.eq(AiProduct::getCompanyId, dto.getCompanyId());
        }
        query.ne(AiProduct::getState, 2);
        query.orderByDesc(AiProduct::getHotState);
        query.orderByAsc(AiProduct::getSort);
@@ -113,6 +116,7 @@
        entity.setTarget(dto.getTarget());
        entity.setDescription(dto.getDescription());
        entity.setQuestionCount(dto.getQuestionCount());
        entity.setTalkCnt(dto.getTalkCnt());
        entity.setCreatedTime(new Date());
        this.save(entity);
        return new FebsResponse().success().message("操作成功");
@@ -135,6 +139,7 @@
                            .set(AiProduct::getTarget, dto.getTarget())
                            .set(AiProduct::getDescription, dto.getDescription())
                            .set(AiProduct::getQuestionCount, dto.getQuestionCount())
                            .set(AiProduct::getTalkCnt, dto.getTalkCnt())
                            .set(AiProduct::getUpdatedTime, new Date())
                            .eq(AiProduct::getId, id)
            );
@@ -158,11 +163,13 @@
    }

    @Override
    public List<AiProduct> selectList() {
        return aiProductMapper.selectList(
                Wrappers.lambdaQuery(AiProduct.class)
                .ne(AiProduct::getState, 2)
        );
    public List<AiProduct> selectList(String companyId) {
        LambdaQueryWrapper<AiProduct> query = Wrappers.lambdaQuery(AiProduct.class);
        if (StrUtil.isNotEmpty(companyId)){
            query.eq(AiProduct::getCompanyId, companyId);
        }
        query.ne(AiProduct::getState, 2);
        return aiProductMapper.selectList(query);
    }

    @Override
@@ -172,15 +179,18 @@
        List<String> chooseIds = dto.getChooseIds();
        AiProduct aiProduct = this.getById(chooseId);
        if (ObjectUtil.isNotNull(aiProduct)) {
            String companyId = aiProduct.getCompanyId();
            aiProductPointLinkService.deleteByQuery(
                    Wrappers.lambdaQuery(AiProductPointLink.class)
                            .eq(AiProductPointLink::getProductId,chooseId)
                            .eq(AiProductPointLink::getCompanyId,companyId)
            );
            if(CollUtil.isNotEmpty(chooseIds)){
                Date createdTime = new Date();
                for (String item : chooseIds){
                    AiProductPointLink entity = new AiProductPointLink();
                    entity.setId(UUID.getSimpleUUIDString());
                    entity.setCompanyId(companyId);
                    entity.setProductId(chooseId);
                    entity.setProductPointId(item);
                    entity.setCreatedTime(createdTime);
@@ -198,15 +208,18 @@
        List<String> chooseIds = dto.getChooseIds();
        AiProduct aiProduct = this.getById(chooseId);
        if (ObjectUtil.isNotNull(aiProduct)) {
            String companyId = aiProduct.getCompanyId();
            aiProductRoleLinkService.deleteByQuery(
                    Wrappers.lambdaQuery(AiProductRoleLink.class)
                            .eq(AiProductRoleLink::getProductId,chooseId)
                    .eq(AiProductRoleLink::getCompanyId,companyId)
            );
            if(CollUtil.isNotEmpty(chooseIds)){
                Date createdTime = new Date();
                for (String item : chooseIds){
                    AiProductRoleLink entity = new AiProductRoleLink();
                    entity.setId(UUID.getSimpleUUIDString());
                    entity.setCompanyId(companyId);
                    entity.setProductId(chooseId);
                    entity.setProductRoleId(item);
                    entity.setCreatedTime(createdTime);
@@ -223,15 +236,18 @@
        List<String> chooseIds = dto.getChooseIds();
        AiProduct aiProduct = this.getById(chooseId);
        if (ObjectUtil.isNotNull(aiProduct)) {
            String companyId = aiProduct.getCompanyId();
            aiProductQuestionLinkService.deleteByQuery(
                    Wrappers.lambdaQuery(AiProductQuestionLink.class)
                            .eq(AiProductQuestionLink::getProductId,chooseId)
                    .eq(AiProductQuestionLink::getCompanyId,companyId)
            );
            if(CollUtil.isNotEmpty(chooseIds)){
                Date createdTime = new Date();
                for (String item : chooseIds){
                    AiProductQuestionLink entity = new AiProductQuestionLink();
                    entity.setId(UUID.getSimpleUUIDString());
                    entity.setCompanyId(companyId);
                    entity.setProductId(chooseId);
                    entity.setProductQuestionId(item);
                    entity.setCreatedTime(createdTime);
@@ -241,4 +257,18 @@
        }
        return new FebsResponse().success().message("操作成功");
    }

    @Override
    public List<AiProduct> listByCategory(AiProduct dto) {
        LambdaQueryWrapper<AiProduct> query = Wrappers.lambdaQuery(AiProduct.class);
        if (StrUtil.isNotEmpty(dto.getProductCategoryId())){
            query.eq(AiProduct::getProductCategoryId, dto.getProductCategoryId());
        }
        if (StrUtil.isNotEmpty(dto.getCompanyId())){
            query.eq(AiProduct::getCompanyId, dto.getCompanyId());
        }
        query.ne(AiProduct::getState, 2);
        query.orderByAsc(AiProduct::getSort);
        return aiProductMapper.selectList(query);
    }
}