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

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

2

3 from llm_dataset_engine.stages.data_loader_stage import DataLoaderStage

4 from llm_dataset_engine.stages.llm_invocation_stage import LLMInvocationStage

5 from llm_dataset_engine.stages.multi_run_stage import (

6 AggregationStrategy,

7 AllStrategy,

8 AverageStrategy,

9 ConsensusStrategy,

10 FirstSuccessStrategy,

11 MultiRunStage,

12)

13 from llm_dataset_engine.stages.pipeline_stage import PipelineStage

14 from llm_dataset_engine.stages.prompt_formatter_stage import (

15 PromptFormatterStage,

16)

17 from llm_dataset_engine.stages.response_parser_stage import (

18 JSONParser,

19 PydanticParser,

20 RawTextParser,

21 RegexParser,

22 ResponseParser,

23 ResponseParserStage,

24)

25 from 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