From 100bf5fa74c8685150bd7c1640a057e52d55a206 Mon Sep 17 00:00:00 2001 From: limin Date: Wed, 22 Jan 2025 14:31:34 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E5=AE=8C=E6=88=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../module/llm/service/dataset/DatasetServiceImpl.java | 8 ++++++++ 1 file changed, 8 insertions(+) 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) + "】"));