expose retrievers via root module

deepset-ai · Nov 30, 2023 · f21b4e6 · f21b4e6
1 parent 1e0f44a
commit f21b4e6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/integrations/opensearch/src/opensearch_haystack/__init__.py b/integrations/opensearch/src/opensearch_haystack/__init__.py
@@ -1,3 +1,5 @@
 from opensearch_haystack.document_store import OpenSearchDocumentStore
+from opensearch_haystack.bm25_retriever import OpenSearchBM25Retriever
+from opensearch_haystack.embedding_retriever import OpenSearchEmbeddingRetriever
 
-__all__ = ["OpenSearchDocumentStore"]
+__all__ = ["OpenSearchDocumentStore", "OpenSearchBM25Retriever", "OpenSearchEmbeddingRetriever"]