Coverage for haystack/hooks/compaction/__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 "hooks": ["CompactionHook"], 

12 "sliding_window": ["SlidingWindowCompactor"], 

13 "summarization": ["SummarizationCompactor"], 

14 "tool_result_offloading": ["ToolResultOffloadCompactor"], 

15 "tool_result_pruning": ["ToolResultPruningCompactor"], 

16 "types": ["Compactor"], 

17} 

18 

19if TYPE_CHECKING: 

20 from .hooks import CompactionHook as CompactionHook 

21 from .sliding_window import SlidingWindowCompactor as SlidingWindowCompactor 

22 from .summarization import SummarizationCompactor as SummarizationCompactor 

23 from .tool_result_offloading import ToolResultOffloadCompactor as ToolResultOffloadCompactor 

24 from .tool_result_pruning import ToolResultPruningCompactor as ToolResultPruningCompactor 

25 from .types import Compactor as Compactor 

26else: 

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