Have you considered defining custom KaTeX macros instead? I created an example here: https://observablehq.com/d/c19092b8c4686e1b

If I had known they were a thing. Which I do now. Thanks!