|
| 1 | +# SDG API Simplification |
| 2 | + |
| 3 | +## Objective |
| 4 | + |
| 5 | +Identify simplifications to [the original SDG API design](sdg-api-interface.md) based on retrospective insights from working with the implementation of that design. |
| 6 | + |
| 7 | +## Original API Design |
| 8 | + |
| 9 | +Consider the original API sketch: |
| 10 | + |
| 11 | +```python |
| 12 | +from sdg import SDG |
| 13 | +from run_config import SynthDataFlow |
| 14 | +from pipeline import Pipeline |
| 15 | +import yaml |
| 16 | + |
| 17 | +client = openai_client(endpoint) |
| 18 | +model = "model-version" |
| 19 | + |
| 20 | +synth_skills_flow = SynthDataFlow(client, model).get_flow() |
| 21 | +skills_pipe = Pipeline(synth_skills_flow) |
| 22 | + |
| 23 | +cli_sdg = SDG([synth_skills_flow]) # run config has all the variables like num_samples, pipelinesteps etc |
| 24 | +generated_samples = cli_sdg.generate() |
| 25 | +``` |
| 26 | + |
| 27 | +The nouns above are: |
| 28 | + |
| 29 | +* Dataset - this is from Hugging Face's datasets library - used for the return value from `SDG.generate()`, but also what is passed between elements of the data generation pipeline |
| 30 | +* Block - not shown in the code above, but required to understand a pipeline - a block provides a `generate()` method transforms an input dataset and returns an output dataset |
| 31 | +* Block config - a description of how to instantiate and invoke a block, a sequence of these is returned from `get_flow()` above |
| 32 | +* Flow - a class which describes how to render a sequence of block configs for a pipeline |
| 33 | +* Pipeline - a pipeline is created from a sequence of block configs, and provides a generate() method in which it instantiates and invokes blocks in turn, passing the input dataset and collecting the output |
| 34 | +* SDG - an SDG is created from a list of pipelines, and its generate() method calls pipelines in turn |
| 35 | + |
| 36 | +## Simplification Proposals |
| 37 | + |
| 38 | +### Remove `SDG` |
| 39 | + |
| 40 | +We don't need both SDG and Pipeline since Pipeline can already do everything SDG can do. If more advanced orchestration of multiple pipelines is required later, an orchestration abstraction can be added then. |
| 41 | + |
| 42 | +### Remove `Flow` |
| 43 | + |
| 44 | +With flows migrating to a YAML file format (#109), their purpose becomes more clear - they are simply an expression of the configuration of a sequence of blocks, used to create a pipeline. We can simply refer to these YAML files as pipeline descriptions. |
| 45 | + |
| 46 | +### Add PipelineContext |
| 47 | + |
| 48 | +A number of runtime parameters are required by blocks in a pipeline - e.g. every `LLMBlock` requires an OpenAI API client and a model name. These parameters are distinct from configuration that is specified by a pipeline author. |
| 49 | + |
| 50 | +It would be much more straightforward if `Block` were able to access these runtime parameters via their parent `Pipeline`. |
| 51 | + |
| 52 | +In the case where multiple pipelines with the same runtime context is desired, it would also be beneficial to abstract these runtime parameters into a `PipelineContext` class. |
| 53 | + |
| 54 | +## New API Design |
| 55 | + |
| 56 | +```python |
| 57 | +ds = Dataset.from_list(samples) |
| 58 | + |
| 59 | +ctx = PipelineContext(client, "mixtral", teacher_model) |
| 60 | + |
| 61 | +knowledge_pipe = Pipeline.from_configs(ctx, [MMLU_BENCH_PIPELINE, SYNTH_KNOWLEDGE_PIPELINE]) |
| 62 | + |
| 63 | +gen_data = knowledge_pipe.generate(ds) |
| 64 | +``` |
0 commit comments