Coverage for haystack/token_counters/__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 "approximate_counter": ["ApproximateTokenCounter"], 

12 "openai_counter": ["OpenAITokenCounter"], 

13 "types": ["TokenCounter"], 

14 "tiktoken_counter": ["TiktokenCounter"], 

15} 

16 

17if TYPE_CHECKING: 

18 from .approximate_counter import ApproximateTokenCounter as ApproximateTokenCounter 

19 from .openai_counter import OpenAITokenCounter as OpenAITokenCounter 

20 from .tiktoken_counter import TiktokenCounter as TiktokenCounter 

21 from .types import TokenCounter as TokenCounter 

22else: 

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