example_add_thinking.yaml

# Example configuration for adding thinking traces to existing Q&A data
# This demonstrates using the system to augment existing datasets

goal: Add thinking traces to existing Q&A pairs for reasoning model training
teacher_model: gemini-2.5-flash
api_type: vertex

generation_config:
  temperature: 0.7
  top_p: 0.9

vertex_config:
  project_id: your-project-id
  location: us-central1
  bucket_name: your-bucket-name

input_data_metadata:
  path: data/your_qa_data.parquet  # Can be parquet from previous round
  tokens_per_part: 200000000

data_gen_procedures:
  round_1:
    prompt_name: add_thinking_traces
    extraction_config: extraction_configs/add_thinking_traces.yaml
    has_branching: false  # 1:1 transformation
    validation:
      enabled: true
      filter_1: is_valid == True only if the value of column thinking is not null

output:
  output_format: format_templates/chat_format_qa_with_thinking.py