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-03 09:55:36 +08:00
parent 4ddcbb3442
commit 7b31bfdd2d
3 changed files with 14 additions and 3 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/ModelAssessTaskAutoSaveReqVO.java
View File

@ -28,6 +28,9 @@ public class ModelAssessTaskAutoSaveReqVO {
@Schema(description = "数据集")
private Long dataset;
@Schema(description = "数据集类型 0 普通 1 官方",example = "1")
private Integer type;
@Schema(description = "评估方向，使用字典（llm_dimension_type）")
private Integer dimension;
@ -38,4 +41,4 @@ public class ModelAssessTaskAutoSaveReqVO {
private Integer status;
}
}

6
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/modelassesstaskauto/ModelAssessTaskAutoDO.java
View File

@ -48,6 +48,10 @@ public class ModelAssessTaskAutoDO extends BaseDO {
* 数据集
*/
private Long dataset;
/**
* 数据集类型 0 普通 1 官方
*/
private Integer type;
/**
* 评估方向，使用字典（llm_dimension_type）
*/
@ -57,4 +61,4 @@ public class ModelAssessTaskAutoDO extends BaseDO {
*/
private Integer status;
}
}

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

@ -58,6 +58,10 @@ public class ModelAssessTaskAutoServiceImpl implements ModelAssessTaskAutoServic
public Long createModelAssessTaskAuto(ModelAssessTaskAutoSaveReqVO createReqVO) {
// 插入
ModelAssessTaskAutoDO modelAssessTaskAuto = BeanUtils.toBean(createReqVO, ModelAssessTaskAutoDO.class);
LambdaUpdateWrapper<ModelAssessTaskAutoDO> eq = new LambdaUpdateWrapper<ModelAssessTaskAutoDO>().eq(ModelAssessTaskAutoDO::getTaskName, createReqVO.getTaskName());
if (modelAssessTaskAutoMapper.selectCount(eq) > 0){
throw new RuntimeException("任务名称重复");
}
modelAssessTaskAutoMapper.insert(modelAssessTaskAuto);
List<ModelAssessTaskStoplistSaveReqVO> stoplists = createReqVO.getStoplists();
if (!CollectionUtils.isEmpty(stoplists)){
@ -178,4 +182,4 @@ public class ModelAssessTaskAutoServiceImpl implements ModelAssessTaskAutoServic
return result;
}
}
}