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