MessageViewModel.java

@AppScoped
public class MessageViewModel extends ViewModel{
    @NonNull
    private final MessageRepository mRepository;

    @Inject
    public MessageViewModel(@NonNull MessageRepository repository) {
        this.mRepository = repository;
    }

    @NonNull
    public MessageUiModel getUiModel(){
        return new MessageUiModel(mRepository.getData());
    }
}