wip: add augment api