From dc03a52541449b55af8a00cfecfc4081f3e2022c Mon Sep 17 00:00:00 2001 From: vasilije Date: Sat, 19 Jul 2025 19:50:36 +0200 Subject: [PATCH] fixes to delete --- cognee/tasks/ingestion/ingest_data.py | 2 +- cognee/tasks/ingestion/plugin_ingest_data.py | 23 ++++++++++++++------ 2 files changed, 17 insertions(+), 8 deletions(-) diff --git a/cognee/tasks/ingestion/ingest_data.py b/cognee/tasks/ingestion/ingest_data.py index 846c183d4..21b91d55f 100644 --- a/cognee/tasks/ingestion/ingest_data.py +++ b/cognee/tasks/ingestion/ingest_data.py @@ -60,7 +60,7 @@ async def ingest_data( else: # Find existing dataset or create a new one existing_datasets = await get_authorized_existing_datasets( - user=user, permission_type="write", datasets=[dataset_name] + datasets=[dataset_name], permission_type="write", user=user ) dataset = await load_or_create_datasets( dataset_names=[dataset_name], diff --git a/cognee/tasks/ingestion/plugin_ingest_data.py b/cognee/tasks/ingestion/plugin_ingest_data.py index 9e7562da6..aefdf4924 100644 --- a/cognee/tasks/ingestion/plugin_ingest_data.py +++ b/cognee/tasks/ingestion/plugin_ingest_data.py @@ -120,13 +120,22 @@ async def plugin_ingest_data( logger.info(f"Plugin-based ingestion starting for dataset: {dataset_name}") # Preserve existing dataset creation and permission logic - existing_datasets = await get_authorized_existing_datasets([dataset_name], "write", user) - - # Use dataset_id if provided, otherwise use dataset_name - dataset_names = [dataset_id] if dataset_id else [dataset_name] - datasets = await load_or_create_datasets(dataset_names, existing_datasets, user) - - dataset = datasets[0] + if dataset_id: + # Retrieve existing dataset by ID + dataset = await get_specific_user_permission_datasets(user.id, "write", [dataset_id]) + # Convert from list to Dataset element + if isinstance(dataset, list): + dataset = dataset[0] + else: + # Find existing dataset or create a new one by name + existing_datasets = await get_authorized_existing_datasets( + datasets=[dataset_name], permission_type="write", user=user + ) + datasets = await load_or_create_datasets( + dataset_names=[dataset_name], existing_datasets=existing_datasets, user=user + ) + if isinstance(datasets, list): + dataset = datasets[0] new_datapoints = [] existing_data_points = []