Add record ids to the documents of split datasets