diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java index dac0c4158..c1fa1eb76 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/dataset/DatasetServiceImpl.java @@ -125,6 +125,10 @@ public class DatasetServiceImpl implements DatasetService { if (annoCount == 0) { status = 0; } + if (status != 2 && dataset.getDatasetType() == 2){ + throw new ServiceException(new ErrorCode( + 20000, "评估数据集只能上传标注完成的数据")); + } if (dataset.getStatus() != status) { throw new ServiceException(new ErrorCode( 20000, "数据集标注状态错误!应该是【" + DatasetStatusEnum.getStatusByName(status) + "】")); @@ -209,6 +213,10 @@ public class DatasetServiceImpl implements DatasetService { if (annoCount == 0) { status=0; } + if (status != 2 && updateObj.getDatasetType() == 2){ + throw new ServiceException(new ErrorCode( + 20000, "评估数据集只能上传标注完成的数据")); + } if (updateObj.getStatus() != status) { throw new ServiceException(new ErrorCode( 20000, "数据集标注状态错误!应该是【" + DatasetStatusEnum.getStatusByName(status) + "】"));