This is exactly what I was looking for!
They feature an "Ask AI" widget in the bottom-right of their own docs site (https://docs.kapa.ai/) and some of their customers are doing the same: https://www.prisma.io/docs | https://docs.mapbox.com/ | https://circleci.com/docs/ | https://fusionauth.io/docs/