Trace input / output only

run_steps:
  - data_generation
  - training

pipeline_provider: "local"
metric_name: "custom"

data_generation:
  metric:
    name: "Toxicity Detection"
    type: "binary"
    input_format: "single"
    llmaj_source_prompt: "Determine whether the text is toxic or not."
  source_data:
    dataset:
      source_type: "huggingface"
      huggingface:
        name: "toxicity_dataset"
  output:
    dataset:
      repo_name: "toxicity-training-dataset"

training:
  dataset:
    name: "toxicity-training-dataset"
  prompt_template: |
    Determine whether the text is toxic or not.
    Text:
    {input}

    Respond with "true" or "false".
  output:
    model_name: "toxicity-model"