Coverage for llm_dataset_engine/stages/__init__.py: 100%

1"""Processing stages for data transformation.""" 

2 

3from llm_dataset_engine.stages.data_loader_stage import DataLoaderStage 

4from llm_dataset_engine.stages.llm_invocation_stage import LLMInvocationStage 

5from llm_dataset_engine.stages.multi_run_stage import ( 

6 AggregationStrategy, 

7 AllStrategy, 

8 AverageStrategy, 

9 ConsensusStrategy, 

10 FirstSuccessStrategy, 

11 MultiRunStage, 

12) 

13from llm_dataset_engine.stages.pipeline_stage import PipelineStage 

14from llm_dataset_engine.stages.prompt_formatter_stage import ( 

15 PromptFormatterStage, 

16) 

17from llm_dataset_engine.stages.response_parser_stage import ( 

18 JSONParser, 

19 PydanticParser, 

20 RawTextParser, 

21 RegexParser, 

22 ResponseParser, 

23 ResponseParserStage, 

24) 

25from llm_dataset_engine.stages.result_writer_stage import ResultWriterStage 

26 

27__all__ = [ 

28 "PipelineStage", 

29 "DataLoaderStage", 

30 "PromptFormatterStage", 

31 "LLMInvocationStage", 

32 "ResponseParserStage", 

33 "ResultWriterStage", 

34 "MultiRunStage", 

35 "ResponseParser", 

36 "RawTextParser", 

37 "JSONParser", 

38 "PydanticParser", 

39 "RegexParser", 

40 "AggregationStrategy", 

41 "ConsensusStrategy", 

42 "FirstSuccessStrategy", 

43 "AllStrategy", 

44 "AverageStrategy", 

45] 

46