Похоже, я отправил этот вопрос в мозг, но, возможно, это поможет кому-то еще в будущем. Все, что вам нужно сделать, это установить ключи при выделении сущностей:
Entity entity = new Entity(key);
Или, если вы ранее извлекли сущности из хранилища данных, ключи уже должны быть установлены.