From 90287e0dac72e9146698193ea9927fd2e005a2b3 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Wed, 27 Nov 2024 15:07:30 +0100 Subject: [PATCH] refactor: Use fastapi exception instead of custom exception Return the use of fastapi exception instead of custom exception Refactor COG-502 --- cognee/modules/users/get_user_manager.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cognee/modules/users/get_user_manager.py b/cognee/modules/users/get_user_manager.py index 0e8adaebd..30410a985 100644 --- a/cognee/modules/users/get_user_manager.py +++ b/cognee/modules/users/get_user_manager.py @@ -8,7 +8,7 @@ from fastapi_users.db import SQLAlchemyUserDatabase from .get_user_db import get_user_db from .models import User from .methods import get_user -from cognee.exceptions import UserNotFoundError +from fastapi_users.exceptions import UserNotExists class UserManager(UUIDIDMixin, BaseUserManager[User, uuid.UUID]): @@ -26,7 +26,7 @@ class UserManager(UUIDIDMixin, BaseUserManager[User, uuid.UUID]): user = await get_user(id) if user is None: - raise UserNotFoundError + raise UserNotExists() return user