1  # SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
 2  #
 3  # SPDX-License-Identifier: Apache-2.0
 4  
 5  import sys
 6  from typing import TYPE_CHECKING
 7  
 8  from lazy_imports import LazyImporter
 9  
10  _import_structure = {
11      "openai": ["OpenAIGenerator"],
12      "azure": ["AzureOpenAIGenerator"],
13      "hugging_face_local": ["HuggingFaceLocalGenerator"],
14      "hugging_face_api": ["HuggingFaceAPIGenerator"],
15      "openai_dalle": ["DALLEImageGenerator"],
16  }
17  
18  if TYPE_CHECKING:
19      from .azure import AzureOpenAIGenerator as AzureOpenAIGenerator
20      from .hugging_face_api import HuggingFaceAPIGenerator as HuggingFaceAPIGenerator
21      from .hugging_face_local import HuggingFaceLocalGenerator as HuggingFaceLocalGenerator
22      from .openai import OpenAIGenerator as OpenAIGenerator
23      from .openai_dalle import DALLEImageGenerator as DALLEImageGenerator
24  
25  else:
26      sys.modules[__name__] = LazyImporter(name=__name__, module_file=__file__, import_structure=_import_structure)