Update missed import refactor

NeonGeckoCom · Nov 4, 2024 · 24d4c58 · 24d4c58
1 parent 5c08464
commit 24d4c58
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/neon_hana/app/routers/auth.py b/neon_hana/app/routers/auth.py
@@ -28,7 +28,7 @@
 
 from neon_hana.app.dependencies import client_manager
 from neon_hana.schema.auth_requests import *
-from neon_users_service.models import User
+from neon_data_models.models.user import User
 
 auth_route = APIRouter(prefix="/auth", tags=["authentication"])
 

diff --git a/neon_hana/auth/client_manager.py b/neon_hana/auth/client_manager.py
@@ -38,7 +38,9 @@
 
 from neon_hana.auth.permissions import ClientPermissions
 from neon_hana.mq_service_api import MQServiceManager
-from neon_users_service.models import User, AccessRoles, TokenConfig, NeonUserConfig, PermissionsConfig
+from neon_data_models.models.user import (User, TokenConfig, NeonUserConfig,
+                                          PermissionsConfig)
+from neon_data_models.enum import AccessRoles
 
 _DEFAULT_USER_PERMISSIONS = PermissionsConfig(klat=AccessRoles.USER,
                                               core=AccessRoles.USER,

diff --git a/neon_hana/schema/auth_requests.py b/neon_hana/schema/auth_requests.py
@@ -30,7 +30,7 @@
 
 from pydantic import BaseModel, Field
 
-from neon_users_service.models import NeonUserConfig
+from neon_data_models.models.user import NeonUserConfig
 
 
 class AuthenticationRequest(BaseModel):