fix: moved function breaks Litestar imports. (#146)

advanced_alchemy/repository/_util.py

@@ -2,12 +2,17 @@
 
 from typing import TYPE_CHECKING, Any, cast
 
+from advanced_alchemy.exceptions import wrap_sqlalchemy_exception as _wrap_sqlalchemy_exception
+
 if TYPE_CHECKING:
     from sqlalchemy.orm import InstrumentedAttribute
 
     from advanced_alchemy.base import ModelProtocol
     from advanced_alchemy.repository.typing import ModelT
 
+# NOTE: For backward compatibility with Litestar - this is imported from here within the litestar codebase.
+wrap_sqlalchemy_exception = _wrap_sqlalchemy_exception
+
 
 def get_instrumented_attr(model: type[ModelProtocol], key: str | InstrumentedAttribute) -> InstrumentedAttribute:
     if isinstance(key, str):

tests/unit/test_extensions/test_litestar/test_litestar_re_exports.py

@@ -0,0 +1,10 @@
+# ruff: noqa: F401
+
+
+def test_repository_re_exports() -> None:
+    from litestar.contrib.sqlalchemy import types
+    from litestar.contrib.sqlalchemy.repository import (
+        SQLAlchemyAsyncRepository,
+        SQLAlchemySyncRepository,
+        wrap_sqlalchemy_exception,
+    )