example2.kt

import androidx.hilt.Assisted
import androidx.hilt.lifecycle.ViewModelInject
import androidx.lifecycle.LiveData
import androidx.lifecycle.SavedStateHandle
import androidx.lifecycle.ViewModel

class ExampleViewModel @ViewModelInject constructor(
    private val someUseCase: SomeUseCase,
    @Assisted private val savedStateHandle: SavedStateHandle
) : ViewModel() {
    //Get plain argument
    val argumentA: String = savedStateHandle.get<>(KEY_ARGUMENT_A)

    // Get argument as LiveData
    val argumentA: LiveData<String> = savedStateHandle.getLiveData(KEY_ARGUMENT_A)
}