Coverage for haystack/hooks/tool_result_offloading/__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": ["ToolResultOffloadHook", "RESULT_STORE_CONTEXT_KEY"], 

12 "policies": ["AlwaysOffload", "NeverOffload", "OffloadOverChars"], 

13 "stores": ["FileSystemToolResultStore"], 

14 "types": ["OffloadPolicy", "ToolResultStore"], 

15} 

16 

17if TYPE_CHECKING: 

18 from .hooks import RESULT_STORE_CONTEXT_KEY as RESULT_STORE_CONTEXT_KEY 

19 from .hooks import ToolResultOffloadHook as ToolResultOffloadHook 

20 from .policies import AlwaysOffload as AlwaysOffload 

21 from .policies import NeverOffload as NeverOffload 

22 from .policies import OffloadOverChars as OffloadOverChars 

23 from .stores import FileSystemToolResultStore as FileSystemToolResultStore 

24 from .types import OffloadPolicy as OffloadPolicy 

25 from .types import ToolResultStore as ToolResultStore 

26else: 

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