langchain/libs/partners/elasticsearch/tests/unit_tests/test_imports.py
Max Jakob 61a2eba081
elasticsearch[patch]: add top-level import, remove obsolete dependency (#18644)
Make `ElasticsearchRetriever` available as top-level import.

The `langchain` package depends on `langchain-community` so we do not
need to depend on it explicitly.
2024-03-07 19:38:31 -08:00

16 lines
367 B
Python

from langchain_elasticsearch import __all__
EXPECTED_ALL = [
"ApproxRetrievalStrategy",
"ElasticsearchChatMessageHistory",
"ElasticsearchEmbeddings",
"ElasticsearchRetriever",
"ElasticsearchStore",
"ExactRetrievalStrategy",
"SparseRetrievalStrategy",
]
def test_all_imports() -> None:
assert sorted(EXPECTED_ALL) == sorted(__all__)