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

refactor: Rename auth layer

Browse source
This commit is contained in:
Igor Ilic 2025-08-26 15:58:53 +02:00
parent 950d29a678
commit 5ea3056564
2 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/modules/pipelines/layers/authorized_user_datasets.py → cognee/modules/pipelines/layers/resolve_authorized_user_datasets.py
View file

@ -12,7 +12,7 @@ from cognee.modules.data.methods import (
)
async def authorized_user_datasets(
async def resolve_authorized_user_datasets(
datasets: Union[str, UUID, list[str], list[UUID]], user: User = None
) -> Tuple[User, List[Dataset]]:
"""

6
cognee/modules/pipelines/operations/pipeline.py
View file

@ -12,7 +12,9 @@ from cognee.modules.pipelines.tasks.task import Task
from cognee.modules.users.models import User
from cognee.modules.pipelines.operations import log_pipeline_run_initiated
from cognee.context_global_variables import set_database_global_context_variables
from cognee.modules.pipelines.layers.authorized_user_datasets import authorized_user_datasets
from cognee.modules.pipelines.layers.resolve_authorized_user_datasets import (
resolve_authorized_user_datasets,
)
from cognee.modules.pipelines.layers.process_pipeline_check import process_pipeline_check
from cognee.infrastructure.databases.relational import (
@ -67,7 +69,7 @@ async def cognee_pipeline(
await test_embedding_connection()
cognee_pipeline.first_run = False # Update flag after first run
user, authorized_datasets = await authorized_user_datasets(datasets, user)
user, authorized_datasets = await resolve_authorized_user_datasets(datasets, user)
for dataset in authorized_datasets:
async for run_info in run_pipeline(