From cb3a53bc4b80ebe88da7ffcd7affb1cc779cf45a Mon Sep 17 00:00:00 2001
From: archer <545436317@qq.com>
Date: Mon, 8 Dec 2025 00:55:30 +0800
Subject: [PATCH] delete dataset

---
 packages/service/core/dataset/controller.ts | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/packages/service/core/dataset/controller.ts b/packages/service/core/dataset/controller.ts
index 6beb7f382..517e9de0c 100644
--- a/packages/service/core/dataset/controller.ts
+++ b/packages/service/core/dataset/controller.ts
@@ -95,6 +95,17 @@ export async function delDatasetRelevantData({
     teamId,
     datasetId: { $in: datasetIds }
   });
+  //Delete dataset_data_texts
+  await MongoDatasetDataText.deleteMany({
+    teamId,
+    datasetId: { $in: datasetIds }
+  });
+  //delete dataset_datas
+  await MongoDatasetData.deleteMany({ teamId, datasetId: { $in: datasetIds } });
+  // Delete collection image and file
+  await delCollectionRelatedSource({ collections });
+  // Delete vector data
+  await deleteDatasetDataVector({ teamId, datasetIds });
 
   for (const datasetId of datasetIds) {
     // Delete dataset_data_texts in batches by datasetId