Move AsyncMinHashLSH from experimental to datasketch.aio

datasketch/__init__.py

@@ -23,21 +23,27 @@
 WeightedMinHashLSH = MinHashLSH
 WeightedMinHashLSHForest = MinHashLSHForest
 
+# Optional async export (requires motor or redis.asyncio)
+try:
+    from datasketch.aio import AsyncMinHashLSH
+except ImportError:
+    AsyncMinHashLSH = None  # type: ignore[misc,assignment]
 
 __all__ = [
+    "AsyncMinHashLSH",
     "HNSW",
     "HyperLogLog",
     "HyperLogLogPlusPlus",
     "LeanMinHash",
     "MinHash",
     "MinHashLSH",
     "MinHashLSHBloom",
     "MinHashLSHEnsemble",
     "MinHashLSHForest",
     "WeightedMinHash",
     "WeightedMinHashGenerator",
     "WeightedMinHashLSH",
     "WeightedMinHashLSHForest",
     "bBitMinHash",
     "sha1_hash32",
 ]

datasketch/aio/__init__.py

@@ -0,0 +1,34 @@
+"""Async MinHash LSH module.
+
+This module provides asynchronous implementations of MinHash LSH for use with
+async storage backends like MongoDB (via motor) and Redis (via redis.asyncio).
+
+Example:
+    .. code-block:: python
+
+        from datasketch.aio import AsyncMinHashLSH
+        from datasketch import MinHash
+
+        async def main():
+            async with AsyncMinHashLSH(
+                storage_config={"type": "aiomongo", "mongo": {"host": "localhost", "port": 27017}},
+                threshold=0.5,
+                num_perm=128
+            ) as lsh:
+                m = MinHash(num_perm=128)
+                m.update(b"data")
+                await lsh.insert("key", m)
+                result = await lsh.query(m)
+"""
+
+from datasketch.aio.lsh import (
+    AsyncMinHashLSH,
+    AsyncMinHashLSHDeleteSession,
+    AsyncMinHashLSHInsertionSession,
+)
+
+__all__ = [
+    "AsyncMinHashLSH",
+    "AsyncMinHashLSHInsertionSession",
+    "AsyncMinHashLSHDeleteSession",
+]