Getting arg from ViewModel with Hilt assisted injection

composables.kt

@HiltViewModel(assistedFactory = DetailsViewModel.Factory::class)
class DetailsViewModel @AssistedInject constructor(
  @Assisted val arg: String,
) : ViewModel() {
  @AssistedFactory interface Factory {
    fun create(arg: String): DetailsViewModel
  }

  private val state = MutableStateFlow(DetailsUiState(arg = arg))
  fun state() = state.asStateFlow()
}

data class DetailsUiState(val arg: String)

@Composable
fun App(
  modifier: Modifier = Modifier,
) {
  Box(
    modifier = modifier
      .fillMaxSize()
      .background(color = colorScheme.background),
  ) {
    val navController = rememberNavController()
    NavHost(navController = navController, startDestination = "start") {
      composable(route = "start") {
        StartScreen(
          onNavigateClick = { arg -> navController.navigate(route = "details/$arg") },
        )
      }
      composable(route = "details/{arg}") { navBackStackEntry ->
        val arg = requireNotNull(navBackStackEntry.arguments?.getString("arg"))
        val viewModel = hiltViewModel<DetailsViewModel, DetailsViewModel.Factory>(
          creationCallback = { factory -> factory.create(arg = arg) },
        )
        val state by viewModel.state().collectAsStateWithLifecycle(lifecycleOwner = navBackStackEntry)
        DetailsScreen(arg = state.arg)
      }
    }
  }
}