1- from typing import Type , Union
1+ from typing import Type
22
33from fastapi import FastAPI
44from fastapi_amis_admin .admin import AdminSite , Settings
55from fastapi_amis_admin .amis .components import ActionType , App , Dialog , Flex , Service
66from fastapi_amis_admin .amis .constants import SizeEnum
77from fastapi_amis_admin .amis .types import AmisAPI
8+ from fastapi_amis_admin .crud .utils import SqlalchemyDatabase
89from fastapi_amis_admin .utils .translation import i18n as _
9- from sqlalchemy .ext .asyncio import AsyncEngine
10- from sqlalchemy .future import Engine
1110from starlette .requests import Request
1211
1312from fastapi_user_auth .app import UserAuthApp as DefaultUserAuthApp
@@ -18,7 +17,7 @@ class AuthAdminSite(AdminSite):
1817 auth : Auth = None
1918 UserAuthApp : Type [DefaultUserAuthApp ] = DefaultUserAuthApp
2019
21- def __init__ (self , settings : Settings , fastapi : FastAPI = None , engine : Union [ Engine , AsyncEngine ] = None , auth : Auth = None ):
20+ def __init__ (self , settings : Settings , fastapi : FastAPI = None , engine : SqlalchemyDatabase = None , auth : Auth = None ):
2221 super ().__init__ (settings , fastapi , engine )
2322 self .auth = auth or self .auth or Auth (db = self .db )
2423 self .register_admin (self .UserAuthApp )
0 commit comments