Knowing me, knowing you: Theory of mind in AI