[
  {
    "instruction": "问题：曰:贫而无谄富而无骄何如?",
    "input": "",
    "output": "可也、未若贫而乐富而好礼者也、"
  },

  {
    "instruction": "问题：曰:夫子至于是邦也必闻其政求之与?抑与之与?",
    "input": "",
    "output": "夫子温、良、恭、俭、让以得之、夫子之求之也其诸异乎人之求之与?"
  },

  {
    "instruction": "问题：何谓四恶?",
    "input": "",
    "output": "不教而杀谓之虐;不戒视成谓之暴;慢令致期谓之贼;犹之与人也出纳之吝谓之有司、"
  }
]

import argparse
import json
from tqdm import tqdm


def format_example(example: dict) -> dict:
    context = f"Instruction: {example['instruction']}\n"
    if example.get("input"):
        context += f"Input: {example['input']}\n"
    context += "Answer: "
    target = example["output"]
    return {"context": context, "target": target}


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--data_path", type=str, default="./answers-lunyu.json")
    parser.add_argument("--save_path", type=str, default="./answers-lunyu.jsonl")

    args = parser.parse_args()
    with open(args.data_path,encoding='utf-8') as f:
        examples = json.load(f)

    with open(args.save_path, 'w', encoding='utf-8') as f:
        for example in tqdm(examples, desc="formatting.."):
            f.write(json.dumps(format_example(example)) + '\n')


if __name__ == "__main__":
    main()