lazy import lancedb

akashmangoai · fm1320 · commit d91638ed7d1f · 2024-12-10T14:40:44.000Z
diff --git a/adalflow/adalflow/components/retriever/__init__.py b/adalflow/adalflow/components/retriever/__init__.py
@@ -27,13 +27,19 @@
     OptionalPackages.QDRANT,
 )
 
+LanceDBRetriever = LazyImport(
+    "adalflow.components.retriever.lancedb_retriver.LanceDBRetriever",
+    OptionalPackages.LANCEDB,
+)
+
 __all__ = [
     "BM25Retriever",
     "LLMRetriever",
     "FAISSRetriever",
     "RerankerRetriever",
     "PostgresRetriever",
     "QdrantRetriever",
+    "LanceDBRetriever",
     "split_text_by_word_fn",
     "split_text_by_word_fn_then_lower_tokenized",
 ]
