You might want to check out the guidance DSL. It is conceptually similar to lmql but works on a lower level than lmql.

https://github.com/guidance-ai/guidance/

The library used to be developed by Microsoft. It seems like they spun it of as a independent company or nonprofit a few months ago.