Coverage for haystack/components/retrievers/__init__.py: 100%

1# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai> 

2# 

3# SPDX-License-Identifier: Apache-2.0 

4 

5import sys 

6from typing import TYPE_CHECKING 

7 

8from lazy_imports import LazyImporter 

9 

10_import_structure = { 

11 "auto_merging_retriever": ["AutoMergingRetriever"], 

12 "filter_retriever": ["FilterRetriever"], 

13 "in_memory": ["InMemoryBM25Retriever", "InMemoryEmbeddingRetriever"], 

14 "multi_retriever": ["MultiRetriever"], 

15 "multi_query_embedding_retriever": ["MultiQueryEmbeddingRetriever"], 

16 "multi_query_text_retriever": ["MultiQueryTextRetriever"], 

17 "sentence_window_retriever": ["SentenceWindowRetriever"], 

18 "text_embedding_retriever": ["TextEmbeddingRetriever"], 

19} 

20 

21if TYPE_CHECKING: 

22 from .auto_merging_retriever import AutoMergingRetriever as AutoMergingRetriever 

23 from .filter_retriever import FilterRetriever as FilterRetriever 

24 from .in_memory import InMemoryBM25Retriever as InMemoryBM25Retriever 

25 from .in_memory import InMemoryEmbeddingRetriever as InMemoryEmbeddingRetriever 

26 from .multi_query_embedding_retriever import MultiQueryEmbeddingRetriever as MultiQueryEmbeddingRetriever 

27 from .multi_query_text_retriever import MultiQueryTextRetriever as MultiQueryTextRetriever 

28 from .multi_retriever import MultiRetriever as MultiRetriever 

29 from .sentence_window_retriever import SentenceWindowRetriever as SentenceWindowRetriever 

30 from .text_embedding_retriever import TextEmbeddingRetriever as TextEmbeddingRetriever 

31 

32else: 

33 sys.modules[__name__] = LazyImporter(name=__name__, module_file=__file__, import_structure=_import_structure)