diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetAnswerMapper.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetAnswerMapper.java index 63b0e4d3f..ed48605fa 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetAnswerMapper.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetAnswerMapper.java @@ -33,4 +33,6 @@ public interface DatasetAnswerMapper extends BaseMapperX { void deleteTrue(@Param("datasetPostId") Long datasetPostId); List getAnswersToYourQuestions(@Param("collected") List collected); + + void deleteTheAnswer(@Param("id") Long id); } diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetQuestionMapper.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetQuestionMapper.java index f47b62cf2..27a004916 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetQuestionMapper.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/dataset/DatasetQuestionMapper.java @@ -35,4 +35,6 @@ public interface DatasetQuestionMapper extends BaseMapperX { void deleteTrue(@Param("datasetPostId") Long datasetPostId); List getAListOfIssues(@Param("datasetId") Long datasetId); + + void deleteTheIssue(@Param("id") Long id); } 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 055a6c2f4..8bc0d4ddc 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 @@ -258,6 +258,10 @@ public class DatasetServiceImpl implements DatasetService { validateDatasetExists(id); // 校验使用 validateDatasetUse(id); + // 删除问题(数据量太大直接删除) + datasetQuestionMapper.deleteTheIssue(id); + // 删除答案 + datasetAnswerMapper.deleteTheAnswer(id); // 删除 datasetMapper.deleteById(id); } diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/dataset/DatasetAnswerMapper.xml b/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/dataset/DatasetAnswerMapper.xml index 11c667c57..c0aa2cd97 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/dataset/DatasetAnswerMapper.xml +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/dataset/DatasetAnswerMapper.xml @@ -1,6 +1,9 @@ + + DELETE FROM llm_dataset_answer WHERE dataset_id = #{id} +