The methods in QueryStoreWriter now accept `EntityDocument` rather than just `Entity`