It's not like this game has real AI.
It's a set of rules they follows. A sort of decision tree probably.
So it's just a matter of adding more checks.
Situational Awareness implies real cognition. If that's what the game is doing, then Larian is wasting talent.
Probably the list of conditions the AI checks for and has rules to handle is way to limited.
If self.checkForDebuf("Blind") then Do "HELP I'm Blind routine"
If ally[i].checkForDebuf("Blind") then Do "try to heal the blind routine"
routine "HELP I'm Blind"
if self.hasHealBlindness() then DUH heal myself
if self.hasSpellType(AOE) then BLAST AWAY and hope we take out more enemies than allies
ect
For a game AI to "lack situational awareness" usually just means the combat decision tree is incomplete