QuivrHQ/quivr
1
1
Fork 0
mirror of https://github.com/QuivrHQ/quivr.git synced 2024-12-15 09:32:22 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix: add user id while creating default brain (#1616)

Browse Source
This commit is contained in:
Mamadou DICKO 2023-11-13 12:16:38 +01:00 committed by GitHub
parent 5dc4f15a6c
commit f24779ff14
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/repository/brain/get_default_user_brain_or_create_new.py
View File

@ -1,17 +1,18 @@
from models import BrainEntity, UserIdentity from models import BrainEntity, UserIdentity
from models.databases.supabase.brains import CreateBrainProperties from models.databases.supabase.brains import CreateBrainProperties
from routes.authorizations.types import RoleEnum
from repository.brain import create_brain, create_brain_user, get_user_default_brain from repository.brain import create_brain, create_brain_user, get_user_default_brain
from repository.brain.create_brain import create_brain from repository.brain.create_brain import create_brain
from repository.brain.create_brain_user import create_brain_user from repository.brain.create_brain_user import create_brain_user
from repository.brain.get_default_user_brain import get_user_default_brain from repository.brain.get_default_user_brain import get_user_default_brain
from routes.authorizations.types import RoleEnum
def get_default_user_brain_or_create_new(user: UserIdentity) -> BrainEntity: def get_default_user_brain_or_create_new(user: UserIdentity) -> BrainEntity:
default_brain = get_user_default_brain(user.id) default_brain = get_user_default_brain(user.id)
if not default_brain: if not default_brain:
default_brain = create_brain(CreateBrainProperties()) default_brain = create_brain(brain=CreateBrainProperties(), user_id=user.id)
create_brain_user(user.id, default_brain.brain_id, RoleEnum.Owner, True) create_brain_user(user.id, default_brain.brain_id, RoleEnum.Owner, True)
return default_brain return default_brain