After upgrade to latest IntelliJ for Android Studio Plugin, the BuildFileKey class symbol it not found

Hi,

I'm working on Android Studio plugin project, below class is no longer existed in the current IntelliJ community edition, is there a replacement or doc about the change?  Thanks!

import com.android.tools.idea.gradle.parser.BuildFileKey