From 24d4c58dbe3aa506ed6435484b0b51e20af94a11 Mon Sep 17 00:00:00 2001
From: Daniel McKnight <daniel@neon.ai>
Date: Mon, 4 Nov 2024 11:50:07 -0800
Subject: [PATCH] Update missed import refactor

---
 neon_hana/app/routers/auth.py     | 2 +-
 neon_hana/auth/client_manager.py  | 4 +++-
 neon_hana/schema/auth_requests.py | 2 +-
 3 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/neon_hana/app/routers/auth.py b/neon_hana/app/routers/auth.py
index efee45e..eccb529 100644
--- 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
index 89904b3..b642baf 100644
--- 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
index 46463ee..e2e1619 100644
--- 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):