Add AI state debugging