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

cleanup of delete dataset methods

Browse source
This commit is contained in:
rajeevrajeshuni 2025-11-30 07:29:42 +05:30
parent 40846df92d
commit 1ac2c24cf5
3 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/cli/commands/delete_command.py
View file

@ -6,7 +6,7 @@ from cognee.cli import DEFAULT_DOCS_URL
import cognee.cli.echo as fmt import cognee.cli.echo as fmt
from cognee.cli.exceptions import CliCommandException, CliCommandInnerException from cognee.cli.exceptions import CliCommandException, CliCommandInnerException
from cognee.modules.data.methods.get_deletion_counts import get_deletion_counts 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.data.methods.delete_data_by_user import delete_data_by_user
from cognee.modules.users.methods import get_default_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: if args.dataset_name:
# Use delete_datasets_by_name for dataset deletion # Use delete_datasets_by_name for dataset deletion
user = await get_default_user() 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"]: if result["not_found"]:
fmt.warning(f"Dataset '{args.dataset_name}' not found") fmt.warning(f"Dataset '{args.dataset_name}' not found")

4
cognee/cli/tui/delete_screen.py
View file

@ -6,7 +6,7 @@ from textual.containers import Container, Vertical, Horizontal
from textual.binding import Binding from textual.binding import Binding
from cognee.cli.tui.base_screen import BaseTUIScreen 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.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.data.methods.delete_data_by_user import delete_data_by_user
from cognee.modules.users.methods import get_default_user from cognee.modules.users.methods import get_default_user
@ -151,7 +151,7 @@ class DeleteTUIScreen(BaseTUIScreen):
if dataset_name: if dataset_name:
# Use delete_datasets_by_name for dataset deletion # Use delete_datasets_by_name for dataset deletion
user = await get_default_user() 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"]: if result["not_found"]:
status.update(f"⚠️ Dataset '{dataset_name}' not found") status.update(f"⚠️ Dataset '{dataset_name}' not found")

2
cognee/modules/data/methods/__init__.py
View file

@ -16,7 +16,7 @@ from .get_dataset_ids import get_dataset_ids
# Delete # Delete
from .delete_dataset import delete_dataset 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_by_user import delete_data_by_user
from .delete_data import delete_data from .delete_data import delete_data