Make it like in the alpha where if the Ai lost sight of you it wouldn't instantly lose you but have a timer that will count down from something like 20 and when the Ai would spot you again the timer will reset and freeze until it loses sight of you where it will start counting down again
So basically the AI will still chase you when they lose sight of you but when that timer runs out they will lose you