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 87a405e50..ca4a5d2c6 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 @@ -321,6 +321,18 @@ public class ModelAssessTaskManualServiceImpl implements ModelAssessTaskManualSe ModelAssessTaskManualDO::getId, ModelAssessTaskManualDO::getTaskName, modelServiceId); } + /** + * 根据 数据集id 获取 人工评估 + * + * @param datasetId 数据集id + * @return 人工评估 + */ + @Override + public Map getModelAssessmentTaskByDatasetId (Long datasetId) { + return getModelAssessmentTaskMap((wrapper, id) -> wrapper.eq(ModelAssessTaskManualDO::getDataset, datasetId), + ModelAssessTaskManualDO::getId, ModelAssessTaskManualDO::getTaskName, datasetId); + } + @Override public List gitViewTheReportDetails (Long id) { List manualEvaluationReportRespVOS = new ArrayList<>();