From 1ac2c24cf5eedc656c3ce59ef6162bb23b5b32ea Mon Sep 17 00:00:00 2001 From: rajeevrajeshuni Date: Sun, 30 Nov 2025 07:29:42 +0530 Subject: [PATCH] cleanup of delete dataset methods --- cognee/cli/commands/delete_command.py | 4 ++-- cognee/cli/tui/delete_screen.py | 4 ++-- cognee/modules/data/methods/__init__.py | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/cognee/cli/commands/delete_command.py b/cognee/cli/commands/delete_command.py index d50571fca..df0a1cbe3 100644 --- a/cognee/cli/commands/delete_command.py +++ b/cognee/cli/commands/delete_command.py @@ -6,7 +6,7 @@ from cognee.cli import DEFAULT_DOCS_URL import cognee.cli.echo as fmt from cognee.cli.exceptions import CliCommandException, CliCommandInnerException from cognee.modules.data.methods.get_deletion_counts import get_deletion_counts -from cognee.modules.data.methods.delete_dataset_by_name import delete_datasets_by_name +from cognee.modules.data.methods.delete_dataset_by_name import delete_dataset_by_name from cognee.modules.data.methods.delete_data_by_user import delete_data_by_user from cognee.modules.users.methods import get_default_user @@ -98,7 +98,7 @@ Be careful with deletion operations as they are irreversible. if args.dataset_name: # Use delete_datasets_by_name for dataset deletion user = await get_default_user() - result = await delete_datasets_by_name(args.dataset_name, user.id) + result = await delete_dataset_by_name(args.dataset_name, user.id) if result["not_found"]: fmt.warning(f"Dataset '{args.dataset_name}' not found") diff --git a/cognee/cli/tui/delete_screen.py b/cognee/cli/tui/delete_screen.py index 80bb9586d..9b7e45c16 100644 --- a/cognee/cli/tui/delete_screen.py +++ b/cognee/cli/tui/delete_screen.py @@ -6,7 +6,7 @@ from textual.containers import Container, Vertical, Horizontal from textual.binding import Binding from cognee.cli.tui.base_screen import BaseTUIScreen from cognee.modules.data.methods.get_deletion_counts import get_deletion_counts -from cognee.modules.data.methods.delete_dataset_by_name import delete_datasets_by_name +from cognee.modules.data.methods.delete_dataset_by_name import delete_dataset_by_name from cognee.modules.data.methods.delete_data_by_user import delete_data_by_user from cognee.modules.users.methods import get_default_user @@ -151,7 +151,7 @@ class DeleteTUIScreen(BaseTUIScreen): if dataset_name: # Use delete_datasets_by_name for dataset deletion user = await get_default_user() - result = await delete_datasets_by_name(dataset_name, user.id) + result = await delete_dataset_by_name(dataset_name, user.id) if result["not_found"]: status.update(f"⚠️ Dataset '{dataset_name}' not found") diff --git a/cognee/modules/data/methods/__init__.py b/cognee/modules/data/methods/__init__.py index 17073cd08..da8d64dea 100644 --- a/cognee/modules/data/methods/__init__.py +++ b/cognee/modules/data/methods/__init__.py @@ -16,7 +16,7 @@ from .get_dataset_ids import get_dataset_ids # Delete from .delete_dataset import delete_dataset -from .delete_dataset_by_name import delete_datasets_by_name +from .delete_dataset_by_name import delete_dataset_by_name from .delete_data_by_user import delete_data_by_user from .delete_data import delete_data