On the lower left of the Studio window there’s a docked view called “Build Variants”.
Open it and choose whichever variant you want to work on. This will impact which variant is built but also the enabled source folders, which will drive completion in code and resources editors.