Add methods for getting specific entity AI