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

Fixes to the ACL model

Browse source
This commit is contained in:
Vasilije 2024-07-27 12:48:03 +02:00
parent b4d1a739be
commit 7930586017
2 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
cognee/api/v1/cognify/cognify_v2.py
View file

@ -10,7 +10,7 @@ from sqlalchemy.ext.asyncio import AsyncSession
from cognee.infrastructure.databases.graph import get_graph_config from cognee.infrastructure.databases.graph import get_graph_config
from cognee.infrastructure.databases.relational.user_authentication.authentication_db import async_session_maker from cognee.infrastructure.databases.relational.user_authentication.authentication_db import async_session_maker
from cognee.infrastructure.databases.relational.user_authentication.users import has_permission_document, \ from cognee.infrastructure.databases.relational.user_authentication.users import has_permission_document, \
get_user_permissions, get_async_session_context get_user_permissions, get_async_session_context, fast_api_users_init
# from cognee.infrastructure.databases.relational.user_authentication.authentication_db import async_session_maker # from cognee.infrastructure.databases.relational.user_authentication.authentication_db import async_session_maker
# from cognee.infrastructure.databases.relational.user_authentication.users import get_user_permissions, fastapi_users # from cognee.infrastructure.databases.relational.user_authentication.users import get_user_permissions, fastapi_users
from cognee.modules.cognify.config import get_cognify_config from cognee.modules.cognify.config import get_cognify_config
@ -40,7 +40,7 @@ class PermissionDeniedException(Exception):
self.message = message self.message = message
super().__init__(self.message) super().__init__(self.message)
async def cognify(datasets: Union[str, list[str]] = None, root_node_id: str = None, user_id:str="default_user"): async def cognify(datasets: Union[str, list[str]] = None, root_node_id: str = None):
relational_config = get_relationaldb_config() relational_config = get_relationaldb_config()
db_engine = relational_config.database_engine db_engine = relational_config.database_engine
@ -57,8 +57,9 @@ async def cognify(datasets: Union[str, list[str]] = None, root_node_id: str = No
file["name"] = file["name"].replace(" ", "_") file["name"] = file["name"].replace(" ", "_")
async with get_async_session_context() as session: async with get_async_session_context() as session:
active_user = await fast_api_users_init()
out = await has_permission_document(user_id, file["id"], "write", session) out = await has_permission_document(active_user.current_user(active=True), file["id"], "write", session)
async with update_status_lock: async with update_status_lock:

3
cognee/infrastructure/databases/relational/user_authentication/users.py
View file

@ -61,6 +61,9 @@ auth_backend = AuthenticationBackend(
fastapi_users = FastAPIUsers[User, uuid.UUID](get_user_manager, [auth_backend]) fastapi_users = FastAPIUsers[User, uuid.UUID](get_user_manager, [auth_backend])
async def fast_api_users_init():
return fastapi_users
current_active_user = fastapi_users.current_user(active=True) current_active_user = fastapi_users.current_user(active=True)