I've seen KK a few times....
Agassi was imitating KK during that particular match to get under KKs skin -- pure gamesmanship. Not as a tit-for-tat.
Obviously I can only infer his mental state. But at least I have some basis for doing so. I saw the match.