import copy import jsonlines data_path = '/home/ubuntu/sas0/lzy/mars-mcp/generate_data/agent_questions_solution_turn5_ans_no_none.jsonl' #output_path = './agent_questions_solutions_qwq1.jsonl' with jsonlines.open(data_path, mode='r') as reader: datas = [line for line in reader if len(line['messages']) == 4] print(datas[0])