2023-12-11 21:53:30 +00:00
|
|
|
from langchain_community.retrievers import __all__
|
2023-11-02 22:32:55 +00:00
|
|
|
|
|
|
|
EXPECTED_ALL = [
|
|
|
|
"AmazonKendraRetriever",
|
2023-11-28 22:10:23 +00:00
|
|
|
"AmazonKnowledgeBasesRetriever",
|
2023-11-02 22:32:55 +00:00
|
|
|
"ArceeRetriever",
|
|
|
|
"ArxivRetriever",
|
|
|
|
"AzureCognitiveSearchRetriever",
|
|
|
|
"ChatGPTPluginRetriever",
|
|
|
|
"ChaindeskRetriever",
|
|
|
|
"CohereRagRetriever",
|
|
|
|
"ElasticSearchBM25Retriever",
|
2023-11-20 01:35:03 +00:00
|
|
|
"EmbedchainRetriever",
|
2023-11-02 22:32:55 +00:00
|
|
|
"GoogleDocumentAIWarehouseRetriever",
|
|
|
|
"GoogleCloudEnterpriseSearchRetriever",
|
|
|
|
"GoogleVertexAIMultiTurnSearchRetriever",
|
|
|
|
"GoogleVertexAISearchRetriever",
|
|
|
|
"KayAiRetriever",
|
|
|
|
"KNNRetriever",
|
|
|
|
"LlamaIndexGraphRetriever",
|
|
|
|
"LlamaIndexRetriever",
|
|
|
|
"MetalRetriever",
|
|
|
|
"MilvusRetriever",
|
2023-11-27 02:56:12 +00:00
|
|
|
"OutlineRetriever",
|
2023-11-02 22:32:55 +00:00
|
|
|
"PineconeHybridSearchRetriever",
|
|
|
|
"PubMedRetriever",
|
2023-12-20 07:22:19 +00:00
|
|
|
"QdrantSparseVectorRetriever",
|
2023-11-02 22:32:55 +00:00
|
|
|
"RemoteLangChainRetriever",
|
|
|
|
"SVMRetriever",
|
|
|
|
"TavilySearchAPIRetriever",
|
|
|
|
"TFIDFRetriever",
|
|
|
|
"BM25Retriever",
|
|
|
|
"VespaRetriever",
|
|
|
|
"WeaviateHybridSearchRetriever",
|
|
|
|
"WikipediaRetriever",
|
|
|
|
"ZepRetriever",
|
|
|
|
"ZillizRetriever",
|
|
|
|
"DocArrayRetriever",
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
|
|
def test_all_imports() -> None:
|
|
|
|
assert set(__all__) == set(EXPECTED_ALL)
|