gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fixes to delete

Browse source
This commit is contained in:
vasilije 2025-07-19 19:50:36 +02:00
parent f0ba618f0c
commit dc03a52541
2 changed files with 17 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/tasks/ingestion/ingest_data.py
View file

@ -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],

23
cognee/tasks/ingestion/plugin_ingest_data.py
View file

@ -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 = []