Coverage for haystack/components/embedders/__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 "azure_document_embedder": ["AzureOpenAIDocumentEmbedder"], 

12 "azure_text_embedder": ["AzureOpenAITextEmbedder"], 

13 "mock_document_embedder": ["MockDocumentEmbedder"], 

14 "mock_text_embedder": ["MockTextEmbedder"], 

15 "openai_document_embedder": ["OpenAIDocumentEmbedder"], 

16 "openai_text_embedder": ["OpenAITextEmbedder"], 

17} 

18 

19if TYPE_CHECKING: 

20 from .azure_document_embedder import AzureOpenAIDocumentEmbedder as AzureOpenAIDocumentEmbedder 

21 from .azure_text_embedder import AzureOpenAITextEmbedder as AzureOpenAITextEmbedder 

22 from .mock_document_embedder import MockDocumentEmbedder as MockDocumentEmbedder 

23 from .mock_text_embedder import MockTextEmbedder as MockTextEmbedder 

24 from .openai_document_embedder import OpenAIDocumentEmbedder as OpenAIDocumentEmbedder 

25 from .openai_text_embedder import OpenAITextEmbedder as OpenAITextEmbedder 

26 

27else: 

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