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:
limin 2024-12-27 15:51:51 +08:00
parent e2a24cd81e
commit eb52395737
4 changed files with 47 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/modelassesstaskauto/ModelAssessTaskAutoController.java
View File

@ -72,21 +72,17 @@ public class ModelAssessTaskAutoController {
@Parameter(name = "id", description = "编号", required = true, example = "1024")
@PreAuthorize("@ss.hasPermission('llm:model-assess-task-auto:query')")
public CommonResult<ModelAssessTaskAutoRespVO> getModelAssessTaskAuto(@RequestParam("id") Long id) {
ModelAssessTaskAutoDO modelAssessTaskAuto = modelAssessTaskAutoService.getModelAssessTaskAuto(id);
ModelAssessTaskAutoRespVO bean = BeanUtils.toBean(modelAssessTaskAuto, ModelAssessTaskAutoRespVO.class);
List<ModelAssessTaskStoplistDO> modelAssessTaskStoplistDOS = modelingTaskStoplistMapper.selectList(new LambdaQueryWrapper<ModelAssessTaskStoplistDO>()
.eq(ModelAssessTaskStoplistDO::getTaskId, id)
.eq(ModelAssessTaskStoplistDO::getDeleted, false));
bean.setStoplists(modelAssessTaskStoplistDOS);
return success(bean);
ModelAssessTaskAutoRespVO result = modelAssessTaskAutoService.getModelAssessTaskAuto(id);
return success(result);
}
@GetMapping("/page")
@Operation(summary = "获得模型评估任务 —— 自动评估分页")
@PreAuthorize("@ss.hasPermission('llm:model-assess-task-auto:query')")
public CommonResult<PageResult<ModelAssessTaskAutoRespVO>> getModelAssessTaskAutoPage(@Valid ModelAssessTaskAutoPageReqVO pageReqVO) {
PageResult<ModelAssessTaskAutoDO> pageResult = modelAssessTaskAutoService.getModelAssessTaskAutoPage(pageReqVO);
return success(BeanUtils.toBean(pageResult, ModelAssessTaskAutoRespVO.class));
PageResult<ModelAssessTaskAutoRespVO> pageResult = modelAssessTaskAutoService.getModelAssessTaskAutoPage1(pageReqVO);
return success(pageResult);
}
@GetMapping("/export-excel")

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

@ -51,4 +51,6 @@ public class ModelAssessTaskAutoRespVO {
@Schema(description = "模型评估任务状态，使用字典（llm_model_assess_task_status）")
private Integer status;
private String modelServiceName;
}

3
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoService.java
View File

@ -42,7 +42,7 @@ public interface ModelAssessTaskAutoService {
* @param id 编号
* @return 模型评估任务 —— 自动评估
*/
ModelAssessTaskAutoDO getModelAssessTaskAuto(Long id);
ModelAssessTaskAutoRespVO getModelAssessTaskAuto(Long id);
/**
* 获得模型评估任务 —— 自动评估分页
@ -52,4 +52,5 @@ public interface ModelAssessTaskAutoService {
*/
PageResult<ModelAssessTaskAutoDO> getModelAssessTaskAutoPage(ModelAssessTaskAutoPageReqVO pageReqVO);
PageResult<ModelAssessTaskAutoRespVO> getModelAssessTaskAutoPage1(ModelAssessTaskAutoPageReqVO pageReqVO);
}

40
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskauto/ModelAssessTaskAutoServiceImpl.java
View File

@ -1,7 +1,11 @@
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.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.modelassesstaskauto.ModelAssessTaskStoplistMapper;
import cn.iocoder.yudao.module.llm.dal.mysql.modelservice.ModelServiceMapper;
import com.alibaba.druid.util.StringUtils;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
@ -38,6 +42,9 @@ public class ModelAssessTaskAutoServiceImpl implements ModelAssessTaskAutoServic
private ModelAssessTaskAutoMapper modelAssessTaskAutoMapper;
@Resource
private ModelAssessTaskStoplistMapper modelingTaskStoplistMapper;
@Resource
private ModelServiceMapper modelServiceMapper;
@Override
public Long createModelAssessTaskAuto(ModelAssessTaskAutoSaveReqVO createReqVO) {
@ -99,13 +106,42 @@ public class ModelAssessTaskAutoServiceImpl implements ModelAssessTaskAutoServic
}
@Override
public ModelAssessTaskAutoDO getModelAssessTaskAuto(Long id) {
return modelAssessTaskAutoMapper.selectById(id);
public ModelAssessTaskAutoRespVO getModelAssessTaskAuto(Long id) {
ModelAssessTaskAutoDO modelAssessTaskAutoDO = modelAssessTaskAutoMapper.selectById(id);
ModelAssessTaskAutoRespVO result = BeanUtils.toBean(modelAssessTaskAutoDO, ModelAssessTaskAutoRespVO.class);
List<ModelAssessTaskStoplistDO> modelAssessTaskStoplistDOS = modelingTaskStoplistMapper.selectList(new LambdaQueryWrapper<ModelAssessTaskStoplistDO>()
.eq(ModelAssessTaskStoplistDO::getTaskId, id)
.eq(ModelAssessTaskStoplistDO::getDeleted, false));
result.setStoplists(modelAssessTaskStoplistDOS);
ModelServiceDO modelServiceDO = modelServiceMapper.selectById(modelAssessTaskAutoDO.getModelService());
if (modelServiceDO != null){
result.setModelServiceName(modelServiceDO.getServiceName());
}
return result;
}
@Override
public PageResult<ModelAssessTaskAutoDO> getModelAssessTaskAutoPage(ModelAssessTaskAutoPageReqVO pageReqVO) {
return modelAssessTaskAutoMapper.selectPage(pageReqVO);
}
@Override
public PageResult<ModelAssessTaskAutoRespVO> getModelAssessTaskAutoPage1(ModelAssessTaskAutoPageReqVO pageReqVO) {
PageResult<ModelAssessTaskAutoDO> modelAssessTaskAutoDOPageResult = modelAssessTaskAutoMapper.selectPage(pageReqVO);
PageResult<ModelAssessTaskAutoRespVO> result = BeanUtils.toBean(modelAssessTaskAutoDOPageResult, ModelAssessTaskAutoRespVO.class);
List<Long> serviceIds = modelAssessTaskAutoDOPageResult.getList().stream().map(ModelAssessTaskAutoDO::getModelService).collect(Collectors.toList());
LambdaQueryWrapper<ModelServiceDO> wrapper = new LambdaQueryWrapper<>();
wrapper.in(ModelServiceDO::getId,serviceIds);
List<ModelServiceDO> modelServiceDOS = modelServiceMapper.selectList(wrapper);
Map<Long, ModelServiceDO> 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());
}
});
return result;
}
}