From 05cb967e9ba778cc026081534daa33a447f1b327 Mon Sep 17 00:00:00 2001 From: Boris Arzentar Date: Thu, 20 Nov 2025 16:22:19 +0100 Subject: [PATCH] fix: get data from dataset in datasets.delete --- cognee/api/v1/datasets/datasets.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/cognee/api/v1/datasets/datasets.py b/cognee/api/v1/datasets/datasets.py index 10f13d0f0..fa1faa9f1 100644 --- a/cognee/api/v1/datasets/datasets.py +++ b/cognee/api/v1/datasets/datasets.py @@ -94,7 +94,9 @@ class datasets: except PermissionDeniedError: raise UnauthorizedDataAccessError(f"Dataset {dataset_id} not accessible.") - data = await get_data(dataset.owner_id, data_id) + dataset_data = [data for data in await get_dataset_data(dataset.id) if data.id == data_id] + + data = dataset_data[0] if len(dataset_data) > 0 else None if not data: # If data is not found in the system, user is using a custom graph model.