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

fix: unit tests

Browse source
This commit is contained in:
Boris Arzentar 2025-10-16 18:27:43 +02:00
parent a46b5010e2
commit cd8b952ca8
No known key found for this signature in database
GPG key ID: D5CC274C784807B7
3 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
cognee/api/v1/cognify/cognify.py
View file

@ -17,7 +17,6 @@ from cognee.modules.ontology.get_default_ontology_resolver import (
get_ontology_resolver_from_env, get_ontology_resolver_from_env,
) )
from cognee.modules.users.models import User from cognee.modules.users.models import User
from cognee.modules.users.methods import get_default_user
from cognee.tasks.documents import ( from cognee.tasks.documents import (
check_permissions_on_dataset, check_permissions_on_dataset,

2
cognee/api/v1/datasets/datasets.py
View file

@ -13,7 +13,7 @@ from cognee.modules.users.models import User
class datasets: class datasets:
@staticmethod @staticmethod
async def list_datasets(user: Optional[User] = None): async def list_datasets(user: Optional[User] = None):
if user == None: if user is None:
user = await get_default_user() user = await get_default_user()
return await get_authorized_existing_datasets([], "read", user) return await get_authorized_existing_datasets([], "read", user)

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

@ -8,6 +8,7 @@ from cognee.cli.exceptions import CliCommandException, CliCommandInnerException
from cognee.api.v1.datasets.datasets import datasets as cognee_datasets from cognee.api.v1.datasets.datasets import datasets as cognee_datasets
from cognee.modules.data.methods import get_datasets_by_name from cognee.modules.data.methods import get_datasets_by_name
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.users.methods import get_default_user, get_user
class DeleteCommand(SupportsCliCommand): class DeleteCommand(SupportsCliCommand):
@ -127,7 +128,12 @@ Be careful with deletion operations as they are irreversible.
"No user ID provided for deletion. Please specify using --user-id param." "No user ID provided for deletion. Please specify using --user-id param."
) )
await cognee_datasets.delete_dataset(dataset_id=dataset_id, user=args.user) if not args.user_id:
user = await get_default_user()
else:
user = await get_user(args.user_id)
await cognee_datasets.delete_dataset(dataset_id=dataset_id, user=user)
elif hasattr(args, "dataset_id") and hasattr(args, "data_id"): elif hasattr(args, "dataset_id") and hasattr(args, "data_id"):
await cognee_datasets.delete_data(args.dataset_id, args.data_id) await cognee_datasets.delete_data(args.dataset_id, args.data_id)
except Exception as e: except Exception as e: