I know this could be very hard to do because you would have to create different entities with different recipes for each quality, but you could change the success rate to be dependent of the quality of the augmentor like you use the mining_drill_resource_drain_multiplier of the quality to determine more success of the machine.
