I’m pretty sure you’re going to get the same system that they released on console. I just can’t see them going back to the drawing board on this when they’ve already released an AI customization interface on one of their platforms.
If you don’t know, you can customize your defence team AI using 7 sliders that increase or decrease the priority of each of the 7 gem types (6 colours plus skulls). Apparently, these sliders not only affect the priority for gem matching, but also influence spell selection in some unspecified way. I’ll see if I can find the dev post that described this and link it below if I do. It doesn’t do exactly what you’re describing, but it does have a noticeable impact on gameplay.
Edit:
Here is the post I was thinking of. It is in response to a question of whether changing the gem priority sliders would also affect the choice of gem colour that the Guardians would choose as the 2nd colour when casting their spells. This was an effect that I don’t think any of us had ever thought about before.