Coverage for haystack/components/generators/chat/__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 "openai": ["OpenAIChatGenerator"], 

12 "openai_responses": ["OpenAIResponsesChatGenerator"], 

13 "azure": ["AzureOpenAIChatGenerator"], 

14 "azure_responses": ["AzureOpenAIResponsesChatGenerator"], 

15 "fallback": ["FallbackChatGenerator"], 

16 "llm": ["LLM"], 

17 "mock": ["MockChatGenerator"], 

18} 

19 

20if TYPE_CHECKING: 

21 from .azure import AzureOpenAIChatGenerator as AzureOpenAIChatGenerator 

22 from .azure_responses import AzureOpenAIResponsesChatGenerator as AzureOpenAIResponsesChatGenerator 

23 from .fallback import FallbackChatGenerator as FallbackChatGenerator 

24 from .llm import LLM as LLM 

25 from .mock import MockChatGenerator as MockChatGenerator 

26 from .openai import OpenAIChatGenerator as OpenAIChatGenerator 

27 from .openai_responses import OpenAIResponsesChatGenerator as OpenAIResponsesChatGenerator 

28 

29else: 

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