feat: add additional exports for the litestar plugin (#144)

litestar-org · Mar 19, 2024 · 2a8bf18 · 2a8bf18
1 parent 3364b6e
commit 2a8bf18
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/advanced_alchemy/extensions/litestar/__init__.py b/advanced_alchemy/extensions/litestar/__init__.py
@@ -1,4 +1,4 @@
-from advanced_alchemy import base, repository, service, types
+from advanced_alchemy import base, mixins, repository, service, types
 from advanced_alchemy.config import AlembicAsyncConfig, AlembicSyncConfig, AsyncSessionConfig, SyncSessionConfig
 
 from .alembic import AlembicCommands
@@ -11,12 +11,21 @@
     SQLAlchemySerializationPlugin,
     SQLAlchemySyncConfig,
 )
+from .plugins.init.config.asyncio import autocommit_before_send_handler as async_autocommit_before_send_handler
+from .plugins.init.config.asyncio import default_before_send_handler as async_default_before_send_handler
+from .plugins.init.config.sync import autocommit_before_send_handler as sync_autocommit_before_send_handler
+from .plugins.init.config.sync import default_before_send_handler as sync_default_before_send_handler
 
 __all__ = (
     "base",
     "types",
     "repository",
     "service",
+    "mixins",
+    "sync_autocommit_before_send_handler",
+    "async_autocommit_before_send_handler",
+    "sync_default_before_send_handler",
+    "async_default_before_send_handler",
     "AlembicCommands",
     "AlembicAsyncConfig",
     "AlembicSyncConfig",