ML agents difficulty modulation for game

I’m considering using Reinforcement Learning (like Unity ML-Agents) to train the enemy bots for my game instead of coding traditional behavior trees/FSMs. I want the bots to feel like they’re making smart human decisions in a complex environment, but I also want a good control over how difficult they are. Do you have ideas on how to achieve this?