Create base AI state classes