That requires tool use or some similar specific action at inference time.

The technique I suggested would, I think, work on existing model inference methods. The ability already exists in the architecture. It's just a training adjustment to produce the parameters required to do so.