Coverage for haystack/hooks/human_in_the_loop/__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 "dataclasses": ["ConfirmationUIResult", "ToolExecutionDecision"], 

12 "hooks": ["ConfirmationHook"], 

13 "policies": ["AlwaysAskPolicy", "AskOncePolicy", "NeverAskPolicy"], 

14 "strategies": ["BlockingConfirmationStrategy"], 

15 "user_interfaces": ["RichConsoleUI", "SimpleConsoleUI"], 

16} 

17 

18if TYPE_CHECKING: 

19 from .dataclasses import ConfirmationUIResult as ConfirmationUIResult 

20 from .dataclasses import ToolExecutionDecision as ToolExecutionDecision 

21 from .hooks import ConfirmationHook as ConfirmationHook 

22 from .policies import AlwaysAskPolicy as AlwaysAskPolicy 

23 from .policies import AskOncePolicy as AskOncePolicy 

24 from .policies import NeverAskPolicy as NeverAskPolicy 

25 from .strategies import BlockingConfirmationStrategy as BlockingConfirmationStrategy 

26 from .user_interfaces import RichConsoleUI as RichConsoleUI 

27 from .user_interfaces import SimpleConsoleUI as SimpleConsoleUI 

28 

29else: 

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