Expand search for a row's entity and hold to declared return type

https://www.drupal.org/project/swiper_formatter/issues/3500394

Closes #3500394