refactor: setup sub-agen helpers for apply usage

Refactors the code with no behavior changes in order to make the
following behavior change as easy as possible

Author: sigilioso