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