Script demonstrating the use of @AssistedInject & @AssistedFactory in case parameterized viewmodel

SampleActivity.kt

@AndroidEntryPoint
class SampleActivity : AppCompatActivity() {
  
  @Inject lateinit var factory: SampleViewModelFactory
  
  val viewModel: SampleViewModel by viewModels {
        SampleViewModel.provideFactory(
            factory,
            "data"
        )
    }
}

SampleRepository.kt

class SampleRepository @Inject constructor {
    
  ........
  ........
  
}

SampleViewModel.kt

@HiltViewModel
class SampleViewModel @AssistedInject constructor(
    @Assisted("key") key : String?,
    var sampleRepository: SampleRepository
) : ViewModel() {
  
  ........
  ........
  
  
  companion object {

        fun provideFactory(
            assistedFactory: SampleViewModelFactory,
            key: String?
        ): ViewModelProvider.Factory = object : ViewModelProvider.Factory {
            @Suppress("UNCHECKED_CAST")
            override fun <T : ViewModel> create(modelClass: Class<T>): T {
                return assistedFactory.create(key) as T
            }
        }
    }

}

SampleViewModelFactory.kt

@AssistedFactory
interface SampleViewModelFactory {
    fun create(
        @Assisted("key") key: String?
    ) : SampleViewModel
}