Add method to get entity to entity AI interface