From c15363be6a42b55308ae7fa66907bbaeea743db2 Mon Sep 17 00:00:00 2001 From: sunxiqing <2240398334@qq.com> Date: Fri, 3 Jan 2025 14:49:04 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E8=AF=84=E4=BC=B0=E6=8A=A5?= =?UTF-8?q?=E5=91=8A=E6=9F=A5=E7=9C=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../vo/ModelAssessTaskAutoRespVO.java | 5 ++- .../ModelAssessTaskAutoServiceImpl.java | 33 ++++++++++++++++--- 2 files changed, 32 insertions(+), 6 deletions(-) diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/vo/ModelAssessTaskAutoRespVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/vo/ModelAssessTaskAutoRespVO.java index ef206e785..ca7c04d2f 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/vo/ModelAssessTaskAutoRespVO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/vo/ModelAssessTaskAutoRespVO.java @@ -33,6 +33,9 @@ public class ModelAssessTaskAutoRespVO { @ExcelProperty("模型服务") private Long modelService; + @Schema(description = "模型类型（0普通，1官方）") + private Integer modelType; + @Schema(description = "数据集") @ExcelProperty("数据集") private Long dataset; @@ -56,4 +59,4 @@ public class ModelAssessTaskAutoRespVO { @Schema(description = "数据集名称") private String datasetName; -} \ No newline at end of file +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoServiceImpl.java index f82a6e740..f9674f6a2 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoServiceImpl.java @@ -1,11 +1,13 @@ package cn.iocoder.yudao.module.llm.service.modelassesstaskauto; import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanual.vo.ModelAssessTaskManualRespVO; +import cn.iocoder.yudao.module.llm.dal.dataobject.basemodel.BaseModelDO; import cn.iocoder.yudao.module.llm.dal.dataobject.dataset.DatasetDO; import cn.iocoder.yudao.module.llm.dal.dataobject.modelassessstoplist.ModelAssessStoplistDO; import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskauto.ModelAssessTaskStoplistDO; import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskmanual.ModelAssessTaskManualDO; import cn.iocoder.yudao.module.llm.dal.dataobject.modelservice.ModelServiceDO; +import cn.iocoder.yudao.module.llm.dal.mysql.basemodel.BaseModelMapper; import cn.iocoder.yudao.module.llm.dal.mysql.dataset.DatasetMapper; import cn.iocoder.yudao.module.llm.dal.mysql.modelassessstoplist.ModelAssessStoplistMapper; import cn.iocoder.yudao.module.llm.dal.mysql.modelassesstaskauto.ModelAssessTaskStoplistMapper; @@ -52,6 +54,8 @@ public class ModelAssessTaskAutoServiceImpl implements ModelAssessTaskAutoServic private DatasetMapper datasetMapper; @Resource private ModelAssessStoplistMapper modelAssessStoplistMapper; + @Resource + private BaseModelMapper baseModelMapper; @Override @@ -166,17 +170,36 @@ public class ModelAssessTaskAutoServiceImpl implements ModelAssessTaskAutoServic PageResult modelAssessTaskAutoDOPageResult = modelAssessTaskAutoMapper.selectPage(pageReqVO); PageResult result = BeanUtils.toBean(modelAssessTaskAutoDOPageResult, ModelAssessTaskAutoRespVO.class); if (CollectionUtils.isNotEmpty(result.getList())){ - List serviceIds = modelAssessTaskAutoDOPageResult.getList().stream().map(ModelAssessTaskAutoDO::getModelService).collect(Collectors.toList()); + // 我的模型获取校验 + List serviceIds = modelAssessTaskAutoDOPageResult.getList().stream(). + filter(modelAssessTaskManualDO -> modelAssessTaskManualDO.getModelType() == 0) + .map(ModelAssessTaskAutoDO::getModelService).collect(Collectors.toList()); LambdaQueryWrapper wrapper = new LambdaQueryWrapper<>(); wrapper.in(ModelServiceDO::getId,serviceIds); List modelServiceDOS = modelServiceMapper.selectList(wrapper); Map longModelServiceDOMap = cn.iocoder.yudao.framework.common.util.collection. CollectionUtils.convertMap(modelServiceDOS, ModelServiceDO::getId); - result.getList().stream().forEach(modelAssessTaskAutoRespVO -> { - ModelServiceDO modelServiceDO = longModelServiceDOMap.get(modelAssessTaskAutoRespVO.getModelService()); - if(modelServiceDO != null){ - modelAssessTaskAutoRespVO.setModelServiceName(modelServiceDO.getServiceName()); + // 人工模型获取校验 + List serviceForeseeIds = modelAssessTaskAutoDOPageResult.getList().stream(). + filter(modelAssessTaskManualDO -> modelAssessTaskManualDO.getModelType() == 1) + .map(ModelAssessTaskAutoDO::getModelService).collect(Collectors.toList()); + List 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 -> { + if (modelAssessTaskManualRespVO.getModelType() == 0){ + ModelServiceDO modelServiceDO = longModelServiceDOMap.get(modelAssessTaskManualRespVO.getModelService()); + if(modelServiceDO != null){ + modelAssessTaskManualRespVO.setModelServiceName(modelServiceDO.getServiceName()); + } + }else { + BaseModelDO baseModelDO = serviceForeseeMaps.get(modelAssessTaskManualRespVO.getModelService()); + if(baseModelDO != null){ + modelAssessTaskManualRespVO.setModelServiceName(baseModelDO.getModelName()); + } } + }); } return result;