I'm trying to write some custom rules, but the IDE cannot complete resolve any of the classes in the

org.jetbrains.kotlin.psi

package. Everything compiles but the IDE not finding the classes make it hard to use. Is there something I need to configure to get this working?

This is a project with a custom rule related with JUnit. IntelliJ had no issues finding all the classes: https://github.com/BraisGabin/detekt-junit-rules check if there is any difference in build.gradle

I don't see anything that jumps out. Not sure if it is me using Android Studio (need to upgrade to latest) or that this project is part of a multi platform project. But I got it to almost work by added kotlin-compiler and kotlin-compiler-embedded as a compile time only dependency. This is completely wrong but it is working for now.