Gradle team created and maintains their own Gradle commands completion. Maybe it's worth to bundle their implementation as a plugin to oh-my-zsh. Default Gradle support does not look alive to me.

Related issue
gradle/gradle-completion#13