Kotlin Gradle Script add generated kotlin sources

I’m trying to use grpc kotlin. Generating sources from proto in src/main/generated folder.

Following gradle snippet adds java sources in compilation path

sourceSets {
	main {
		java {
			srcDir("src/main/generated/main/java")
			srcDir("src/main/generated/main/grpc")
		}
		proto {
			srcDir("src/main/social-model")
		}
	}
}

However, couldn’t find any way to add kotlin sources in a kotlin gradle script file…
sourceSets { main { kotlin { } } } doesn’t work.

https://kotlinlang.org/docs/reference/using-gradle.html

kotlin {
sourceSets[“main”].apply {
kotlin.srcDir(“src/main/myKotlin”)
}
}