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

feat: Add multiuser support

Browse source
This commit is contained in:
Igor Ilic 2025-06-04 14:23:01 +02:00
parent 814c49332a
commit 289e395d0e
3 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
cognee/api/v1/crewai/routers/get_crewai_router.py
View file

@ -95,9 +95,9 @@ def get_crewai_router() -> APIRouter:
# from string import ascii_letters, digits
# hash6 = "".join(choice(ascii_letters + digits) for _ in range(6))
dataset_name = "final_reports"
await add(payload.feedback, node_set=["final_report"], dataset_name=dataset_name)
await cognify(datasets=dataset_name, is_stream_info_enabled=True)
dataset_name = "Github"
await add(payload.feedback, node_set=["final_report"], dataset_name=dataset_name, user=user)
await cognify(datasets=dataset_name, is_stream_info_enabled=True, user=user)
@router.websocket("/subscribe")
async def subscribe_to_crewai_info(websocket: WebSocket):

1
cognee/modules/users/authentication/auth0/auth0_jwt_strategy.py
View file

@ -28,6 +28,5 @@ class Auth0JWTStrategy(JWTStrategy):
new_user = BaseUserCreate(email=email, password="NOT IMPORTANT FOR AUTH0")
user = await user_manager.create(new_user)
await create_tenant(tenant_name="My organization", user_id=user.id)
return user

2
cognee/modules/users/models/ACL.py
View file

@ -13,7 +13,7 @@ class ACL(Base):
created_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
updated_at = Column(DateTime(timezone=True), onupdate=lambda: datetime.now(timezone.utc))
principal_id = Column(UUID, ForeignKey("principals.id"))
principal_id = Column(UUID, ForeignKey("principals.id", ondelete="CASCADE"))
permission_id = Column(UUID, ForeignKey("permissions.id"))
dataset_id = Column(UUID, ForeignKey("datasets.id", ondelete="CASCADE"))