I would check out if Not Diamond (no personal affiliation) [0] fits your use case. They had a really interesting blog post recently where they open sourced their older model router [1].
[0] https://www.notdiamond.ai/features
[1] Blog - Routing on Random Forests - https://www.notdiamond.ai/blog/rorf