From f47bdb372e365723ccb16306f8199599926ed15b Mon Sep 17 00:00:00 2001 From: limin Date: Sat, 4 Jan 2025 21:30:07 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../ModelAssessTaskManualServiceImpl.java | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskmanual/ModelAssessTaskManualServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskmanual/ModelAssessTaskManualServiceImpl.java index 92dcc253a..176b58a74 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskmanual/ModelAssessTaskManualServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskmanual/ModelAssessTaskManualServiceImpl.java @@ -207,16 +207,22 @@ public class ModelAssessTaskManualServiceImpl implements ModelAssessTaskManualSe List serviceMyIds = modelAssessTaskManualDOPageResult.getList().stream(). filter(modelAssessTaskManualDO -> modelAssessTaskManualDO.getModelType() == 0) .map(ModelAssessTaskManualDO::getModelService).collect(Collectors.toList()); - List modelServiceDOS = modelServiceMapper.selectList(new LambdaQueryWrapper() - .in(ModelServiceDO::getId,serviceMyIds)); + List modelServiceDOS = new ArrayList<>(); + if (CollectionUtils.isNotEmpty(serviceMyIds)) { + modelServiceDOS = modelServiceMapper.selectList(new LambdaQueryWrapper() + .in(ModelServiceDO::getId, serviceMyIds)); + } Map serviceMyMaps = cn.iocoder.yudao.framework.common.util.collection. CollectionUtils.convertMap(modelServiceDOS, ModelServiceDO::getId); // 人工模型获取校验 List serviceForeseeIds = modelAssessTaskManualDOPageResult.getList().stream(). filter(modelAssessTaskManualDO -> modelAssessTaskManualDO.getModelType() == 1) .map(ModelAssessTaskManualDO::getModelService).collect(Collectors.toList()); - List baseModelDOS = baseModelMapper.selectList(new LambdaQueryWrapper() - .in(BaseModelDO::getId, serviceForeseeIds)); + List baseModelDOS = new ArrayList<>(); + if (CollectionUtils.isNotEmpty(serviceForeseeIds)) { + baseModelDOS = baseModelMapper.selectList(new LambdaQueryWrapper() + .in(BaseModelDO::getId, serviceForeseeIds)); + } Map serviceForeseeMaps = cn.iocoder.yudao.framework.common.util.collection. CollectionUtils.convertMap(baseModelDOS, BaseModelDO::getId); result.getList().stream().forEach(modelAssessTaskManualRespVO -> {