MuratcanKoylan/Marketing-Memory-Routing-8B

1import json2from synthetic_data.pipeline import SyntheticDataPipeline3 4def generate_sample():5    pipeline = SyntheticDataPipeline()6    print("Generating sample batch...")7    results = pipeline.run_batch(count=2, category="company.brand_core")8    9    with open("synthetic_data/sample_batch.json", "w") as f:10        json.dump(results, f, indent=2)11    print(f"Saved {len(results)} examples to synthetic_data/sample_batch.json")12 13if __name__ == "__main__":14    generate_sample()15 16