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 84318ff88..4e96d6d82 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 @@ -87,6 +87,9 @@ public class DatasetServiceImpl implements DatasetService { dataset.setAnnotateProgress(formattedRatio); } dataset.setStatus(status); + if(annoCount == 0){ + dataset.setStatus(0); + } datasetMapper.updateById(dataset); } return dataset.getId(); @@ -132,6 +135,9 @@ public class DatasetServiceImpl implements DatasetService { updateObj.setAnnotateProgress(formattedRatio); } updateObj.setStatus(status); + if(annoCount == 0){ + updateObj.setStatus(0); + } } datasetMapper.updateById(updateObj); }