Changed sqlalchemy adapter base model typing

Author: miladmahmoodi

archipy/adapters/base/sqlalchemy/adapters.py

@@ -165,7 +165,7 @@ class SQLAlchemySortMixin:
     """
 
     @staticmethod
-    def _apply_sorting(entity: type[BaseEntity], query: Select, sort_info: SortDTO | None) -> Select:
+    def _apply_sorting(entity: type[T], query: Select, sort_info: SortDTO | None) -> Select:
         """Apply sorting to a SQLAlchemy query.
 
         Args:
@@ -238,12 +238,12 @@ def _create_session_manager(self, configs: ConfigT) -> BaseSQLAlchemySessionMana
     @override
     def execute_search_query(
         self,
-        entity: type[BaseEntity],
+        entity: type[T],
         query: Select,
         pagination: PaginationDTO | None = None,
         sort_info: SortDTO | None = None,
         has_multiple_entities: bool = False,
-    ) -> tuple[list[BaseEntity], int]:
+    ) -> tuple[list[T], int]:
         """Execute a search query with pagination and sorting.
 
         Args:
@@ -398,7 +398,7 @@ def get_by_uuid(self, entity_type: type[T], entity_uuid: UUID) -> T | None:
             return result
 
     @override
-    def delete(self, entity: BaseEntity) -> None:
+    def delete(self, entity: T) -> None:
         """Delete an entity from the database.
 
         Args:
@@ -427,7 +427,7 @@ def delete(self, entity: BaseEntity) -> None:
             self._handle_db_exception(e, self.session_manager._get_database_name())
 
     @override
-    def bulk_delete(self, entities: list[BaseEntity]) -> None:
+    def bulk_delete(self, entities: list[T]) -> None:
         """Delete multiple entities from the database.
 
         Args:
@@ -549,12 +549,12 @@ def _create_async_session_manager(self, configs: ConfigT) -> AsyncBaseSQLAlchemy
     @override
     async def execute_search_query(
         self,
-        entity: type[BaseEntity],
+        entity: type[T],
         query: Select,
         pagination: PaginationDTO | None,
         sort_info: SortDTO | None = None,
         has_multiple_entities: bool = False,
-    ) -> tuple[list[BaseEntity], int]:
+    ) -> tuple[list[T], int]:
         """Execute a search query with pagination and sorting.
 
         Args:
@@ -739,7 +739,7 @@ async def delete(self, entity: BaseEntity) -> None:
             self._handle_db_exception(e, self.session_manager._get_database_name())
 
     @override
-    async def bulk_delete(self, entities: list[BaseEntity]) -> None:
+    async def bulk_delete(self, entities: list[T]) -> None:
         """Delete multiple entities from the database.
 
         Args: