I am assuming that would be solved by having the AI also take in inputs of where your troops and allies are located. Perhaps with something like the Blue Force Tracker [0].

One of the first priorities of an operation is not knowing where your enemy is, but where you are.

0: https://www.viasat.com/products/blue-force-tracking-2