A rule of thumb for a productive career: work with people you want to spend time with.

As long as the reason you like spending time with them is not just social, but also that they inspire you, you don't need much more.

That's all there is to it.