I was able to find the vnet by looking at the agent profiles:

for (pool <- k.innerModel().agentPoolProfiles().asScala) { 
val subnet = pool.vnetSubnetId()
}