From 85069103bd0b1a801924c660daad78aedd0bfb6c Mon Sep 17 00:00:00 2001 From: Administrator <15274802129@163.com> Date: Thu, 18 Sep 2025 16:52:52 +0800 Subject: [PATCH] ``` feat(AiMemberTalkStreamService): 添加answerV3方法以支持流式生成报告 --- src/main/java/cc/mrbird/febs/ai/service/impl/AiProductCategoryServiceImpl.java | 52 +++++++++++++++++++++++++++++++++++++++++++++++----- 1 files changed, 47 insertions(+), 5 deletions(-) diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductCategoryServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductCategoryServiceImpl.java index 771f4ef..f59f046 100644 --- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductCategoryServiceImpl.java +++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiProductCategoryServiceImpl.java @@ -103,11 +103,53 @@ } @Override - public FebsResponse categoryList(ApiProductCategoryPageDto dto) { - // 创建分页对象,传入当前页和每页大小 - Page<ApiProductCategoryVo> page = new Page<>(dto.getPageNow(), dto.getPageSize()); - Page<ApiProductCategoryVo> pageListByQuery = this.getPageListByQuery(page, dto); - return new FebsResponse().success().data(pageListByQuery); + public FebsResponse hotV2(ApiProductCategoryHotDto dto) { + + List<ApiProductCategoryVo> list = new ArrayList<>(); + + LambdaQueryWrapper<AiProductCategory> query = Wrappers.lambdaQuery(AiProductCategory.class); + if (StrUtil.isNotEmpty(dto.getCompanyId())){ + query.eq(AiProductCategory::getCompanyId, dto.getCompanyId()); + }else{ + query.isNull(AiProductCategory::getCompanyId); + } + query.eq(AiProductCategory::getLevel, ProductCategoryLevelEnum.LEVEL_ONE.getLevel()); + query.eq(AiProductCategory::getState, 1); + query.eq(AiProductCategory::getHotState, 1); + query.orderByAsc(AiProductCategory::getSort); + List<AiProductCategory> listByQuery = this.getListByQuery(query); + if (CollUtil.isNotEmpty(listByQuery)){ + for (AiProductCategory aiProductCategory : listByQuery){ + ApiProductCategoryVo apiProductCategoryVo = new ApiProductCategoryVo(); + apiProductCategoryVo.setId(aiProductCategory.getId()); + apiProductCategoryVo.setName(aiProductCategory.getName()); + apiProductCategoryVo.setBackImg(aiProductCategory.getBackImg()); + apiProductCategoryVo.setIconImg(aiProductCategory.getIconImg()); + list.add(apiProductCategoryVo); + } + } + return new FebsResponse().success().data(list); + } + + @Override + public FebsResponse categoryChildList(ApiProductCategoryPageDto dto) { + List<ApiProductCategoryVo> vos = new ArrayList<>(); + + List<AiProductCategory> aiProductCategories = aiProductCategoryMapper.selectList( + Wrappers.lambdaQuery(AiProductCategory.class) + .eq(AiProductCategory::getParentId, dto.getParentId()) + .eq(AiProductCategory::getState, 1) + .orderByAsc(AiProductCategory::getSort) + ); + if (CollUtil.isNotEmpty(aiProductCategories)){ + for (AiProductCategory aiProductCategory : aiProductCategories) { + ApiProductCategoryVo vo = new ApiProductCategoryVo(); + vo.setId(aiProductCategory.getId()); + vo.setName(aiProductCategory.getName()); + vos.add(vo); + } + } + return new FebsResponse().success().data(vos); } @Override -- Gitblit v1.9.1