DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

修改自动评估备份返回值类型

Browse Source
This commit is contained in:
sunxiqing 2025-01-24 17:50:15 +08:00
parent 2e255c059f
commit ae03d22e4a
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/vo/ModelAssessTaskAutoRespVO.java
View File

@ -61,4 +61,9 @@ public class ModelAssessTaskAutoRespVO {
@Schema(description = "备份ID")
private Long backupId;
/**
* 评估任务ID
*/
private Long taskId;
}

5
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskmanualbackup/ModelAssessTaskManualBackupServiceImpl.java
View File

@ -272,7 +272,8 @@ public class ModelAssessTaskManualBackupServiceImpl implements ModelAssessTaskMa
public PageResult<ModelAssessTaskAutoRespVO> getModelAssessTaskAutoPage1 (ModelAssessTaskAutoPageReqVO pageReqVO) {
PageResult<ModelAssessTaskAutoBackupDO> modelAssessTaskAutoDOPageResult = modelAssessTaskAutoBaackupMapper.selectPage(pageReqVO);
PageResult<ModelAssessTaskAutoRespVO> result = BeanUtils.toBean(modelAssessTaskAutoDOPageResult, ModelAssessTaskAutoRespVO.class);
if (CollectionUtils.isNotEmpty(result.getList())) {
List<ModelAssessTaskAutoRespVO> resultList = result.getList();
if (CollectionUtils.isNotEmpty(resultList)) {
// 我的模型获取校验
List<Long> serviceIds = modelAssessTaskAutoDOPageResult.getList().stream().
filter(modelAssessTaskManualDO -> modelAssessTaskManualDO.getModelType() == 0)
@ -296,7 +297,7 @@ public class ModelAssessTaskManualBackupServiceImpl implements ModelAssessTaskMa
serviceForeseeMaps = cn.iocoder.yudao.framework.common.util.collection.
CollectionUtils.convertMap(baseModelDOS, BaseModelDO::getId);
}
for (ModelAssessTaskAutoRespVO modelAssessTaskManualRespVO : result.getList()) {
for (ModelAssessTaskAutoRespVO modelAssessTaskManualRespVO : resultList) {
if (modelAssessTaskManualRespVO.getModelType() == 0 && longModelServiceDOMap != null) {
ModelServiceDO modelServiceDO = longModelServiceDOMap.get(modelAssessTaskManualRespVO.getModelService());
if (modelServiceDO != null) {