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

refactor: Use fastapi exception instead of custom exception

Browse source 
Return the use of fastapi exception instead of custom exception

Refactor COG-502
This commit is contained in:
Igor Ilic 2024-11-27 15:07:30 +01:00
parent 6eecc39db0
commit 90287e0dac
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/modules/users/get_user_manager.py
View file

@ -8,7 +8,7 @@ from fastapi_users.db import SQLAlchemyUserDatabase
from .get_user_db import get_user_db from .get_user_db import get_user_db
from .models import User from .models import User
from .methods import get_user from .methods import get_user
from cognee.exceptions import UserNotFoundError from fastapi_users.exceptions import UserNotExists
class UserManager(UUIDIDMixin, BaseUserManager[User, uuid.UUID]): class UserManager(UUIDIDMixin, BaseUserManager[User, uuid.UUID]):
@ -26,7 +26,7 @@ class UserManager(UUIDIDMixin, BaseUserManager[User, uuid.UUID]):
user = await get_user(id) user = await get_user(id)
if user is None: if user is None:
raise UserNotFoundError raise UserNotExists()
return user return user