Instead of gaining mana from matching gems on the board, maybe some troops could only gain mana from 4- and 5-gem matches (i.e. on extra turns). If these troops could also charge during the opponent’s turn, then it would offer the AI a bit of a counter to control teams.
Edit: dang. Didn’t think about how this would work with traitstones. This is left as an exercise to the reader.