import json,pathlib,sys
root=pathlib.Path(sys.argv[1])
for mode in ['block','context']:
 messages=[[{'role':m['role'],'text':m['text']} for m in json.loads((root/f'{side}-{mode}-final.json').read_text())['messages']] for side in ['base','head']]
 assert messages[0]==messages[1], mode
 (root/f'{mode}-messages.json').write_text(json.dumps(messages[0],indent=2)+'\n')
 print(f'{mode}: base and head user/assistant message text matches byte for byte')